@itwin/tree-widget-react 4.0.0-alpha.12 → 4.0.0-alpha.14

package/README.md

@@ -201,8 +201,6 @@ Based on the returned paths, the displayed hierarchy consists only of the target
 Use `getFilteredPaths` when you need more control over which nodes are shown. Here are some example use cases:
 
 - **Filter by known instance keys**: You already have a list of `InstanceKey` items that should remain in the tree. Pass them as `targetItems` to `createInstanceKeyPaths`.
-  <!-- [[include: [TreeWidget.GetFilteredPathsComponentWithTargetItemsExample], tsx]] -->
-  <!-- BEGIN EXTRACTION -->
 
   ```tsx
   type UseModelsTreeProps = Parameters<typeof useModelsTree>[0];
@@ -243,11 +241,7 @@ Use `getFilteredPaths` when you need more control over which nodes are shown. He
   }
   ```
 
-  <!-- END EXTRACTION -->
-
 - **Post-process the paths created `createInstanceKeyPaths`**: Use `filter` string to generate the paths, then apply additional filtering - e.g., remove paths that are too long.
-  <!-- [[include: [TreeWidget.GetFilteredPathsComponentWithPostProcessingExample], tsx]] -->
-  <!-- BEGIN EXTRACTION -->
 
   ```tsx
   function CustomModelsTreeComponentWithPostProcessing({
@@ -286,11 +280,8 @@ Use `getFilteredPaths` when you need more control over which nodes are shown. He
   }
   ```
 
-  <!-- END EXTRACTION -->
-
 - **Apply custom logic to generate instance keys**: Generate instance keys using custom implementation. For example: query elements that have specified filter in their user label and provide them as targetItems.
-    <!-- [[include: [TreeWidget.GetFilteredPathsComponentWithFilterAndTargetItemsExample], tsx]] -->
-    <!-- BEGIN EXTRACTION -->
+
   ```tsx
   function CustomModelsTreeComponentWithFilterAndTargetItems({
     viewport,
@@ -333,7 +324,6 @@ Use `getFilteredPaths` when you need more control over which nodes are shown. He
     );
   }
   ```
-    <!-- END EXTRACTION -->
 
 ### Categories tree
 

package/lib/esm/tree-widget-react/components/TreeWidgetUiItemsProvider.js.map

@@ -1 +1 @@
-{"version":3,"file":"TreeWidgetUiItemsProvider.js","sourceRoot":"","sources":["../../../../src/tree-widget-react/components/TreeWidgetUiItemsProvider.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,iCAAiC,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAC/B,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAC9F,OAAO,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAC9C,OAAO,gBAAgB,MAAM,qCAAqC,CAAC;AACnE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AACzD,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAyBvE;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAsB;IACrD,OAAO;QACL,EAAE,EAAE,yBAAyB;QAC7B,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,cAAc,CAAC;QAC3C,IAAI,EAAE,KAAC,IAAI,IAAC,IAAI,EAAE,gBAAgB,GAAI;QACtC,OAAO,EAAE;YACP,QAAQ,EAAE;gBACR,OAAO,EAAE,iBAAiB,CAAC,KAAK;gBAChC,QAAQ,EAAE,kBAAkB,CAAC,KAAK;aACnC;SACF;QACD,OAAO,EAAE,KAAC,mBAAmB,IAAC,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,qBAAqB,EAAE,KAAK,CAAC,qBAAqB,EAAE,aAAa,EAAE,KAAK,CAAC,aAAa,GAAI;KAC7I,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAAsB;IACxD,MAAM,GAAG,GAAG,2BAA2B,EAAE,CAAC;IAC1C,OAAO,CACL,cAAK,GAAG,EAAE,GAAG,EAAE,SAAS,EAAC,aAAa,YACpC,KAAC,aAAa,IAAC,iBAAiB,EAAE,UAAU,YAC1C,KAAC,uBAAuB,OAAK,KAAK,GAAI,GACxB,GACZ,CACP,CAAC;AACJ,CAAC;AAED,SAAS,2BAA2B;IAClC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,cAAc,EAAE,CAAC;IACnD,iBAAiB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACpC,OAAO,GAAG,CAAC;AACb,CAAC;AAQD,SAAS,cAAc;IACrB,MAAM,GAAG,GAAG,MAAM,CAAiB,IAAI,CAAC,CAAC;IACzC,MAAM,SAAS,GAAG,MAAM,EAAsB,CAAC;IAE/C,MAAM,YAAY,GAAG,GAAG,EAAE;QACxB,OAAO,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,6BAA6B,CAAC,CAAC;IACnE,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QACjC,SAAS,CAAC,OAAO,GAAG,SAAS,EAAE,SAAS,CAAC;IAC3C,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;YACjC,IAAI,SAAS,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;gBACnC,SAAS,CAAC,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC;YAC1C,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;AACnC,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 \"./TreeWidgetUiItemsProvider.css\";\nimport { useRef } from \"react\";\nimport { ErrorBoundary } from \"react-error-boundary\";\nimport { StagePanelLocation, StagePanelSection, useTransientState } from \"@itwin/appui-react\";\nimport { Icon } from \"@stratakit/foundations\";\nimport hierarchyTreeSvg from \"@stratakit/icons/hierarchy-tree.svg\";\nimport { TreeWidget } from \"../TreeWidget.js\";\nimport { ErrorState } from \"./tree-header/ErrorState.js\";\nimport { TreeWidgetComponentImpl } from \"./TreeWidgetComponentImpl.js\";\n\nimport type { Ref } from \"react\";\nimport type { Widget } from \"@itwin/appui-react\";\nimport type { TreeDefinition } from \"./TreeWidgetComponentImpl.js\";\n/**\n * Props for `createWidget`.\n * @public\n */\ninterface TreeWidgetProps {\n  /**\n   * Trees to show in the widget.\n   * @see ModelsTreeComponent\n   * @see CategoriesTreeComponent\n   * @see ExternalSourcesTreeComponent\n   * @see IModelContentTreeComponent\n   */\n  trees: TreeDefinition[];\n  /** Callback that is invoked when performance of tracked feature is measured. */\n  onPerformanceMeasured?: (feature: string, elapsedTime: number) => void;\n  /** Callback that is invoked when a tracked feature is used. */\n  onFeatureUsed?: (feature: string) => void;\n}\n\n\n/**\n * Creates a tree widget definition that should be returned from `UiItemsProvider.getWidgets()`.\n * @public\n */\nexport function createTreeWidget(props: TreeWidgetProps): Widget {\n  return {\n    id: \"tree-widget-react:trees\",\n    label: TreeWidget.translate(\"widget.label\"),\n    icon: <Icon href={hierarchyTreeSvg} />,\n    layouts: {\n      standard: {\n        section: StagePanelSection.Start,\n        location: StagePanelLocation.Right,\n      },\n    },\n    content: <TreeWidgetComponent trees={props.trees} onPerformanceMeasured={props.onPerformanceMeasured} onFeatureUsed={props.onFeatureUsed} />,\n  };\n}\n\n/**\n * Tree widget component which allows selecting which tree to render.\n * @public\n */\nexport function TreeWidgetComponent(props: TreeWidgetProps) {\n  const ref = useTreeWidgetTransientState();\n  return (\n    <div ref={ref} className=\"tree-widget\">\n      <ErrorBoundary FallbackComponent={ErrorState}>\n        <TreeWidgetComponentImpl {...props} />\n      </ErrorBoundary>\n    </div>\n  );\n}\n\nfunction useTreeWidgetTransientState() {\n  const { ref, persist, restore } = useTreeStorage();\n  useTransientState(persist, restore);\n  return ref;\n}\n\ninterface UseTreeStorageResult {\n  ref: Ref<HTMLDivElement>;\n  persist: () => void;\n  restore: () => void;\n}\n\nfunction useTreeStorage(): UseTreeStorageResult {\n  const ref = useRef<HTMLDivElement>(null);\n  const scrollTop = useRef<number | undefined>();\n\n  const getContainer = () => {\n    return ref.current?.querySelector(\"#tw-tree-renderer-container\");\n  };\n\n  const persist = () => {\n    const container = getContainer();\n    scrollTop.current = container?.scrollTop;\n  };\n\n  const restore = () => {\n    setTimeout(() => {\n      const container = getContainer();\n      if (container && scrollTop.current) {\n        container.scrollTop = scrollTop.current;\n      }\n    });\n  };\n\n  return { ref, persist, restore };\n}\n"]}
+{"version":3,"file":"TreeWidgetUiItemsProvider.js","sourceRoot":"","sources":["../../../../src/tree-widget-react/components/TreeWidgetUiItemsProvider.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,iCAAiC,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAC/B,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAC9F,OAAO,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAC9C,OAAO,gBAAgB,MAAM,qCAAqC,CAAC;AACnE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AACzD,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAwBvE;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAsB;IACrD,OAAO;QACL,EAAE,EAAE,yBAAyB;QAC7B,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,cAAc,CAAC;QAC3C,IAAI,EAAE,KAAC,IAAI,IAAC,IAAI,EAAE,gBAAgB,GAAI;QACtC,OAAO,EAAE;YACP,QAAQ,EAAE;gBACR,OAAO,EAAE,iBAAiB,CAAC,KAAK;gBAChC,QAAQ,EAAE,kBAAkB,CAAC,KAAK;aACnC;SACF;QACD,OAAO,EAAE,KAAC,mBAAmB,IAAC,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,qBAAqB,EAAE,KAAK,CAAC,qBAAqB,EAAE,aAAa,EAAE,KAAK,CAAC,aAAa,GAAI;KAC7I,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAAsB;IACxD,MAAM,GAAG,GAAG,2BAA2B,EAAE,CAAC;IAC1C,OAAO,CACL,cAAK,GAAG,EAAE,GAAG,EAAE,SAAS,EAAC,aAAa,YACpC,KAAC,aAAa,IAAC,iBAAiB,EAAE,UAAU,YAC1C,KAAC,uBAAuB,OAAK,KAAK,GAAI,GACxB,GACZ,CACP,CAAC;AACJ,CAAC;AAED,SAAS,2BAA2B;IAClC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,cAAc,EAAE,CAAC;IACnD,iBAAiB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACpC,OAAO,GAAG,CAAC;AACb,CAAC;AAQD,SAAS,cAAc;IACrB,MAAM,GAAG,GAAG,MAAM,CAAiB,IAAI,CAAC,CAAC;IACzC,MAAM,SAAS,GAAG,MAAM,EAAsB,CAAC;IAE/C,MAAM,YAAY,GAAG,GAAG,EAAE;QACxB,OAAO,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,6BAA6B,CAAC,CAAC;IACnE,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QACjC,SAAS,CAAC,OAAO,GAAG,SAAS,EAAE,SAAS,CAAC;IAC3C,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;YACjC,IAAI,SAAS,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;gBACnC,SAAS,CAAC,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC;YAC1C,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;AACnC,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 \"./TreeWidgetUiItemsProvider.css\";\nimport { useRef } from \"react\";\nimport { ErrorBoundary } from \"react-error-boundary\";\nimport { StagePanelLocation, StagePanelSection, useTransientState } from \"@itwin/appui-react\";\nimport { Icon } from \"@stratakit/foundations\";\nimport hierarchyTreeSvg from \"@stratakit/icons/hierarchy-tree.svg\";\nimport { TreeWidget } from \"../TreeWidget.js\";\nimport { ErrorState } from \"./tree-header/ErrorState.js\";\nimport { TreeWidgetComponentImpl } from \"./TreeWidgetComponentImpl.js\";\n\nimport type { Ref } from \"react\";\nimport type { Widget } from \"@itwin/appui-react\";\nimport type { TreeDefinition } from \"./TreeWidgetComponentImpl.js\";\n/**\n * Props for `createWidget`.\n * @public\n */\ninterface TreeWidgetProps {\n  /**\n   * Trees to show in the widget.\n   * @see ModelsTreeComponent\n   * @see CategoriesTreeComponent\n   * @see ExternalSourcesTreeComponent\n   * @see IModelContentTreeComponent\n   */\n  trees: TreeDefinition[];\n  /** Callback that is invoked when performance of tracked feature is measured. */\n  onPerformanceMeasured?: (feature: string, elapsedTime: number) => void;\n  /** Callback that is invoked when a tracked feature is used. */\n  onFeatureUsed?: (feature: string) => void;\n}\n\n/**\n * Creates a tree widget definition that should be returned from `UiItemsProvider.getWidgets()`.\n * @public\n */\nexport function createTreeWidget(props: TreeWidgetProps): Widget {\n  return {\n    id: \"tree-widget-react:trees\",\n    label: TreeWidget.translate(\"widget.label\"),\n    icon: <Icon href={hierarchyTreeSvg} />,\n    layouts: {\n      standard: {\n        section: StagePanelSection.Start,\n        location: StagePanelLocation.Right,\n      },\n    },\n    content: <TreeWidgetComponent trees={props.trees} onPerformanceMeasured={props.onPerformanceMeasured} onFeatureUsed={props.onFeatureUsed} />,\n  };\n}\n\n/**\n * Tree widget component which allows selecting which tree to render.\n * @public\n */\nexport function TreeWidgetComponent(props: TreeWidgetProps) {\n  const ref = useTreeWidgetTransientState();\n  return (\n    <div ref={ref} className=\"tree-widget\">\n      <ErrorBoundary FallbackComponent={ErrorState}>\n        <TreeWidgetComponentImpl {...props} />\n      </ErrorBoundary>\n    </div>\n  );\n}\n\nfunction useTreeWidgetTransientState() {\n  const { ref, persist, restore } = useTreeStorage();\n  useTransientState(persist, restore);\n  return ref;\n}\n\ninterface UseTreeStorageResult {\n  ref: Ref<HTMLDivElement>;\n  persist: () => void;\n  restore: () => void;\n}\n\nfunction useTreeStorage(): UseTreeStorageResult {\n  const ref = useRef<HTMLDivElement>(null);\n  const scrollTop = useRef<number | undefined>();\n\n  const getContainer = () => {\n    return ref.current?.querySelector(\"#tw-tree-renderer-container\");\n  };\n\n  const persist = () => {\n    const container = getContainer();\n    scrollTop.current = container?.scrollTop;\n  };\n\n  const restore = () => {\n    setTimeout(() => {\n      const container = getContainer();\n      if (container && scrollTop.current) {\n        container.scrollTop = scrollTop.current;\n      }\n    });\n  };\n\n  return { ref, persist, restore };\n}\n"]}

package/lib/esm/tree-widget-react/components/tree-header/WidgetHeader.css

@@ -13,7 +13,7 @@
     display: flex;
     vertical-align: middle;
     padding: var(--iui-size-s) var(--iui-size-s) var(--iui-size-xs) var(--iui-size-s);
-    gap:  var(--iui-size-s);
+    gap: var(--iui-size-s);
 
     .tw-content-header-selector {
       width: 100%;

package/lib/esm/tree-widget-react/components/trees/categories-tree/UseCategoriesTree.js

@@ -3,7 +3,8 @@ import { jsx as _jsx } from "react/jsx-runtime";
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { useCallback, useEffect, useMemo, useState } from "react";
+import { useCallback, useEffect, useMemo } from "react";
+import { createECSqlQueryExecutor } from "@itwin/presentation-core-interop";
 import { Icon } from "@stratakit/foundations";
 import categorySvg from "@stratakit/icons/bis-category-3d.svg";
 import subcategorySvg from "@stratakit/icons/bis-category-subcategory.svg";
@@ -11,10 +12,12 @@ import classSvg from "@stratakit/icons/bis-class.svg";
 import definitionContainerSvg from "@stratakit/icons/bis-definitions-container.svg";
 import elementSvg from "@stratakit/icons/bis-element.svg";
 import { EmptyTreeContent, FilterUnknownError, NoFilterMatches, TooManyFilterMatches } from "../common/components/EmptyTree.js";
+import { useCachedVisibility } from "../common/internal/useTreeHooks/UseCachedVisibility.js";
+import { useIdsCache } from "../common/internal/useTreeHooks/UseIdsCache.js";
 import { CategoriesTreeDefinition, defaultHierarchyConfiguration } from "./CategoriesTreeDefinition.js";
+import { CategoriesTreeIdsCache } from "./internal/CategoriesTreeIdsCache.js";
 import { createCategoriesTreeVisibilityHandler } from "./internal/CategoriesTreeVisibilityHandler.js";
 import { useFilteredPaths } from "./internal/UseFilteredPaths.js";
-import { useIdsCache } from "./internal/UseIdsCache.js";
 /**
  * Custom hook to create and manage state for the categories tree.
  * @beta
@@ -27,7 +30,16 @@ export function useCategoriesTree({ filter, activeView, onCategoriesFiltered, em
     // eslint-disable-next-line react-hooks/exhaustive-deps
     Object.values(hierarchyConfig ?? {}));
     const viewType = activeView.view.is2d() ? "2d" : "3d";
-    const { getCategoriesTreeIdsCache, visibilityHandlerFactory, onFilteredPathsChanged } = useCachedVisibility(activeView, hierarchyConfiguration, viewType);
+    const { getCache: getCategoriesTreeIdsCache } = useIdsCache({
+        imodel: activeView.iModel,
+        createCache,
+        cacheSpecificProps: useMemo(() => ({ viewType }), [viewType]),
+    });
+    const { visibilityHandlerFactory, onFilteredPathsChanged } = useCategoriesCachedVisibility({
+        activeView,
+        getCache: getCategoriesTreeIdsCache,
+        hierarchyConfig: hierarchyConfiguration,
+    });
     const getHierarchyDefinition = useCallback((props) => {
         return new CategoriesTreeDefinition({ ...props, viewType, idsCache: getCategoriesTreeIdsCache(), hierarchyConfig: hierarchyConfiguration });
     }, [viewType, getCategoriesTreeIdsCache, hierarchyConfiguration]);
@@ -54,21 +66,15 @@ export function useCategoriesTree({ filter, activeView, onCategoriesFiltered, em
         },
     };
 }
-function createVisibilityHandlerFactory(activeView, idsCacheGetter, hierarchyConfig, filteredPaths) {
-    return ({ imodelAccess }) => createCategoriesTreeVisibilityHandler({ viewport: activeView, idsCache: idsCacheGetter(), imodelAccess, filteredPaths, hierarchyConfig });
-}
-function useCachedVisibility(activeView, hierarchyConfig, viewType) {
-    const [filteredPaths, setFilteredPaths] = useState();
-    const { getCache: getCategoriesTreeIdsCache } = useIdsCache(activeView.iModel, viewType, filteredPaths);
-    const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState(() => createVisibilityHandlerFactory(activeView, getCategoriesTreeIdsCache, hierarchyConfig, filteredPaths));
-    useEffect(() => {
-        setVisibilityHandlerFactory(() => createVisibilityHandlerFactory(activeView, getCategoriesTreeIdsCache, hierarchyConfig, filteredPaths));
-    }, [activeView, getCategoriesTreeIdsCache, hierarchyConfig, filteredPaths]);
-    return {
-        getCategoriesTreeIdsCache,
-        visibilityHandlerFactory,
-        onFilteredPathsChanged: useCallback((paths) => setFilteredPaths(paths), []),
-    };
+function createVisibilityHandlerFactory(props) {
+    const { activeView, factoryProps, idsCacheGetter, filteredPaths } = props;
+    return ({ imodelAccess }) => createCategoriesTreeVisibilityHandler({
+        viewport: activeView,
+        idsCache: idsCacheGetter(),
+        imodelAccess,
+        filteredPaths,
+        hierarchyConfig: factoryProps.hierarchyConfig,
+    });
 }
 function getEmptyTreeContentComponent(filter, error, emptyTreeContent) {
     if (error) {
@@ -111,4 +117,23 @@ export function CategoriesTreeIcon({ node }) {
 function getSublabel(node) {
     return node.nodeData.extendedData?.description;
 }
+function useCategoriesCachedVisibility(props) {
+    const { activeView, getCache, hierarchyConfig } = props;
+    const { visibilityHandlerFactory, filteredPaths, onFilteredPathsChanged } = useCachedVisibility({
+        activeView,
+        getCache,
+        factoryProps: useMemo(() => ({ hierarchyConfig }), [hierarchyConfig]),
+        createFactory: createVisibilityHandlerFactory,
+    });
+    useEffect(() => {
+        getCache().clearFilteredElementsModels();
+    }, [filteredPaths, getCache]);
+    return {
+        visibilityHandlerFactory,
+        onFilteredPathsChanged,
+    };
+}
+function createCache(props) {
+    return new CategoriesTreeIdsCache(createECSqlQueryExecutor(props.imodel), props.specificProps.viewType);
+}
 //# sourceMappingURL=UseCategoriesTree.js.map

package/lib/esm/tree-widget-react/components/trees/categories-tree/UseCategoriesTree.js.map

@@ -1 +1 @@
-{"version":3,"file":"UseCategoriesTree.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/categories-tree/UseCategoriesTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAClE,OAAO,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAC9C,OAAO,WAAW,MAAM,sCAAsC,CAAC;AAC/D,OAAO,cAAc,MAAM,+CAA+C,CAAC;AAC3E,OAAO,QAAQ,MAAM,gCAAgC,CAAC;AACtD,OAAO,sBAAsB,MAAM,gDAAgD,CAAC;AACpF,OAAO,UAAU,MAAM,kCAAkC,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AAChI,OAAO,EAAE,wBAAwB,EAAE,6BAA6B,EAAE,MAAM,+BAA+B,CAAC;AACxG,OAAO,EAAE,qCAAqC,EAAE,MAAM,+CAA+C,CAAC;AACtG,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAgCxD;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,EAChC,MAAM,EACN,UAAU,EACV,oBAAoB,EACpB,gBAAgB,EAChB,eAAe,GACQ;IACvB,MAAM,sBAAsB,GAAG,OAAO,CACpC,GAAG,EAAE,CAAC,CAAC;QACL,GAAG,6BAA6B;QAChC,GAAG,eAAe;KACnB,CAAC;IACF,uDAAuD;IACvD,MAAM,CAAC,MAAM,CAAC,eAAe,IAAI,EAAE,CAAC,CACrC,CAAC;IACF,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IAEtD,MAAM,EAAE,yBAAyB,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,GAAG,mBAAmB,CAAC,UAAU,EAAE,sBAAsB,EAAE,QAAQ,CAAC,CAAC;IAE1J,MAAM,sBAAsB,GAAG,WAAW,CACxC,CAAC,KAAK,EAAE,EAAE;QACR,OAAO,IAAI,wBAAwB,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,yBAAyB,EAAE,EAAE,eAAe,EAAE,sBAAsB,EAAE,CAAC,CAAC;IAC9I,CAAC,EACD,CAAC,QAAQ,EAAE,yBAAyB,EAAE,sBAAsB,CAAC,CAC9D,CAAC;IAEF,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,gBAAgB,CAAC;QACpD,sBAAsB;QACtB,MAAM;QACN,yBAAyB;QACzB,sBAAsB;QACtB,QAAQ;QACR,oBAAoB;KACrB,CAAC,CAAC;IAEH,OAAO;QACL,mBAAmB,EAAE;YACnB,QAAQ,EAAE,oBAAoB;YAC9B,sBAAsB;YACtB,gBAAgB,EAAE,QAAQ;YAC1B,wBAAwB;YACxB,gBAAgB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,4BAA4B,CAAC,MAAM,EAAE,cAAc,EAAE,gBAAgB,CAAC,EAAE,CAAC,MAAM,EAAE,cAAc,EAAE,gBAAgB,CAAC,CAAC;YACnJ,SAAS,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;SAC5E;QACD,aAAa,EAAE;YACb,cAAc,EAAE,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAC,kBAAkB,IAAC,IAAI,EAAE,IAAI,GAAI,EAAE,EAAE,CAAC;YAC7E,WAAW;SACZ;KACF,CAAC;AACJ,CAAC;AAED,SAAS,8BAA8B,CACrC,UAAoB,EACpB,cAA4C,EAC5C,eAAqD,EACrD,aAAwC;IAExC,OAAO,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAC1B,qCAAqC,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,cAAc,EAAE,EAAE,YAAY,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,CAAC;AAC9I,CAAC;AAED,SAAS,mBAAmB,CAAC,UAAoB,EAAE,eAAqD,EAAE,QAAqB;IAC7H,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,EAA4B,CAAC;IAC/E,MAAM,EAAE,QAAQ,EAAE,yBAAyB,EAAE,GAAG,WAAW,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC;IAExG,MAAM,CAAC,wBAAwB,EAAE,2BAA2B,CAAC,GAAG,QAAQ,CAAkD,GAAG,EAAE,CAC7H,8BAA8B,CAAC,UAAU,EAAE,yBAAyB,EAAE,eAAe,EAAE,aAAa,CAAC,CACtG,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,2BAA2B,CAAC,GAAG,EAAE,CAAC,8BAA8B,CAAC,UAAU,EAAE,yBAAyB,EAAE,eAAe,EAAE,aAAa,CAAC,CAAC,CAAC;IAC3I,CAAC,EAAE,CAAC,UAAU,EAAE,yBAAyB,EAAE,eAAe,EAAE,aAAa,CAAC,CAAC,CAAC;IAE5E,OAAO;QACL,yBAAyB;QACzB,wBAAwB;QACxB,sBAAsB,EAAE,WAAW,CAAC,CAAC,KAA2C,EAAE,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;KAClH,CAAC;AACJ,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAe,EAAE,KAAoC,EAAE,gBAAkC;IAC7H,IAAI,KAAK,EAAE,CAAC;QACV,IAAI,KAAK,KAAK,sBAAsB,EAAE,CAAC;YACrC,OAAO,KAAC,oBAAoB,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;QAC1D,CAAC;QACD,OAAO,KAAC,kBAAkB,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;IACxD,CAAC;IACD,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,KAAC,eAAe,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;IACrD,CAAC;IACD,IAAI,gBAAgB,EAAE,CAAC;QACrB,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IACD,OAAO,KAAC,gBAAgB,IAAC,IAAI,EAAE,WAAW,GAAI,CAAC;AACjD,CAAC;AAED,YAAY;AACZ,MAAM,UAAU,kBAAkB,CAAC,EAAE,IAAI,EAAuC;IAC9E,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;QACtD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,QAAQ,IAAI,CAAC,QAAQ,CAAC,YAAa,CAAC,OAAO,EAAE,CAAC;YAC5C,KAAK,aAAa;gBAChB,OAAO,WAAW,CAAC;YACrB,KAAK,qBAAqB;gBACxB,OAAO,cAAc,CAAC;YACxB,KAAK,2BAA2B;gBAC9B,OAAO,sBAAsB,CAAC;YAChC,KAAK,WAAW;gBACd,OAAO,UAAU,CAAC;YACpB,KAAK,eAAe;gBAClB,OAAO,QAAQ,CAAC;YAClB;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,KAAC,IAAI,IAAC,IAAI,EAAE,OAAO,EAAE,GAAI,CAAC;AACnC,CAAC;AAED,SAAS,WAAW,CAAC,IAA+B;IAClD,OAAO,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,WAAW,CAAC;AACjD,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useEffect, useMemo, useState } from \"react\";\nimport { Icon } from \"@stratakit/foundations\";\nimport categorySvg from \"@stratakit/icons/bis-category-3d.svg\";\nimport subcategorySvg from \"@stratakit/icons/bis-category-subcategory.svg\";\nimport classSvg from \"@stratakit/icons/bis-class.svg\";\nimport definitionContainerSvg from \"@stratakit/icons/bis-definitions-container.svg\";\nimport elementSvg from \"@stratakit/icons/bis-element.svg\";\nimport { EmptyTreeContent, FilterUnknownError, NoFilterMatches, TooManyFilterMatches } from \"../common/components/EmptyTree.js\";\nimport { CategoriesTreeDefinition, defaultHierarchyConfiguration } from \"./CategoriesTreeDefinition.js\";\nimport { createCategoriesTreeVisibilityHandler } from \"./internal/CategoriesTreeVisibilityHandler.js\";\nimport { useFilteredPaths } from \"./internal/UseFilteredPaths.js\";\nimport { useIdsCache } from \"./internal/UseIdsCache.js\";\n\nimport type { CategoriesTreeFilteringError } from \"./internal/UseFilteredPaths.js\";\nimport type { HierarchyFilteringPath } from \"@itwin/presentation-hierarchies\";\nimport type { CategoriesTreeIdsCache } from \"./internal/CategoriesTreeIdsCache.js\";\nimport type { ReactNode } from \"react\";\nimport type { Id64Array } from \"@itwin/core-bentley\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { VisibilityTreeProps } from \"../common/components/VisibilityTree.js\";\nimport type { VisibilityTreeRendererProps } from \"../common/components/VisibilityTreeRenderer.js\";\nimport type { CategoryInfo } from \"../common/CategoriesVisibilityUtils.js\";\nimport type { CategoriesTreeHierarchyConfiguration } from \"./CategoriesTreeDefinition.js\";\n\n/** @beta */\nexport interface UseCategoriesTreeProps {\n  activeView: Viewport;\n  onCategoriesFiltered?: (props: { categories: CategoryInfo[] | undefined; models?: Id64Array }) => void;\n  filter?: string;\n  emptyTreeContent?: ReactNode;\n  hierarchyConfig?: Partial<CategoriesTreeHierarchyConfiguration>;\n}\n\n/** @beta */\ninterface UseCategoriesTreeResult {\n  categoriesTreeProps: Pick<\n    VisibilityTreeProps,\n    \"treeName\" | \"getHierarchyDefinition\" | \"getFilteredPaths\" | \"visibilityHandlerFactory\" | \"highlight\" | \"emptyTreeContent\"\n  >;\n  rendererProps: Required<Pick<VisibilityTreeRendererProps, \"getDecorations\" | \"getSublabel\">>;\n}\n\n/**\n * Custom hook to create and manage state for the categories tree.\n * @beta\n */\nexport function useCategoriesTree({\n  filter,\n  activeView,\n  onCategoriesFiltered,\n  emptyTreeContent,\n  hierarchyConfig,\n}: UseCategoriesTreeProps): UseCategoriesTreeResult {\n  const hierarchyConfiguration = useMemo<CategoriesTreeHierarchyConfiguration>(\n    () => ({\n      ...defaultHierarchyConfiguration,\n      ...hierarchyConfig,\n    }),\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    Object.values(hierarchyConfig ?? {}),\n  );\n  const viewType = activeView.view.is2d() ? \"2d\" : \"3d\";\n\n  const { getCategoriesTreeIdsCache, visibilityHandlerFactory, onFilteredPathsChanged } = useCachedVisibility(activeView, hierarchyConfiguration, viewType);\n\n  const getHierarchyDefinition = useCallback<VisibilityTreeProps[\"getHierarchyDefinition\"]>(\n    (props) => {\n      return new CategoriesTreeDefinition({ ...props, viewType, idsCache: getCategoriesTreeIdsCache(), hierarchyConfig: hierarchyConfiguration });\n    },\n    [viewType, getCategoriesTreeIdsCache, hierarchyConfiguration],\n  );\n\n  const { getPaths, filteringError } = useFilteredPaths({\n    hierarchyConfiguration,\n    filter,\n    getCategoriesTreeIdsCache,\n    onFilteredPathsChanged,\n    viewType,\n    onCategoriesFiltered,\n  });\n\n  return {\n    categoriesTreeProps: {\n      treeName: \"categories-tree-v2\",\n      getHierarchyDefinition,\n      getFilteredPaths: getPaths,\n      visibilityHandlerFactory,\n      emptyTreeContent: useMemo(() => getEmptyTreeContentComponent(filter, filteringError, emptyTreeContent), [filter, filteringError, emptyTreeContent]),\n      highlight: useMemo(() => (filter ? { text: filter } : undefined), [filter]),\n    },\n    rendererProps: {\n      getDecorations: useCallback((node) => <CategoriesTreeIcon node={node} />, []),\n      getSublabel,\n    },\n  };\n}\n\nfunction createVisibilityHandlerFactory(\n  activeView: Viewport,\n  idsCacheGetter: () => CategoriesTreeIdsCache,\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration,\n  filteredPaths?: HierarchyFilteringPath[],\n): VisibilityTreeProps[\"visibilityHandlerFactory\"] {\n  return ({ imodelAccess }) =>\n    createCategoriesTreeVisibilityHandler({ viewport: activeView, idsCache: idsCacheGetter(), imodelAccess, filteredPaths, hierarchyConfig });\n}\n\nfunction useCachedVisibility(activeView: Viewport, hierarchyConfig: CategoriesTreeHierarchyConfiguration, viewType: \"2d\" | \"3d\") {\n  const [filteredPaths, setFilteredPaths] = useState<HierarchyFilteringPath[]>();\n  const { getCache: getCategoriesTreeIdsCache } = useIdsCache(activeView.iModel, viewType, filteredPaths);\n\n  const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState<VisibilityTreeProps[\"visibilityHandlerFactory\"]>(() =>\n    createVisibilityHandlerFactory(activeView, getCategoriesTreeIdsCache, hierarchyConfig, filteredPaths),\n  );\n\n  useEffect(() => {\n    setVisibilityHandlerFactory(() => createVisibilityHandlerFactory(activeView, getCategoriesTreeIdsCache, hierarchyConfig, filteredPaths));\n  }, [activeView, getCategoriesTreeIdsCache, hierarchyConfig, filteredPaths]);\n\n  return {\n    getCategoriesTreeIdsCache,\n    visibilityHandlerFactory,\n    onFilteredPathsChanged: useCallback((paths: HierarchyFilteringPath[] | undefined) => setFilteredPaths(paths), []),\n  };\n}\n\nfunction getEmptyTreeContentComponent(filter?: string, error?: CategoriesTreeFilteringError, emptyTreeContent?: React.ReactNode) {\n  if (error) {\n    if (error === \"tooManyFilterMatches\") {\n      return <TooManyFilterMatches base={\"categoriesTree\"} />;\n    }\n    return <FilterUnknownError base={\"categoriesTree\"} />;\n  }\n  if (filter) {\n    return <NoFilterMatches base={\"categoriesTree\"} />;\n  }\n  if (emptyTreeContent) {\n    return emptyTreeContent;\n  }\n  return <EmptyTreeContent icon={categorySvg} />;\n}\n\n/** @beta */\nexport function CategoriesTreeIcon({ node }: { node: PresentationHierarchyNode }) {\n  if (node.nodeData.extendedData?.imageId === undefined) {\n    return undefined;\n  }\n\n  const getIcon = () => {\n    switch (node.nodeData.extendedData!.imageId) {\n      case \"icon-layers\":\n        return categorySvg;\n      case \"icon-layers-isolate\":\n        return subcategorySvg;\n      case \"icon-definition-container\":\n        return definitionContainerSvg;\n      case \"icon-item\":\n        return elementSvg;\n      case \"icon-ec-class\":\n        return classSvg;\n      default:\n        return undefined;\n    }\n  };\n\n  return <Icon href={getIcon()} />;\n}\n\nfunction getSublabel(node: PresentationHierarchyNode) {\n  return node.nodeData.extendedData?.description;\n}\n"]}
+{"version":3,"file":"UseCategoriesTree.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/categories-tree/UseCategoriesTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AACxD,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAC9C,OAAO,WAAW,MAAM,sCAAsC,CAAC;AAC/D,OAAO,cAAc,MAAM,+CAA+C,CAAC;AAC3E,OAAO,QAAQ,MAAM,gCAAgC,CAAC;AACtD,OAAO,sBAAsB,MAAM,gDAAgD,CAAC;AACpF,OAAO,UAAU,MAAM,kCAAkC,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AAChI,OAAO,EAAE,mBAAmB,EAAE,MAAM,wDAAwD,CAAC;AAC7F,OAAO,EAAE,WAAW,EAAE,MAAM,gDAAgD,CAAC;AAC7E,OAAO,EAAE,wBAAwB,EAAE,6BAA6B,EAAE,MAAM,+BAA+B,CAAC;AACxG,OAAO,EAAE,sBAAsB,EAAE,MAAM,sCAAsC,CAAC;AAC9E,OAAO,EAAE,qCAAqC,EAAE,MAAM,+CAA+C,CAAC;AACtG,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAgClE;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,EAChC,MAAM,EACN,UAAU,EACV,oBAAoB,EACpB,gBAAgB,EAChB,eAAe,GACQ;IACvB,MAAM,sBAAsB,GAAG,OAAO,CACpC,GAAG,EAAE,CAAC,CAAC;QACL,GAAG,6BAA6B;QAChC,GAAG,eAAe;KACnB,CAAC;IACF,uDAAuD;IACvD,MAAM,CAAC,MAAM,CAAC,eAAe,IAAI,EAAE,CAAC,CACrC,CAAC;IACF,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IAEtD,MAAM,EAAE,QAAQ,EAAE,yBAAyB,EAAE,GAAG,WAAW,CAAoD;QAC7G,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,WAAW;QACX,kBAAkB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;KAC9D,CAAC,CAAC;IAEH,MAAM,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,GAAG,6BAA6B,CAAC;QACzF,UAAU;QACV,QAAQ,EAAE,yBAAyB;QACnC,eAAe,EAAE,sBAAsB;KACxC,CAAC,CAAC;IAEH,MAAM,sBAAsB,GAAG,WAAW,CACxC,CAAC,KAAK,EAAE,EAAE;QACR,OAAO,IAAI,wBAAwB,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,yBAAyB,EAAE,EAAE,eAAe,EAAE,sBAAsB,EAAE,CAAC,CAAC;IAC9I,CAAC,EACD,CAAC,QAAQ,EAAE,yBAAyB,EAAE,sBAAsB,CAAC,CAC9D,CAAC;IAEF,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,gBAAgB,CAAC;QACpD,sBAAsB;QACtB,MAAM;QACN,yBAAyB;QACzB,sBAAsB;QACtB,QAAQ;QACR,oBAAoB;KACrB,CAAC,CAAC;IAEH,OAAO;QACL,mBAAmB,EAAE;YACnB,QAAQ,EAAE,oBAAoB;YAC9B,sBAAsB;YACtB,gBAAgB,EAAE,QAAQ;YAC1B,wBAAwB;YACxB,gBAAgB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,4BAA4B,CAAC,MAAM,EAAE,cAAc,EAAE,gBAAgB,CAAC,EAAE,CAAC,MAAM,EAAE,cAAc,EAAE,gBAAgB,CAAC,CAAC;YACnJ,SAAS,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;SAC5E;QACD,aAAa,EAAE;YACb,cAAc,EAAE,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAC,kBAAkB,IAAC,IAAI,EAAE,IAAI,GAAI,EAAE,EAAE,CAAC;YAC7E,WAAW;SACZ;KACF,CAAC;AACJ,CAAC;AAED,SAAS,8BAA8B,CACrC,KAA4G;IAE5G,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,cAAc,EAAE,aAAa,EAAE,GAAG,KAAK,CAAC;IAC1E,OAAO,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAC1B,qCAAqC,CAAC;QACpC,QAAQ,EAAE,UAAU;QACpB,QAAQ,EAAE,cAAc,EAAE;QAC1B,YAAY;QACZ,aAAa;QACb,eAAe,EAAE,YAAY,CAAC,eAAe;KAC9C,CAAC,CAAC;AACP,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAe,EAAE,KAAoC,EAAE,gBAAkC;IAC7H,IAAI,KAAK,EAAE,CAAC;QACV,IAAI,KAAK,KAAK,sBAAsB,EAAE,CAAC;YACrC,OAAO,KAAC,oBAAoB,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;QAC1D,CAAC;QACD,OAAO,KAAC,kBAAkB,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;IACxD,CAAC;IACD,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,KAAC,eAAe,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;IACrD,CAAC;IACD,IAAI,gBAAgB,EAAE,CAAC;QACrB,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IACD,OAAO,KAAC,gBAAgB,IAAC,IAAI,EAAE,WAAW,GAAI,CAAC;AACjD,CAAC;AAED,YAAY;AACZ,MAAM,UAAU,kBAAkB,CAAC,EAAE,IAAI,EAAuC;IAC9E,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;QACtD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,QAAQ,IAAI,CAAC,QAAQ,CAAC,YAAa,CAAC,OAAO,EAAE,CAAC;YAC5C,KAAK,aAAa;gBAChB,OAAO,WAAW,CAAC;YACrB,KAAK,qBAAqB;gBACxB,OAAO,cAAc,CAAC;YACxB,KAAK,2BAA2B;gBAC9B,OAAO,sBAAsB,CAAC;YAChC,KAAK,WAAW;gBACd,OAAO,UAAU,CAAC;YACpB,KAAK,eAAe;gBAClB,OAAO,QAAQ,CAAC;YAClB;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,KAAC,IAAI,IAAC,IAAI,EAAE,OAAO,EAAE,GAAI,CAAC;AACnC,CAAC;AAED,SAAS,WAAW,CAAC,IAA+B;IAClD,OAAO,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,WAAW,CAAC;AACjD,CAAC;AAED,SAAS,6BAA6B,CAAC,KAItC;IACC,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC;IACxD,MAAM,EAAE,wBAAwB,EAAE,aAAa,EAAE,sBAAsB,EAAE,GAAG,mBAAmB,CAG7F;QACA,UAAU;QACV,QAAQ;QACR,YAAY,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC;QACrE,aAAa,EAAE,8BAA8B;KAC9C,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,QAAQ,EAAE,CAAC,2BAA2B,EAAE,CAAC;IAC3C,CAAC,EAAE,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC,CAAC;IAE9B,OAAO;QACL,wBAAwB;QACxB,sBAAsB;KACvB,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,KAAkD;IACrE,OAAO,IAAI,sBAAsB,CAAC,wBAAwB,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;AAC1G,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useEffect, useMemo } from \"react\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { Icon } from \"@stratakit/foundations\";\nimport categorySvg from \"@stratakit/icons/bis-category-3d.svg\";\nimport subcategorySvg from \"@stratakit/icons/bis-category-subcategory.svg\";\nimport classSvg from \"@stratakit/icons/bis-class.svg\";\nimport definitionContainerSvg from \"@stratakit/icons/bis-definitions-container.svg\";\nimport elementSvg from \"@stratakit/icons/bis-element.svg\";\nimport { EmptyTreeContent, FilterUnknownError, NoFilterMatches, TooManyFilterMatches } from \"../common/components/EmptyTree.js\";\nimport { useCachedVisibility } from \"../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport { useIdsCache } from \"../common/internal/useTreeHooks/UseIdsCache.js\";\nimport { CategoriesTreeDefinition, defaultHierarchyConfiguration } from \"./CategoriesTreeDefinition.js\";\nimport { CategoriesTreeIdsCache } from \"./internal/CategoriesTreeIdsCache.js\";\nimport { createCategoriesTreeVisibilityHandler } from \"./internal/CategoriesTreeVisibilityHandler.js\";\nimport { useFilteredPaths } from \"./internal/UseFilteredPaths.js\";\n\nimport type { CreateCacheProps } from \"../common/internal/useTreeHooks/UseIdsCache.js\";\nimport type { CreateFactoryProps } from \"../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport type { CategoriesTreeFilteringError } from \"./internal/UseFilteredPaths.js\";\nimport type { ReactNode } from \"react\";\nimport type { Id64Array } from \"@itwin/core-bentley\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { VisibilityTreeProps } from \"../common/components/VisibilityTree.js\";\nimport type { VisibilityTreeRendererProps } from \"../common/components/VisibilityTreeRenderer.js\";\nimport type { CategoryInfo } from \"../common/CategoriesVisibilityUtils.js\";\nimport type { CategoriesTreeHierarchyConfiguration } from \"./CategoriesTreeDefinition.js\";\n\n/** @beta */\nexport interface UseCategoriesTreeProps {\n  activeView: Viewport;\n  onCategoriesFiltered?: (props: { categories: CategoryInfo[] | undefined; models?: Id64Array }) => void;\n  filter?: string;\n  emptyTreeContent?: ReactNode;\n  hierarchyConfig?: Partial<CategoriesTreeHierarchyConfiguration>;\n}\n\n/** @beta */\ninterface UseCategoriesTreeResult {\n  categoriesTreeProps: Pick<\n    VisibilityTreeProps,\n    \"treeName\" | \"getHierarchyDefinition\" | \"getFilteredPaths\" | \"visibilityHandlerFactory\" | \"highlight\" | \"emptyTreeContent\"\n  >;\n  rendererProps: Required<Pick<VisibilityTreeRendererProps, \"getDecorations\" | \"getSublabel\">>;\n}\n\n/**\n * Custom hook to create and manage state for the categories tree.\n * @beta\n */\nexport function useCategoriesTree({\n  filter,\n  activeView,\n  onCategoriesFiltered,\n  emptyTreeContent,\n  hierarchyConfig,\n}: UseCategoriesTreeProps): UseCategoriesTreeResult {\n  const hierarchyConfiguration = useMemo<CategoriesTreeHierarchyConfiguration>(\n    () => ({\n      ...defaultHierarchyConfiguration,\n      ...hierarchyConfig,\n    }),\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    Object.values(hierarchyConfig ?? {}),\n  );\n  const viewType = activeView.view.is2d() ? \"2d\" : \"3d\";\n\n  const { getCache: getCategoriesTreeIdsCache } = useIdsCache<CategoriesTreeIdsCache, { viewType: \"2d\" | \"3d\" }>({\n    imodel: activeView.iModel,\n    createCache,\n    cacheSpecificProps: useMemo(() => ({ viewType }), [viewType]),\n  });\n\n  const { visibilityHandlerFactory, onFilteredPathsChanged } = useCategoriesCachedVisibility({\n    activeView,\n    getCache: getCategoriesTreeIdsCache,\n    hierarchyConfig: hierarchyConfiguration,\n  });\n\n  const getHierarchyDefinition = useCallback<VisibilityTreeProps[\"getHierarchyDefinition\"]>(\n    (props) => {\n      return new CategoriesTreeDefinition({ ...props, viewType, idsCache: getCategoriesTreeIdsCache(), hierarchyConfig: hierarchyConfiguration });\n    },\n    [viewType, getCategoriesTreeIdsCache, hierarchyConfiguration],\n  );\n\n  const { getPaths, filteringError } = useFilteredPaths({\n    hierarchyConfiguration,\n    filter,\n    getCategoriesTreeIdsCache,\n    onFilteredPathsChanged,\n    viewType,\n    onCategoriesFiltered,\n  });\n\n  return {\n    categoriesTreeProps: {\n      treeName: \"categories-tree-v2\",\n      getHierarchyDefinition,\n      getFilteredPaths: getPaths,\n      visibilityHandlerFactory,\n      emptyTreeContent: useMemo(() => getEmptyTreeContentComponent(filter, filteringError, emptyTreeContent), [filter, filteringError, emptyTreeContent]),\n      highlight: useMemo(() => (filter ? { text: filter } : undefined), [filter]),\n    },\n    rendererProps: {\n      getDecorations: useCallback((node) => <CategoriesTreeIcon node={node} />, []),\n      getSublabel,\n    },\n  };\n}\n\nfunction createVisibilityHandlerFactory(\n  props: CreateFactoryProps<CategoriesTreeIdsCache, { hierarchyConfig: CategoriesTreeHierarchyConfiguration }>,\n): VisibilityTreeProps[\"visibilityHandlerFactory\"] {\n  const { activeView, factoryProps, idsCacheGetter, filteredPaths } = props;\n  return ({ imodelAccess }) =>\n    createCategoriesTreeVisibilityHandler({\n      viewport: activeView,\n      idsCache: idsCacheGetter(),\n      imodelAccess,\n      filteredPaths,\n      hierarchyConfig: factoryProps.hierarchyConfig,\n    });\n}\n\nfunction getEmptyTreeContentComponent(filter?: string, error?: CategoriesTreeFilteringError, emptyTreeContent?: React.ReactNode) {\n  if (error) {\n    if (error === \"tooManyFilterMatches\") {\n      return <TooManyFilterMatches base={\"categoriesTree\"} />;\n    }\n    return <FilterUnknownError base={\"categoriesTree\"} />;\n  }\n  if (filter) {\n    return <NoFilterMatches base={\"categoriesTree\"} />;\n  }\n  if (emptyTreeContent) {\n    return emptyTreeContent;\n  }\n  return <EmptyTreeContent icon={categorySvg} />;\n}\n\n/** @beta */\nexport function CategoriesTreeIcon({ node }: { node: PresentationHierarchyNode }) {\n  if (node.nodeData.extendedData?.imageId === undefined) {\n    return undefined;\n  }\n\n  const getIcon = () => {\n    switch (node.nodeData.extendedData!.imageId) {\n      case \"icon-layers\":\n        return categorySvg;\n      case \"icon-layers-isolate\":\n        return subcategorySvg;\n      case \"icon-definition-container\":\n        return definitionContainerSvg;\n      case \"icon-item\":\n        return elementSvg;\n      case \"icon-ec-class\":\n        return classSvg;\n      default:\n        return undefined;\n    }\n  };\n\n  return <Icon href={getIcon()} />;\n}\n\nfunction getSublabel(node: PresentationHierarchyNode) {\n  return node.nodeData.extendedData?.description;\n}\n\nfunction useCategoriesCachedVisibility(props: {\n  activeView: Viewport;\n  getCache: () => CategoriesTreeIdsCache;\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration;\n}) {\n  const { activeView, getCache, hierarchyConfig } = props;\n  const { visibilityHandlerFactory, filteredPaths, onFilteredPathsChanged } = useCachedVisibility<\n    CategoriesTreeIdsCache,\n    { hierarchyConfig: CategoriesTreeHierarchyConfiguration }\n  >({\n    activeView,\n    getCache,\n    factoryProps: useMemo(() => ({ hierarchyConfig }), [hierarchyConfig]),\n    createFactory: createVisibilityHandlerFactory,\n  });\n\n  useEffect(() => {\n    getCache().clearFilteredElementsModels();\n  }, [filteredPaths, getCache]);\n\n  return {\n    visibilityHandlerFactory,\n    onFilteredPathsChanged,\n  };\n}\n\nfunction createCache(props: CreateCacheProps<{ viewType: \"2d\" | \"3d\" }>) {\n  return new CategoriesTreeIdsCache(createECSqlQueryExecutor(props.imodel), props.specificProps.viewType);\n}\n"]}

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

@@ -42,7 +42,7 @@ export declare class CategoriesTreeIdsCache implements Disposable {
         categories: CategoryInfo[];
         definitionContainers: Array<DefinitionContainerId>;
     }>;
-    getCategoriesElementModels(categoryIds: Id64Array, includeSubModels?: boolean): Promise<Map<CategoryId, Array<ModelId>>>;
+    getCategoriesElementModels(categoryIds: Id64Array, includeSubModels?: boolean): Promise<Map<CategoryId, Set<ModelId>>>;
     getModelCategoryIds(modelId: Id64String): Promise<Id64Array>;
     hasSubModel(elementId: Id64String): Promise<boolean>;
     getAllContainedCategories(definitionContainerIds: Id64Array): Promise<Id64Array>;

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

@@ -324,10 +324,10 @@ export class CategoriesTreeIdsCache {
                 if ((includeSubModels || !isSubModel) && categories.has(categoryId)) {
                     let categoryModels = result.get(categoryId);
                     if (!categoryModels) {
-                        categoryModels = new Array();
+                        categoryModels = new Set();
                         result.set(categoryId, categoryModels);
                     }
-                    categoryModels.push(modelId);
+                    categoryModels.add(modelId);
                 }
             }
         }

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

@@ -1 +1 @@
-{"version":3,"file":"CategoriesTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,8BAA8B,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,+CAA+C,CAAC;AACzI,OAAO,EAAE,+BAA+B,EAAE,MAAM,0DAA0D,CAAC;AAC3G,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,gCAAgC,CAAC;AA+BxF,gBAAgB;AAChB,MAAM,OAAO,sBAAsB;IAcvB;IAbF,yBAAyB,CAA2E;IACpG,qBAAqB,CAAoD;IACzE,wBAAwB,CAAmF;IAC3G,kBAAkB,CAA2D;IACpE,sBAAsB,CAAkC;IACjE,iCAAiC,CAA6D;IAC9F,cAAc,CAAS;IACvB,qBAAqB,CAAS;IAC9B,mBAAmB,CAAS;IAC5B,+BAA+B,CAA+B;IAC9D,uBAAuB,CAA+C;IAE9E,YACU,cAA0C,EAClD,QAAqB;QADb,mBAAc,GAAd,cAAc,CAA4B;QAGlD,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC/E,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,qBAAqB,GAAG,YAAY,CAAC;QAC1C,IAAI,CAAC,mBAAmB,GAAG,UAAU,CAAC;QACtC,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,cAAc,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;IACpG,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,2BAA2B,CAAC,kBAA6B;QAItE,MAAM,KAAK,GAAG;;aAEL,IAAI,CAAC,qBAAqB;+BACR,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC;KACvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACzI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC;QAC7C,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,kBAA6B;QAClE,IAAI,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpC,OAAO,IAAI,GAAG,EAAsB,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,uBAAuB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC3C,MAAM,sBAAsB,GAAG,IAAI,GAAG,EAAE,CAAC;YACzC,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,IAAI,CAAC,2BAA2B,CAAC,kBAAkB,CAAC,EAAE,CAAC;gBACzF,sBAAsB,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAC1C,CAAC;YACD,OAAO,sBAAsB,CAAC;QAChC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;IAEM,2BAA2B;QAChC,IAAI,CAAC,uBAAuB,GAAG,SAAS,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,CAAC,2BAA2B;QAIxC,MAAM,KAAK,GAAG;;aAEL,IAAI,CAAC,mBAAmB;aACxB,IAAI,CAAC,qBAAqB;;;KAGlC,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,iEAAiE,EAAE,CACzI,EAAE,CAAC;YACF,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QAC7D,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,eAAe;QAM5B,MAAM,8BAA8B,GAAG,MAAM,IAAI,CAAC,iCAAiC,EAAE,CAAC;QACtF,MAAM,eAAe,GAAG;;;;;UAMlB,8BAA8B;YAC5B,CAAC,CAAC;gEACkD,8BAA8B;;;cAGhF;YACF,CAAC,CAAC,OACN;;UAEE,IAAI,CAAC,cAAc;eACd,sBAAsB;eACtB,gBAAgB;;;;oCAIK,IAAI,CAAC,qBAAqB;;KAEzD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,eAAe,EAAE,EAC1B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,8CAA8C,EAAE,CACtH,EAAE,CAAC;YACF,MAAM;gBACJ,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,+BAA+B,EAAE,GAAG,CAAC,+BAA+B;gBACpE,qBAAqB,EAAE,GAAG,CAAC,qBAAqB;aACjD,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,oCAAoC;QAChD,MAAM,KAAK,GAAG;;;;;;;;;KASb,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC5D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,YAAY,EAAE,qEAAqE,EAAE,CACxF,EAAE,CAAC;YACF,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB;QACtC,iKAAiK;QACjK,2EAA2E;QAC3E,2GAA2G;QAC3G,kFAAkF;QAClF,MAAM,yBAAyB,GAAG,sBAAsB,CAAC;QACzD,MAAM,IAAI,GAAG;YACX;UACI,yBAAyB;;;;;cAKrB,8BAA8B;;yDAEa,IAAI,CAAC,cAAc,sDAAsD,IAAI,CAAC,qBAAqB;;;;;;;;;cAS9I,yBAAyB;mBACpB,8BAA8B;;;;OAI1C;SACF,CAAC;QACF,MAAM,gBAAgB,GAAG;2DAC8B,yBAAyB;KAC/E,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,EACjC,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,yDAAyD,EAAE,CACjI,EAAE,CAAC;YACF,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;QAC7C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB,CAAC,WAAsB;QAC7D,MAAM,gBAAgB,GAAG;;;;;UAKnB,sBAAsB;;;+BAGD,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC;KAC/C,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAC3B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,kDAAkD,EAAE,CAC1H,EAAE,CAAC;YACF,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QACjD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,uBAAuB;QACnC,IAAI,CAAC,qBAAqB,KAAK,CAAC,KAAK,IAAI,EAAE;YACzC,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA2B,CAAC;YAC/D,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,eAAe,EAAE,EAAE,CAAC;gBAC3D,IAAI,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBACvE,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;oBAClC,eAAe,GAAG,EAAE,+BAA+B,EAAE,eAAe,CAAC,+BAA+B,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;oBAC5H,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;gBACpE,CAAC;gBACD,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,eAAe,CAAC,EAAE,EAAE,qBAAqB,EAAE,eAAe,CAAC,qBAAqB,EAAE,CAAC,CAAC;YACjI,CAAC;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEO,KAAK,CAAC,0BAA0B;QACtC,IAAI,CAAC,wBAAwB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC5C,MAAM,CAAC,eAAe,EAAE,gCAAgC,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC5E,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAAqC,CAAC;oBAC7E,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,2BAA2B,EAAE,EAAE,CAAC;wBACvE,IAAI,UAAU,GAAG,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;wBACtE,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;4BAC7B,UAAU,GAAG,EAAE,WAAW,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;4BACxC,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;wBACnE,CAAC;wBACD,UAAU,CAAC,WAAW,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;oBACzD,CAAC;oBACD,OAAO,uBAAuB,CAAC;gBACjC,CAAC,CAAC,EAAE;gBACJ,IAAI,CAAC,mCAAmC,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkE,CAAC;YACzF,MAAM,SAAS,GAAG,oBAAoB,CAAC,gCAAgC,CAAC,CAAC;YACzE,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,eAAe,EAAE,CAAC;gBACpD,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC1C,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,UAAU,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC;YAC3E,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,wBAAwB,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QAMjC,MAAM,KAAK,GAAG;;;;;aAKL,IAAI,CAAC,mBAAmB;aACxB,IAAI,CAAC,qBAAqB;;;sDAGe,IAAI,CAAC,qBAAqB;KAC3E,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,oDAAoD,EAAE,CAC5H,EAAE,CAAC;YACF,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,cAAc,EAAE,GAAG,CAAC,cAAc,EAAE,CAAC;QACzI,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mCAAmC;QAC/C,IAAI,CAAC,iCAAiC,KAAK,CAAC,KAAK,IAAI,EAAE;YACrD,MAAM,gCAAgC,GAAG,IAAI,GAAG,EAAoC,CAAC;YACrF,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBAC1F,MAAM,GAAG,GAAqB,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;gBACzD,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACxD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,gCAAgC,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;gBACzE,CAAC;qBAAM,CAAC;oBACN,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,OAAO,gCAAgC,CAAC;QAC1C,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iCAAiC,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,4BAA4B,CAAC,OAAmB,EAAE,WAAsB;QACnF,MAAM,gCAAgC,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;QAC1F,MAAM,MAAM,GAAG,IAAI,KAAK,EAAa,CAAC;QACtC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;YAC/E,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,oBAAoB;QAChC,IAAI,CAAC,kBAAkB,KAAK,CAAC,KAAK,IAAI,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAkC,CAAC;YACnE,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClE,MAAM,oCAAoC,GAAG,IAAI,KAAK,EAAc,CAAC;YACrE,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE,CAAC;gBAChE,oCAAoC,CAAC,IAAI,CACvC,GAAG,mBAAmB,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,qBAAqB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC,CAC/I,CAAC;YACJ,CAAC;YAED,IAAI,oCAAoC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtD,OAAO,gBAAgB,CAAC;YAC1B,CAAC;YAED,IAAI,KAAK,EAAE,MAAM,kBAAkB,IAAI,IAAI,CAAC,yBAAyB,CAAC,oCAAoC,CAAC,EAAE,CAAC;gBAC5G,gBAAgB,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,kBAAkB,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC3F,CAAC;YACD,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,2BAA2B;QACvC,IAAI,CAAC,yBAAyB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC7C,MAAM,wBAAwB,GAAG,IAAI,GAAG,EAAkD,CAAC;YAC3F,MAAM,CAAC,8BAA8B,EAAE,oBAAoB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC/E,IAAI,CAAC,iCAAiC,EAAE;gBACxC,IAAI,CAAC,uBAAuB,EAAE;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,8BAA8B,IAAI,oBAAoB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACvE,OAAO,wBAAwB,CAAC;YAClC,CAAC;YAED,IAAI,KAAK,EAAE,MAAM,0BAA0B,IAAI,IAAI,CAAC,yBAAyB,EAAE,EAAE,CAAC;gBAChF,MAAM,mBAAmB,GAAG,oBAAoB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC;gBAEpF,wBAAwB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,EAAE;oBAC1D,eAAe,EAAE,mBAAmB,EAAE,eAAe,IAAI,EAAE;oBAC3D,OAAO,EAAE,0BAA0B,CAAC,OAAO;oBAC3C,2BAA2B,EAAE,EAAE;oBAC/B,+BAA+B,EAAE,KAAK;iBACvC,CAAC,CAAC;YACL,CAAC;YAED,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,wBAAwB,EAAE,CAAC;gBACxF,MAAM,yBAAyB,GAAG,wBAAwB,CAAC,GAAG,CAAC,uBAAuB,CAAC,OAAO,CAAC,CAAC;gBAChG,IAAI,yBAAyB,KAAK,SAAS,EAAE,CAAC;oBAC5C,yBAAyB,CAAC,2BAA2B,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;oBAClF,uBAAuB,CAAC,+BAA+B,GAAG,IAAI,CAAC;gBACjE,CAAC;YACH,CAAC;YAED,OAAO,wBAAwB,CAAC;QAClC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,yBAAyB,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,+CAA+C,CAC1D,4BAAuC;QAEvC,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAE1E,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAyB,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QAEnH,4BAA4B,CAAC,OAAO,CAAC,CAAC,2BAA2B,EAAE,EAAE;YACnE,MAAM,6BAA6B,GAAG,wBAAwB,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;YAChG,IAAI,6BAA6B,KAAK,SAAS,EAAE,CAAC;gBAChD,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,2BAA2B,CAAC,CAAC;gBAC/F,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,eAAe,CAAC,CAAC;YAC3E,CAAC;QACH,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,0BAA0B,CAAC,WAAsB,EAAE,gBAA0B;QACxF,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,MAAM,MAAM,GAAG,IAAI,GAAG,EAA8B,CAAC;QACrD,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,IAAI,uBAAuB,EAAE,CAAC;gBACzF,IAAI,CAAC,gBAAgB,IAAI,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;oBACpE,IAAI,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC5C,IAAI,CAAC,cAAc,EAAE,CAAC;wBACpB,cAAc,GAAG,IAAI,KAAK,EAAW,CAAC;wBACtC,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;oBACzC,CAAC;oBACD,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAAC,OAAmB;QAClD,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC;IACxE,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,SAAqB;QAC5C,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,OAAO,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,sBAAiC;QACtE,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QAEvC,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC1E,MAAM,kBAAkB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC1C,sBAAsB,CAAC,GAAG,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE;YACzD,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;YACpF,IAAI,uBAAuB,KAAK,SAAS,EAAE,CAAC;gBAC1C,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,uBAAuB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;YACvF,OAAO,IAAI,CAAC,yBAAyB,CAAC,uBAAuB,CAAC,2BAA2B,CAAC,CAAC;QAC7F,CAAC,CAAC,CACH,CAAC;QACF,KAAK,MAAM,UAAU,IAAI,kBAAkB,EAAE,CAAC;YAC5C,MAAM,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAC9B,KAAyG;QAEzG,IAAI,eAAe,IAAI,KAAK,EAAE,CAAC;YAC7B,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5D,MAAM,eAAe,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACnE,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;gBAClC,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,sBAAsB,EAAE,CAAC,CAAC;QAC3J,CAAC;QAED,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;YAC1B,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClE,KAAK,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC,IAAI,oBAAoB,EAAE,CAAC;gBAClE,IAAI,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;oBACvG,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE,CAAC;wBACzD,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;oBACpE,CAAC;oBAED,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;gBAC7I,CAAC;YACH,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC1E,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC1F,IAAI,uBAAuB,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE,CAAC;YAC7D,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,8BAA8B,EAAE,CAAC,CAAC;QAC1F,CAAC;QAED,OAAO;YACL,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,uBAAuB,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/F,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,8BAA8B,EAAE;SAC/E,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,KAAK,CAAC,uCAAuC;QAClD,MAAM,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,CAAC,GAAG,wBAAwB,CAAC,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,KAAK,EAAc,EAAE,CAAC;QACnH,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE,CAAC;YAChE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC;QAC1G,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,wCAAwC;QACnD,MAAM,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QACxG,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE,CAAC;YAChE,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE,CAAC;gBACzD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QAED,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,wBAAwB,EAAE,CAAC;YACxF,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE,CAAC;gBAC7D,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,WAAsB;QAClD,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAoC,CAAC;QAC3D,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,KAAK,MAAM,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,iBAAiB,EAAE,CAAC;gBACjE,IAAI,eAAe,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;oBAC9C,IAAI,aAAa,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC3C,IAAI,CAAC,aAAa,EAAE,CAAC;wBACnB,aAAa,GAAG,EAAE,CAAC;wBACnB,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;oBACxC,CAAC;oBACD,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBACpC,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,iCAAiC;QAC5C,IAAI,CAAC,+BAA+B,KAAK,IAAI,CAAC,oCAAoC,EAAE,CAAC;QACrF,OAAO,IAAI,CAAC,+BAA+B,CAAC;IAC9C,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { CLASS_NAME_DefinitionContainer, CLASS_NAME_Model, CLASS_NAME_SubCategory } from \"../../common/internal/ClassNameDefinitions.js\";\nimport { ModelCategoryElementsCountCache } from \"../../common/internal/ModelCategoryElementsCountCache.js\";\nimport { getClassesByView, getDistinctMapValues } from \"../../common/internal/Utils.js\";\n\nimport type { CategoryId, DefinitionContainerId, ElementId, ModelId, SubCategoryId } from \"../../common/internal/Types.js\";\nimport type { Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\n\ninterface DefinitionContainerInfo {\n  modelId: Id64String;\n  parentDefinitionContainerExists: boolean;\n  childCategories: CategoryInfo[];\n  childDefinitionContainerIds: Id64Array;\n}\n\ninterface CategoriesInfo {\n  childCategories: CategoryInfo[];\n  parentDefinitionContainerExists: boolean;\n}\n\n/** @internal */\nexport interface CategoryInfo {\n  id: CategoryId;\n  subCategoryChildCount: number;\n}\n\ninterface SubCategoryInfo {\n  categoryId: Id64String;\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\n/** @internal */\nexport class CategoriesTreeIdsCache implements Disposable {\n  private _definitionContainersInfo: Promise<Map<DefinitionContainerId, DefinitionContainerInfo>> | undefined;\n  private _modelsCategoriesInfo: Promise<Map<ModelId, CategoriesInfo>> | undefined;\n  private _elementModelsCategories: Promise<Map<ModelId, { categoryIds: Id64Set; isSubModel: boolean }>> | undefined;\n  private _subCategoriesInfo: Promise<Map<SubCategoryId, SubCategoryInfo>> | undefined;\n  private readonly _categoryElementCounts: ModelCategoryElementsCountCache;\n  private _modelWithCategoryModeledElements: Promise<Map<ModelCategoryKey, Set<ElementId>>> | undefined;\n  private _categoryClass: string;\n  private _categoryElementClass: string;\n  private _categoryModelClass: string;\n  private _isDefinitionContainerSupported: Promise<boolean> | undefined;\n  private _filteredElementsModels: Promise<Map<ElementId, ModelId>> | undefined;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    viewType: \"3d\" | \"2d\",\n  ) {\n    const { categoryClass, elementClass, modelClass } = getClassesByView(viewType);\n    this._categoryClass = categoryClass;\n    this._categoryElementClass = elementClass;\n    this._categoryModelClass = modelClass;\n    this._categoryElementCounts = new ModelCategoryElementsCountCache(_queryExecutor, [elementClass]);\n  }\n\n  public [Symbol.dispose]() {\n    this._categoryElementCounts[Symbol.dispose]();\n  }\n\n  private async *queryFilteredElementsModels(filteredElementIds: Id64Array): AsyncIterableIterator<{\n    modelId: Id64String;\n    id: ElementId;\n  }> {\n    const query = `\n      SELECT Model.Id modelId, ECInstanceId id\n      FROM ${this._categoryElementClass}\n      WHERE ECInstanceId IN (${filteredElementIds.join(\", \")})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, id: row.id };\n    }\n  }\n\n  public async getFilteredElementsModels(filteredElementIds: Id64Array) {\n    if (filteredElementIds.length === 0) {\n      return new Map<ElementId, ModelId>();\n    }\n\n    this._filteredElementsModels ??= (async () => {\n      const filteredElementsModels = new Map();\n      for await (const { modelId, id } of this.queryFilteredElementsModels(filteredElementIds)) {\n        filteredElementsModels.set(id, modelId);\n      }\n      return filteredElementsModels;\n    })();\n    return this._filteredElementsModels;\n  }\n\n  public clearFilteredElementsModels() {\n    this._filteredElementsModels = undefined;\n  }\n\n  private async *queryElementModelCategories(): AsyncIterableIterator<{\n    modelId: Id64String;\n    categoryId: Id64String;\n  }> {\n    const query = `\n      SELECT this.Model.Id modelId, this.Category.Id categoryId\n      FROM ${this._categoryModelClass} m\n      JOIN ${this._categoryElementClass} this ON m.ECInstanceId = this.Model.Id\n      WHERE this.Parent.Id IS NULL AND m.IsPrivate = false\n      GROUP BY modelId, categoryId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/element-models-and-categories-query\" },\n    )) {\n      yield { modelId: row.modelId, categoryId: row.categoryId };\n    }\n  }\n\n  private async *queryCategories(): AsyncIterableIterator<{\n    id: CategoryId;\n    modelId: Id64String;\n    parentDefinitionContainerExists: boolean;\n    subCategoryChildCount: number;\n  }> {\n    const isDefinitionContainerSupported = await this.getIsDefinitionContainerSupported();\n    const categoriesQuery = `\n      SELECT\n        this.ECInstanceId id,\n        COUNT(sc.ECInstanceId) subCategoryChildCount,\n        this.Model.Id modelId,\n        ${\n          isDefinitionContainerSupported\n            ? `\n            IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${CLASS_NAME_DefinitionContainer} dc),\n              true,\n              false\n            )`\n            : \"false\"\n        } parentDefinitionContainerExists\n      FROM\n        ${this._categoryClass} this\n        JOIN ${CLASS_NAME_SubCategory} sc ON sc.Parent.Id = this.ECInstanceId\n        JOIN ${CLASS_NAME_Model} m ON m.ECInstanceId = this.Model.Id\n      WHERE\n        NOT this.IsPrivate\n        AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))\n        AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)\n      GROUP BY this.ECInstanceId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: categoriesQuery },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/categories-query\" },\n    )) {\n      yield {\n        id: row.id,\n        modelId: row.modelId,\n        parentDefinitionContainerExists: row.parentDefinitionContainerExists,\n        subCategoryChildCount: row.subCategoryChildCount,\n      };\n    }\n  }\n\n  private async queryIsDefinitionContainersSupported(): Promise<boolean> {\n    const query = `\n      SELECT\n        1\n      FROM\n        ECDbMeta.ECSchemaDef s\n        JOIN ECDbMeta.ECClassDef c ON c.Schema.Id = s.ECInstanceId\n      WHERE\n        s.Name = 'BisCore'\n        AND c.Name = 'DefinitionContainer'\n    `;\n\n    for await (const _row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { restartToken: \"tree-widget/categories-tree/is-definition-container-supported-query\" },\n    )) {\n      return true;\n    }\n    return false;\n  }\n\n  private async *queryDefinitionContainers(): AsyncIterableIterator<{ id: DefinitionContainerId; modelId: Id64String }> {\n    // DefinitionModel ECInstanceId will always be the same as modeled DefinitionContainer ECInstanceId, if this wasn't the case, we would need to do something like:\n    //  JOIN BisCore.DefinitionModel dm ON dm.ECInstanceId = ${modelIdAccessor}\n    //  JOIN BisCore.DefinitionModelBreaksDownDefinitionContainer dr ON dr.SourceECInstanceId = dm.ECInstanceId\n    //  JOIN BisCore.DefinitionContainer dc ON dc.ECInstanceId = dr.TargetECInstanceId\n    const DEFINITION_CONTAINERS_CTE = \"DefinitionContainers\";\n    const ctes = [\n      `\n        ${DEFINITION_CONTAINERS_CTE}(ECInstanceId, ModelId) AS (\n          SELECT\n            dc.ECInstanceId,\n            dc.Model.Id\n          FROM\n            ${CLASS_NAME_DefinitionContainer} dc\n          WHERE\n            dc.ECInstanceId IN (SELECT c.Model.Id FROM ${this._categoryClass} c WHERE NOT c.IsPrivate AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = c.ECInstanceId))\n            AND NOT dc.IsPrivate\n\n          UNION ALL\n\n          SELECT\n            pdc.ECInstanceId,\n            pdc.Model.Id\n          FROM\n            ${DEFINITION_CONTAINERS_CTE} cdc\n            JOIN ${CLASS_NAME_DefinitionContainer} pdc ON pdc.ECInstanceId = cdc.ModelId\n          WHERE\n            NOT pdc.IsPrivate\n        )\n      `,\n    ];\n    const definitionsQuery = `\n      SELECT dc.ECInstanceId id, dc.ModelId modelId FROM ${DEFINITION_CONTAINERS_CTE} dc GROUP BY dc.ECInstanceId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ctes, ecsql: definitionsQuery },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/definition-containers-query\" },\n    )) {\n      yield { id: row.id, modelId: row.modelId };\n    }\n  }\n\n  private async *queryVisibleSubCategories(categoryIds: Id64Array): AsyncIterableIterator<{ id: SubCategoryId; parentId: CategoryId }> {\n    const definitionsQuery = `\n      SELECT\n        sc.ECInstanceId id,\n        sc.Parent.Id categoryId\n      FROM\n        ${CLASS_NAME_SubCategory} sc\n      WHERE\n        NOT sc.IsPrivate\n        AND sc.Parent.Id IN (${categoryIds.join(\",\")})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: definitionsQuery },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/sub-categories-query\" },\n    )) {\n      yield { id: row.id, parentId: row.categoryId };\n    }\n  }\n\n  private async getModelsCategoriesInfo() {\n    this._modelsCategoriesInfo ??= (async () => {\n      const allModelsCategories = new Map<ModelId, CategoriesInfo>();\n      for await (const queriedCategory of this.queryCategories()) {\n        let modelCategories = allModelsCategories.get(queriedCategory.modelId);\n        if (modelCategories === undefined) {\n          modelCategories = { parentDefinitionContainerExists: queriedCategory.parentDefinitionContainerExists, childCategories: [] };\n          allModelsCategories.set(queriedCategory.modelId, modelCategories);\n        }\n        modelCategories.childCategories.push({ id: queriedCategory.id, subCategoryChildCount: queriedCategory.subCategoryChildCount });\n      }\n      return allModelsCategories;\n    })();\n    return this._modelsCategoriesInfo;\n  }\n\n  private async getElementModelsCategories() {\n    this._elementModelsCategories ??= (async () => {\n      const [modelCategories, modelWithCategoryModeledElements] = await Promise.all([\n        (async () => {\n          const elementModelsCategories = new Map<ModelId, { categoryIds: Id64Set }>();\n          for await (const queriedCategory of this.queryElementModelCategories()) {\n            let modelEntry = elementModelsCategories.get(queriedCategory.modelId);\n            if (modelEntry === undefined) {\n              modelEntry = { categoryIds: new Set() };\n              elementModelsCategories.set(queriedCategory.modelId, modelEntry);\n            }\n            modelEntry.categoryIds.add(queriedCategory.categoryId);\n          }\n          return elementModelsCategories;\n        })(),\n        this.getModelWithCategoryModeledElements(),\n      ]);\n      const result = new Map<ModelId, { categoryIds: Set<CategoryId>; isSubModel: boolean }>();\n      const subModels = getDistinctMapValues(modelWithCategoryModeledElements);\n      for (const [modelId, modelEntry] of modelCategories) {\n        const isSubModel = subModels.has(modelId);\n        result.set(modelId, { categoryIds: modelEntry.categoryIds, isSubModel });\n      }\n      return result;\n    })();\n    return this._elementModelsCategories;\n  }\n\n  private async *queryModeledElements(): AsyncIterableIterator<{\n    modelId: Id64String;\n    modeledElementId: Id64String;\n    categoryId: Id64String;\n    rootCategoryId: Id64String;\n  }> {\n    const query = `\n      SELECT\n        pe.ECInstanceId modeledElementId,\n        pe.Category.Id categoryId,\n        pe.Model.Id modelId\n      FROM ${this._categoryModelClass} m\n      JOIN ${this._categoryElementClass} pe ON pe.ECInstanceId = m.ModeledElement.Id\n      WHERE\n        m.IsPrivate = false\n        AND m.ECInstanceId IN (SELECT Model.Id FROM ${this._categoryElementClass})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/modeled-elements-query\" },\n    )) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };\n    }\n  }\n\n  private async getModelWithCategoryModeledElements() {\n    this._modelWithCategoryModeledElements ??= (async () => {\n      const modelWithCategoryModeledElements = new Map<ModelCategoryKey, Set<ElementId>>();\n      for await (const { modelId, categoryId, modeledElementId } of this.queryModeledElements()) {\n        const key: ModelCategoryKey = `${modelId}-${categoryId}`;\n        const entry = modelWithCategoryModeledElements.get(key);\n        if (entry === undefined) {\n          modelWithCategoryModeledElements.set(key, new Set([modeledElementId]));\n        } else {\n          entry.add(modeledElementId);\n        }\n      }\n      return modelWithCategoryModeledElements;\n    })();\n    return this._modelWithCategoryModeledElements;\n  }\n\n  public async getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Array): Promise<Id64Array> {\n    const modelWithCategoryModeledElements = await this.getModelWithCategoryModeledElements();\n    const result = new Array<ElementId>();\n    for (const categoryId of categoryIds) {\n      const entry = modelWithCategoryModeledElements.get(`${modelId}-${categoryId}`);\n      if (entry !== undefined) {\n        result.push(...entry);\n      }\n    }\n    return result;\n  }\n\n  private async getSubCategoriesInfo() {\n    this._subCategoriesInfo ??= (async () => {\n      const allSubCategories = new Map<SubCategoryId, SubCategoryInfo>();\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      const categoriesWithMoreThanOneSubCategory = new Array<CategoryId>();\n      for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n        categoriesWithMoreThanOneSubCategory.push(\n          ...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.subCategoryChildCount > 1).map((categoryInfo) => categoryInfo.id),\n        );\n      }\n\n      if (categoriesWithMoreThanOneSubCategory.length === 0) {\n        return allSubCategories;\n      }\n\n      for await (const queriedSubCategory of this.queryVisibleSubCategories(categoriesWithMoreThanOneSubCategory)) {\n        allSubCategories.set(queriedSubCategory.id, { categoryId: queriedSubCategory.parentId });\n      }\n      return allSubCategories;\n    })();\n    return this._subCategoriesInfo;\n  }\n\n  private async getDefinitionContainersInfo() {\n    this._definitionContainersInfo ??= (async () => {\n      const definitionContainersInfo = new Map<DefinitionContainerId, DefinitionContainerInfo>();\n      const [isDefinitionContainerSupported, modelsCategoriesInfo] = await Promise.all([\n        this.getIsDefinitionContainerSupported(),\n        this.getModelsCategoriesInfo(),\n      ]);\n      if (!isDefinitionContainerSupported || modelsCategoriesInfo.size === 0) {\n        return definitionContainersInfo;\n      }\n\n      for await (const queriedDefinitionContainer of this.queryDefinitionContainers()) {\n        const modelCategoriesInfo = modelsCategoriesInfo.get(queriedDefinitionContainer.id);\n\n        definitionContainersInfo.set(queriedDefinitionContainer.id, {\n          childCategories: modelCategoriesInfo?.childCategories ?? [],\n          modelId: queriedDefinitionContainer.modelId,\n          childDefinitionContainerIds: [],\n          parentDefinitionContainerExists: false,\n        });\n      }\n\n      for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {\n        const parentDefinitionContainer = definitionContainersInfo.get(definitionContainerInfo.modelId);\n        if (parentDefinitionContainer !== undefined) {\n          parentDefinitionContainer.childDefinitionContainerIds.push(definitionContainerId);\n          definitionContainerInfo.parentDefinitionContainerExists = true;\n        }\n      }\n\n      return definitionContainersInfo;\n    })();\n    return this._definitionContainersInfo;\n  }\n\n  public async getDirectChildDefinitionContainersAndCategories(\n    parentDefinitionContainerIds: Id64Array,\n  ): Promise<{ categories: CategoryInfo[]; definitionContainers: Array<DefinitionContainerId> }> {\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n\n    const result = { definitionContainers: new Array<DefinitionContainerId>(), categories: new Array<CategoryInfo>() };\n\n    parentDefinitionContainerIds.forEach((parentDefinitionContainerId) => {\n      const parentDefinitionContainerInfo = definitionContainersInfo.get(parentDefinitionContainerId);\n      if (parentDefinitionContainerInfo !== undefined) {\n        result.definitionContainers.push(...parentDefinitionContainerInfo.childDefinitionContainerIds);\n        result.categories.push(...parentDefinitionContainerInfo.childCategories);\n      }\n    });\n    return result;\n  }\n\n  public async getCategoriesElementModels(categoryIds: Id64Array, includeSubModels?: boolean): Promise<Map<CategoryId, Array<ModelId>>> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    const result = new Map<CategoryId, Array<ModelId>>();\n    for (const categoryId of categoryIds) {\n      for (const [modelId, { categoryIds: categories, isSubModel }] of elementModelsCategories) {\n        if ((includeSubModels || !isSubModel) && categories.has(categoryId)) {\n          let categoryModels = result.get(categoryId);\n          if (!categoryModels) {\n            categoryModels = new Array<ModelId>();\n            result.set(categoryId, categoryModels);\n          }\n          categoryModels.push(modelId);\n        }\n      }\n    }\n    return result;\n  }\n\n  public async getModelCategoryIds(modelId: Id64String): Promise<Id64Array> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    return [...(elementModelsCategories.get(modelId)?.categoryIds ?? [])];\n  }\n\n  public async hasSubModel(elementId: Id64String): Promise<boolean> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    return elementModelsCategories.has(elementId);\n  }\n\n  public async getAllContainedCategories(definitionContainerIds: Id64Array): Promise<Id64Array> {\n    const result = new Array<CategoryId>();\n\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n    const indirectCategories = await Promise.all(\n      definitionContainerIds.map(async (definitionContainerId) => {\n        const definitionContainerInfo = definitionContainersInfo.get(definitionContainerId);\n        if (definitionContainerInfo === undefined) {\n          return [];\n        }\n        result.push(...definitionContainerInfo.childCategories.map((category) => category.id));\n        return this.getAllContainedCategories(definitionContainerInfo.childDefinitionContainerIds);\n      }),\n    );\n    for (const categories of indirectCategories) {\n      result.push(...categories);\n    }\n\n    return result;\n  }\n\n  public async getInstanceKeyPaths(\n    props: { categoryId: Id64String } | { definitionContainerId: Id64String } | { subCategoryId: Id64String },\n  ): Promise<InstanceKey[]> {\n    if (\"subCategoryId\" in props) {\n      const subCategoriesInfo = await this.getSubCategoriesInfo();\n      const subCategoryInfo = subCategoriesInfo.get(props.subCategoryId);\n      if (subCategoryInfo === undefined) {\n        return [];\n      }\n      return [...(await this.getInstanceKeyPaths({ categoryId: subCategoryInfo.categoryId })), { id: props.subCategoryId, className: CLASS_NAME_SubCategory }];\n    }\n\n    if (\"categoryId\" in props) {\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      for (const [modelId, modelCategoriesInfo] of modelsCategoriesInfo) {\n        if (modelCategoriesInfo.childCategories.find((childCategory) => childCategory.id === props.categoryId)) {\n          if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n            return [{ id: props.categoryId, className: this._categoryClass }];\n          }\n\n          return [...(await this.getInstanceKeyPaths({ definitionContainerId: modelId })), { id: props.categoryId, className: this._categoryClass }];\n        }\n      }\n      return [];\n    }\n\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n    const definitionContainerInfo = definitionContainersInfo.get(props.definitionContainerId);\n    if (definitionContainerInfo === undefined) {\n      return [];\n    }\n\n    if (!definitionContainerInfo.parentDefinitionContainerExists) {\n      return [{ id: props.definitionContainerId, className: CLASS_NAME_DefinitionContainer }];\n    }\n\n    return [\n      ...(await this.getInstanceKeyPaths({ definitionContainerId: definitionContainerInfo.modelId })),\n      { id: props.definitionContainerId, className: CLASS_NAME_DefinitionContainer },\n    ];\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  public async getAllDefinitionContainersAndCategories(): Promise<{ categories: Array<CategoryId>; definitionContainers: Array<DefinitionContainerId> }> {\n    const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);\n    const result = { definitionContainers: [...definitionContainersInfo.keys()], categories: new Array<Id64String>() };\n    for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n      result.categories.push(...modelCategoriesInfo.childCategories.map((childCategory) => childCategory.id));\n    }\n\n    return result;\n  }\n\n  public async getRootDefinitionContainersAndCategories(): Promise<{ categories: CategoryInfo[]; definitionContainers: Array<DefinitionContainerId> }> {\n    const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);\n    const result = { definitionContainers: new Array<Id64String>(), categories: new Array<CategoryInfo>() };\n    for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n      if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n        result.categories.push(...modelCategoriesInfo.childCategories);\n      }\n    }\n\n    for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {\n      if (!definitionContainerInfo.parentDefinitionContainerExists) {\n        result.definitionContainers.push(definitionContainerId);\n      }\n    }\n    return result;\n  }\n\n  public async getSubCategories(categoryIds: Id64Array): Promise<Map<CategoryId, Array<SubCategoryId>>> {\n    const subCategoriesInfo = await this.getSubCategoriesInfo();\n    const result = new Map<CategoryId, Array<SubCategoryId>>();\n    for (const categoryId of categoryIds) {\n      for (const [subCategoryId, subCategoryInfo] of subCategoriesInfo) {\n        if (subCategoryInfo.categoryId === categoryId) {\n          let categoryEntry = result.get(categoryId);\n          if (!categoryEntry) {\n            categoryEntry = [];\n            result.set(categoryId, categoryEntry);\n          }\n          categoryEntry.push(subCategoryId);\n        }\n      }\n    }\n\n    return result;\n  }\n\n  public async getIsDefinitionContainerSupported(): Promise<boolean> {\n    this._isDefinitionContainerSupported ??= this.queryIsDefinitionContainersSupported();\n    return this._isDefinitionContainerSupported;\n  }\n}\n"]}
+{"version":3,"file":"CategoriesTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,8BAA8B,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,+CAA+C,CAAC;AACzI,OAAO,EAAE,+BAA+B,EAAE,MAAM,0DAA0D,CAAC;AAC3G,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,gCAAgC,CAAC;AA+BxF,gBAAgB;AAChB,MAAM,OAAO,sBAAsB;IAcvB;IAbF,yBAAyB,CAA2E;IACpG,qBAAqB,CAAoD;IACzE,wBAAwB,CAAmF;IAC3G,kBAAkB,CAA2D;IACpE,sBAAsB,CAAkC;IACjE,iCAAiC,CAA6D;IAC9F,cAAc,CAAS;IACvB,qBAAqB,CAAS;IAC9B,mBAAmB,CAAS;IAC5B,+BAA+B,CAA+B;IAC9D,uBAAuB,CAA+C;IAE9E,YACU,cAA0C,EAClD,QAAqB;QADb,mBAAc,GAAd,cAAc,CAA4B;QAGlD,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC/E,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,qBAAqB,GAAG,YAAY,CAAC;QAC1C,IAAI,CAAC,mBAAmB,GAAG,UAAU,CAAC;QACtC,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,cAAc,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;IACpG,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,2BAA2B,CAAC,kBAA6B;QAItE,MAAM,KAAK,GAAG;;aAEL,IAAI,CAAC,qBAAqB;+BACR,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC;KACvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACzI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC;QAC7C,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,kBAA6B;QAClE,IAAI,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpC,OAAO,IAAI,GAAG,EAAsB,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,uBAAuB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC3C,MAAM,sBAAsB,GAAG,IAAI,GAAG,EAAE,CAAC;YACzC,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,IAAI,CAAC,2BAA2B,CAAC,kBAAkB,CAAC,EAAE,CAAC;gBACzF,sBAAsB,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAC1C,CAAC;YACD,OAAO,sBAAsB,CAAC;QAChC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;IAEM,2BAA2B;QAChC,IAAI,CAAC,uBAAuB,GAAG,SAAS,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,CAAC,2BAA2B;QAIxC,MAAM,KAAK,GAAG;;aAEL,IAAI,CAAC,mBAAmB;aACxB,IAAI,CAAC,qBAAqB;;;KAGlC,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,iEAAiE,EAAE,CACzI,EAAE,CAAC;YACF,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QAC7D,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,eAAe;QAM5B,MAAM,8BAA8B,GAAG,MAAM,IAAI,CAAC,iCAAiC,EAAE,CAAC;QACtF,MAAM,eAAe,GAAG;;;;;UAMlB,8BAA8B;YAC5B,CAAC,CAAC;gEACkD,8BAA8B;;;cAGhF;YACF,CAAC,CAAC,OACN;;UAEE,IAAI,CAAC,cAAc;eACd,sBAAsB;eACtB,gBAAgB;;;;oCAIK,IAAI,CAAC,qBAAqB;;KAEzD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,eAAe,EAAE,EAC1B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,8CAA8C,EAAE,CACtH,EAAE,CAAC;YACF,MAAM;gBACJ,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,+BAA+B,EAAE,GAAG,CAAC,+BAA+B;gBACpE,qBAAqB,EAAE,GAAG,CAAC,qBAAqB;aACjD,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,oCAAoC;QAChD,MAAM,KAAK,GAAG;;;;;;;;;KASb,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC5D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,YAAY,EAAE,qEAAqE,EAAE,CACxF,EAAE,CAAC;YACF,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB;QACtC,iKAAiK;QACjK,2EAA2E;QAC3E,2GAA2G;QAC3G,kFAAkF;QAClF,MAAM,yBAAyB,GAAG,sBAAsB,CAAC;QACzD,MAAM,IAAI,GAAG;YACX;UACI,yBAAyB;;;;;cAKrB,8BAA8B;;yDAEa,IAAI,CAAC,cAAc,sDAAsD,IAAI,CAAC,qBAAqB;;;;;;;;;cAS9I,yBAAyB;mBACpB,8BAA8B;;;;OAI1C;SACF,CAAC;QACF,MAAM,gBAAgB,GAAG;2DAC8B,yBAAyB;KAC/E,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,EACjC,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,yDAAyD,EAAE,CACjI,EAAE,CAAC;YACF,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;QAC7C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB,CAAC,WAAsB;QAC7D,MAAM,gBAAgB,GAAG;;;;;UAKnB,sBAAsB;;;+BAGD,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC;KAC/C,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAC3B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,kDAAkD,EAAE,CAC1H,EAAE,CAAC;YACF,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QACjD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,uBAAuB;QACnC,IAAI,CAAC,qBAAqB,KAAK,CAAC,KAAK,IAAI,EAAE;YACzC,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA2B,CAAC;YAC/D,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,eAAe,EAAE,EAAE,CAAC;gBAC3D,IAAI,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBACvE,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;oBAClC,eAAe,GAAG,EAAE,+BAA+B,EAAE,eAAe,CAAC,+BAA+B,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;oBAC5H,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;gBACpE,CAAC;gBACD,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,eAAe,CAAC,EAAE,EAAE,qBAAqB,EAAE,eAAe,CAAC,qBAAqB,EAAE,CAAC,CAAC;YACjI,CAAC;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEO,KAAK,CAAC,0BAA0B;QACtC,IAAI,CAAC,wBAAwB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC5C,MAAM,CAAC,eAAe,EAAE,gCAAgC,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC5E,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAAqC,CAAC;oBAC7E,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,2BAA2B,EAAE,EAAE,CAAC;wBACvE,IAAI,UAAU,GAAG,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;wBACtE,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;4BAC7B,UAAU,GAAG,EAAE,WAAW,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;4BACxC,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;wBACnE,CAAC;wBACD,UAAU,CAAC,WAAW,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;oBACzD,CAAC;oBACD,OAAO,uBAAuB,CAAC;gBACjC,CAAC,CAAC,EAAE;gBACJ,IAAI,CAAC,mCAAmC,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkE,CAAC;YACzF,MAAM,SAAS,GAAG,oBAAoB,CAAC,gCAAgC,CAAC,CAAC;YACzE,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,eAAe,EAAE,CAAC;gBACpD,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC1C,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,UAAU,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC;YAC3E,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,wBAAwB,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QAMjC,MAAM,KAAK,GAAG;;;;;aAKL,IAAI,CAAC,mBAAmB;aACxB,IAAI,CAAC,qBAAqB;;;sDAGe,IAAI,CAAC,qBAAqB;KAC3E,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,oDAAoD,EAAE,CAC5H,EAAE,CAAC;YACF,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,cAAc,EAAE,GAAG,CAAC,cAAc,EAAE,CAAC;QACzI,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mCAAmC;QAC/C,IAAI,CAAC,iCAAiC,KAAK,CAAC,KAAK,IAAI,EAAE;YACrD,MAAM,gCAAgC,GAAG,IAAI,GAAG,EAAoC,CAAC;YACrF,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBAC1F,MAAM,GAAG,GAAqB,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;gBACzD,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACxD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,gCAAgC,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;gBACzE,CAAC;qBAAM,CAAC;oBACN,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,OAAO,gCAAgC,CAAC;QAC1C,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iCAAiC,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,4BAA4B,CAAC,OAAmB,EAAE,WAAsB;QACnF,MAAM,gCAAgC,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;QAC1F,MAAM,MAAM,GAAG,IAAI,KAAK,EAAa,CAAC;QACtC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;YAC/E,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,oBAAoB;QAChC,IAAI,CAAC,kBAAkB,KAAK,CAAC,KAAK,IAAI,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAkC,CAAC;YACnE,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClE,MAAM,oCAAoC,GAAG,IAAI,KAAK,EAAc,CAAC;YACrE,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE,CAAC;gBAChE,oCAAoC,CAAC,IAAI,CACvC,GAAG,mBAAmB,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,qBAAqB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC,CAC/I,CAAC;YACJ,CAAC;YAED,IAAI,oCAAoC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtD,OAAO,gBAAgB,CAAC;YAC1B,CAAC;YAED,IAAI,KAAK,EAAE,MAAM,kBAAkB,IAAI,IAAI,CAAC,yBAAyB,CAAC,oCAAoC,CAAC,EAAE,CAAC;gBAC5G,gBAAgB,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,kBAAkB,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC3F,CAAC;YACD,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,2BAA2B;QACvC,IAAI,CAAC,yBAAyB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC7C,MAAM,wBAAwB,GAAG,IAAI,GAAG,EAAkD,CAAC;YAC3F,MAAM,CAAC,8BAA8B,EAAE,oBAAoB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC/E,IAAI,CAAC,iCAAiC,EAAE;gBACxC,IAAI,CAAC,uBAAuB,EAAE;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,8BAA8B,IAAI,oBAAoB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACvE,OAAO,wBAAwB,CAAC;YAClC,CAAC;YAED,IAAI,KAAK,EAAE,MAAM,0BAA0B,IAAI,IAAI,CAAC,yBAAyB,EAAE,EAAE,CAAC;gBAChF,MAAM,mBAAmB,GAAG,oBAAoB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC;gBAEpF,wBAAwB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,EAAE;oBAC1D,eAAe,EAAE,mBAAmB,EAAE,eAAe,IAAI,EAAE;oBAC3D,OAAO,EAAE,0BAA0B,CAAC,OAAO;oBAC3C,2BAA2B,EAAE,EAAE;oBAC/B,+BAA+B,EAAE,KAAK;iBACvC,CAAC,CAAC;YACL,CAAC;YAED,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,wBAAwB,EAAE,CAAC;gBACxF,MAAM,yBAAyB,GAAG,wBAAwB,CAAC,GAAG,CAAC,uBAAuB,CAAC,OAAO,CAAC,CAAC;gBAChG,IAAI,yBAAyB,KAAK,SAAS,EAAE,CAAC;oBAC5C,yBAAyB,CAAC,2BAA2B,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;oBAClF,uBAAuB,CAAC,+BAA+B,GAAG,IAAI,CAAC;gBACjE,CAAC;YACH,CAAC;YAED,OAAO,wBAAwB,CAAC;QAClC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,yBAAyB,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,+CAA+C,CAC1D,4BAAuC;QAEvC,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAE1E,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAyB,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QAEnH,4BAA4B,CAAC,OAAO,CAAC,CAAC,2BAA2B,EAAE,EAAE;YACnE,MAAM,6BAA6B,GAAG,wBAAwB,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;YAChG,IAAI,6BAA6B,KAAK,SAAS,EAAE,CAAC;gBAChD,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,2BAA2B,CAAC,CAAC;gBAC/F,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,eAAe,CAAC,CAAC;YAC3E,CAAC;QACH,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,0BAA0B,CAAC,WAAsB,EAAE,gBAA0B;QACxF,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,MAAM,MAAM,GAAG,IAAI,GAAG,EAA4B,CAAC;QACnD,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,IAAI,uBAAuB,EAAE,CAAC;gBACzF,IAAI,CAAC,gBAAgB,IAAI,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;oBACpE,IAAI,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC5C,IAAI,CAAC,cAAc,EAAE,CAAC;wBACpB,cAAc,GAAG,IAAI,GAAG,EAAW,CAAC;wBACpC,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;oBACzC,CAAC;oBACD,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAAC,OAAmB;QAClD,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC;IACxE,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,SAAqB;QAC5C,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,OAAO,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,sBAAiC;QACtE,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QAEvC,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC1E,MAAM,kBAAkB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC1C,sBAAsB,CAAC,GAAG,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE;YACzD,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;YACpF,IAAI,uBAAuB,KAAK,SAAS,EAAE,CAAC;gBAC1C,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,uBAAuB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;YACvF,OAAO,IAAI,CAAC,yBAAyB,CAAC,uBAAuB,CAAC,2BAA2B,CAAC,CAAC;QAC7F,CAAC,CAAC,CACH,CAAC;QACF,KAAK,MAAM,UAAU,IAAI,kBAAkB,EAAE,CAAC;YAC5C,MAAM,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAC9B,KAAyG;QAEzG,IAAI,eAAe,IAAI,KAAK,EAAE,CAAC;YAC7B,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5D,MAAM,eAAe,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACnE,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;gBAClC,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,sBAAsB,EAAE,CAAC,CAAC;QAC3J,CAAC;QAED,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;YAC1B,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClE,KAAK,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC,IAAI,oBAAoB,EAAE,CAAC;gBAClE,IAAI,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;oBACvG,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE,CAAC;wBACzD,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;oBACpE,CAAC;oBAED,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;gBAC7I,CAAC;YACH,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC1E,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC1F,IAAI,uBAAuB,KAAK,SAAS,EAAE,CAAC;YAC1C,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE,CAAC;YAC7D,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,8BAA8B,EAAE,CAAC,CAAC;QAC1F,CAAC;QAED,OAAO;YACL,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,uBAAuB,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/F,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,8BAA8B,EAAE;SAC/E,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,KAAK,CAAC,uCAAuC;QAClD,MAAM,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,CAAC,GAAG,wBAAwB,CAAC,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,KAAK,EAAc,EAAE,CAAC;QACnH,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE,CAAC;YAChE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC;QAC1G,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,wCAAwC;QACnD,MAAM,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QACxG,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE,CAAC;YAChE,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE,CAAC;gBACzD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QAED,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,wBAAwB,EAAE,CAAC;YACxF,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE,CAAC;gBAC7D,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,WAAsB;QAClD,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAoC,CAAC;QAC3D,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,KAAK,MAAM,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,iBAAiB,EAAE,CAAC;gBACjE,IAAI,eAAe,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;oBAC9C,IAAI,aAAa,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC3C,IAAI,CAAC,aAAa,EAAE,CAAC;wBACnB,aAAa,GAAG,EAAE,CAAC;wBACnB,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;oBACxC,CAAC;oBACD,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBACpC,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,iCAAiC;QAC5C,IAAI,CAAC,+BAA+B,KAAK,IAAI,CAAC,oCAAoC,EAAE,CAAC;QACrF,OAAO,IAAI,CAAC,+BAA+B,CAAC;IAC9C,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { CLASS_NAME_DefinitionContainer, CLASS_NAME_Model, CLASS_NAME_SubCategory } from \"../../common/internal/ClassNameDefinitions.js\";\nimport { ModelCategoryElementsCountCache } from \"../../common/internal/ModelCategoryElementsCountCache.js\";\nimport { getClassesByView, getDistinctMapValues } from \"../../common/internal/Utils.js\";\n\nimport type { CategoryId, DefinitionContainerId, ElementId, ModelId, SubCategoryId } from \"../../common/internal/Types.js\";\nimport type { Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\n\ninterface DefinitionContainerInfo {\n  modelId: Id64String;\n  parentDefinitionContainerExists: boolean;\n  childCategories: CategoryInfo[];\n  childDefinitionContainerIds: Id64Array;\n}\n\ninterface CategoriesInfo {\n  childCategories: CategoryInfo[];\n  parentDefinitionContainerExists: boolean;\n}\n\n/** @internal */\nexport interface CategoryInfo {\n  id: CategoryId;\n  subCategoryChildCount: number;\n}\n\ninterface SubCategoryInfo {\n  categoryId: Id64String;\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\n/** @internal */\nexport class CategoriesTreeIdsCache implements Disposable {\n  private _definitionContainersInfo: Promise<Map<DefinitionContainerId, DefinitionContainerInfo>> | undefined;\n  private _modelsCategoriesInfo: Promise<Map<ModelId, CategoriesInfo>> | undefined;\n  private _elementModelsCategories: Promise<Map<ModelId, { categoryIds: Id64Set; isSubModel: boolean }>> | undefined;\n  private _subCategoriesInfo: Promise<Map<SubCategoryId, SubCategoryInfo>> | undefined;\n  private readonly _categoryElementCounts: ModelCategoryElementsCountCache;\n  private _modelWithCategoryModeledElements: Promise<Map<ModelCategoryKey, Set<ElementId>>> | undefined;\n  private _categoryClass: string;\n  private _categoryElementClass: string;\n  private _categoryModelClass: string;\n  private _isDefinitionContainerSupported: Promise<boolean> | undefined;\n  private _filteredElementsModels: Promise<Map<ElementId, ModelId>> | undefined;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    viewType: \"3d\" | \"2d\",\n  ) {\n    const { categoryClass, elementClass, modelClass } = getClassesByView(viewType);\n    this._categoryClass = categoryClass;\n    this._categoryElementClass = elementClass;\n    this._categoryModelClass = modelClass;\n    this._categoryElementCounts = new ModelCategoryElementsCountCache(_queryExecutor, [elementClass]);\n  }\n\n  public [Symbol.dispose]() {\n    this._categoryElementCounts[Symbol.dispose]();\n  }\n\n  private async *queryFilteredElementsModels(filteredElementIds: Id64Array): AsyncIterableIterator<{\n    modelId: Id64String;\n    id: ElementId;\n  }> {\n    const query = `\n      SELECT Model.Id modelId, ECInstanceId id\n      FROM ${this._categoryElementClass}\n      WHERE ECInstanceId IN (${filteredElementIds.join(\", \")})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, id: row.id };\n    }\n  }\n\n  public async getFilteredElementsModels(filteredElementIds: Id64Array) {\n    if (filteredElementIds.length === 0) {\n      return new Map<ElementId, ModelId>();\n    }\n\n    this._filteredElementsModels ??= (async () => {\n      const filteredElementsModels = new Map();\n      for await (const { modelId, id } of this.queryFilteredElementsModels(filteredElementIds)) {\n        filteredElementsModels.set(id, modelId);\n      }\n      return filteredElementsModels;\n    })();\n    return this._filteredElementsModels;\n  }\n\n  public clearFilteredElementsModels() {\n    this._filteredElementsModels = undefined;\n  }\n\n  private async *queryElementModelCategories(): AsyncIterableIterator<{\n    modelId: Id64String;\n    categoryId: Id64String;\n  }> {\n    const query = `\n      SELECT this.Model.Id modelId, this.Category.Id categoryId\n      FROM ${this._categoryModelClass} m\n      JOIN ${this._categoryElementClass} this ON m.ECInstanceId = this.Model.Id\n      WHERE this.Parent.Id IS NULL AND m.IsPrivate = false\n      GROUP BY modelId, categoryId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/element-models-and-categories-query\" },\n    )) {\n      yield { modelId: row.modelId, categoryId: row.categoryId };\n    }\n  }\n\n  private async *queryCategories(): AsyncIterableIterator<{\n    id: CategoryId;\n    modelId: Id64String;\n    parentDefinitionContainerExists: boolean;\n    subCategoryChildCount: number;\n  }> {\n    const isDefinitionContainerSupported = await this.getIsDefinitionContainerSupported();\n    const categoriesQuery = `\n      SELECT\n        this.ECInstanceId id,\n        COUNT(sc.ECInstanceId) subCategoryChildCount,\n        this.Model.Id modelId,\n        ${\n          isDefinitionContainerSupported\n            ? `\n            IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${CLASS_NAME_DefinitionContainer} dc),\n              true,\n              false\n            )`\n            : \"false\"\n        } parentDefinitionContainerExists\n      FROM\n        ${this._categoryClass} this\n        JOIN ${CLASS_NAME_SubCategory} sc ON sc.Parent.Id = this.ECInstanceId\n        JOIN ${CLASS_NAME_Model} m ON m.ECInstanceId = this.Model.Id\n      WHERE\n        NOT this.IsPrivate\n        AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))\n        AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)\n      GROUP BY this.ECInstanceId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: categoriesQuery },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/categories-query\" },\n    )) {\n      yield {\n        id: row.id,\n        modelId: row.modelId,\n        parentDefinitionContainerExists: row.parentDefinitionContainerExists,\n        subCategoryChildCount: row.subCategoryChildCount,\n      };\n    }\n  }\n\n  private async queryIsDefinitionContainersSupported(): Promise<boolean> {\n    const query = `\n      SELECT\n        1\n      FROM\n        ECDbMeta.ECSchemaDef s\n        JOIN ECDbMeta.ECClassDef c ON c.Schema.Id = s.ECInstanceId\n      WHERE\n        s.Name = 'BisCore'\n        AND c.Name = 'DefinitionContainer'\n    `;\n\n    for await (const _row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { restartToken: \"tree-widget/categories-tree/is-definition-container-supported-query\" },\n    )) {\n      return true;\n    }\n    return false;\n  }\n\n  private async *queryDefinitionContainers(): AsyncIterableIterator<{ id: DefinitionContainerId; modelId: Id64String }> {\n    // DefinitionModel ECInstanceId will always be the same as modeled DefinitionContainer ECInstanceId, if this wasn't the case, we would need to do something like:\n    //  JOIN BisCore.DefinitionModel dm ON dm.ECInstanceId = ${modelIdAccessor}\n    //  JOIN BisCore.DefinitionModelBreaksDownDefinitionContainer dr ON dr.SourceECInstanceId = dm.ECInstanceId\n    //  JOIN BisCore.DefinitionContainer dc ON dc.ECInstanceId = dr.TargetECInstanceId\n    const DEFINITION_CONTAINERS_CTE = \"DefinitionContainers\";\n    const ctes = [\n      `\n        ${DEFINITION_CONTAINERS_CTE}(ECInstanceId, ModelId) AS (\n          SELECT\n            dc.ECInstanceId,\n            dc.Model.Id\n          FROM\n            ${CLASS_NAME_DefinitionContainer} dc\n          WHERE\n            dc.ECInstanceId IN (SELECT c.Model.Id FROM ${this._categoryClass} c WHERE NOT c.IsPrivate AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = c.ECInstanceId))\n            AND NOT dc.IsPrivate\n\n          UNION ALL\n\n          SELECT\n            pdc.ECInstanceId,\n            pdc.Model.Id\n          FROM\n            ${DEFINITION_CONTAINERS_CTE} cdc\n            JOIN ${CLASS_NAME_DefinitionContainer} pdc ON pdc.ECInstanceId = cdc.ModelId\n          WHERE\n            NOT pdc.IsPrivate\n        )\n      `,\n    ];\n    const definitionsQuery = `\n      SELECT dc.ECInstanceId id, dc.ModelId modelId FROM ${DEFINITION_CONTAINERS_CTE} dc GROUP BY dc.ECInstanceId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ctes, ecsql: definitionsQuery },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/definition-containers-query\" },\n    )) {\n      yield { id: row.id, modelId: row.modelId };\n    }\n  }\n\n  private async *queryVisibleSubCategories(categoryIds: Id64Array): AsyncIterableIterator<{ id: SubCategoryId; parentId: CategoryId }> {\n    const definitionsQuery = `\n      SELECT\n        sc.ECInstanceId id,\n        sc.Parent.Id categoryId\n      FROM\n        ${CLASS_NAME_SubCategory} sc\n      WHERE\n        NOT sc.IsPrivate\n        AND sc.Parent.Id IN (${categoryIds.join(\",\")})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: definitionsQuery },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/sub-categories-query\" },\n    )) {\n      yield { id: row.id, parentId: row.categoryId };\n    }\n  }\n\n  private async getModelsCategoriesInfo() {\n    this._modelsCategoriesInfo ??= (async () => {\n      const allModelsCategories = new Map<ModelId, CategoriesInfo>();\n      for await (const queriedCategory of this.queryCategories()) {\n        let modelCategories = allModelsCategories.get(queriedCategory.modelId);\n        if (modelCategories === undefined) {\n          modelCategories = { parentDefinitionContainerExists: queriedCategory.parentDefinitionContainerExists, childCategories: [] };\n          allModelsCategories.set(queriedCategory.modelId, modelCategories);\n        }\n        modelCategories.childCategories.push({ id: queriedCategory.id, subCategoryChildCount: queriedCategory.subCategoryChildCount });\n      }\n      return allModelsCategories;\n    })();\n    return this._modelsCategoriesInfo;\n  }\n\n  private async getElementModelsCategories() {\n    this._elementModelsCategories ??= (async () => {\n      const [modelCategories, modelWithCategoryModeledElements] = await Promise.all([\n        (async () => {\n          const elementModelsCategories = new Map<ModelId, { categoryIds: Id64Set }>();\n          for await (const queriedCategory of this.queryElementModelCategories()) {\n            let modelEntry = elementModelsCategories.get(queriedCategory.modelId);\n            if (modelEntry === undefined) {\n              modelEntry = { categoryIds: new Set() };\n              elementModelsCategories.set(queriedCategory.modelId, modelEntry);\n            }\n            modelEntry.categoryIds.add(queriedCategory.categoryId);\n          }\n          return elementModelsCategories;\n        })(),\n        this.getModelWithCategoryModeledElements(),\n      ]);\n      const result = new Map<ModelId, { categoryIds: Set<CategoryId>; isSubModel: boolean }>();\n      const subModels = getDistinctMapValues(modelWithCategoryModeledElements);\n      for (const [modelId, modelEntry] of modelCategories) {\n        const isSubModel = subModels.has(modelId);\n        result.set(modelId, { categoryIds: modelEntry.categoryIds, isSubModel });\n      }\n      return result;\n    })();\n    return this._elementModelsCategories;\n  }\n\n  private async *queryModeledElements(): AsyncIterableIterator<{\n    modelId: Id64String;\n    modeledElementId: Id64String;\n    categoryId: Id64String;\n    rootCategoryId: Id64String;\n  }> {\n    const query = `\n      SELECT\n        pe.ECInstanceId modeledElementId,\n        pe.Category.Id categoryId,\n        pe.Model.Id modelId\n      FROM ${this._categoryModelClass} m\n      JOIN ${this._categoryElementClass} pe ON pe.ECInstanceId = m.ModeledElement.Id\n      WHERE\n        m.IsPrivate = false\n        AND m.ECInstanceId IN (SELECT Model.Id FROM ${this._categoryElementClass})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/modeled-elements-query\" },\n    )) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };\n    }\n  }\n\n  private async getModelWithCategoryModeledElements() {\n    this._modelWithCategoryModeledElements ??= (async () => {\n      const modelWithCategoryModeledElements = new Map<ModelCategoryKey, Set<ElementId>>();\n      for await (const { modelId, categoryId, modeledElementId } of this.queryModeledElements()) {\n        const key: ModelCategoryKey = `${modelId}-${categoryId}`;\n        const entry = modelWithCategoryModeledElements.get(key);\n        if (entry === undefined) {\n          modelWithCategoryModeledElements.set(key, new Set([modeledElementId]));\n        } else {\n          entry.add(modeledElementId);\n        }\n      }\n      return modelWithCategoryModeledElements;\n    })();\n    return this._modelWithCategoryModeledElements;\n  }\n\n  public async getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Array): Promise<Id64Array> {\n    const modelWithCategoryModeledElements = await this.getModelWithCategoryModeledElements();\n    const result = new Array<ElementId>();\n    for (const categoryId of categoryIds) {\n      const entry = modelWithCategoryModeledElements.get(`${modelId}-${categoryId}`);\n      if (entry !== undefined) {\n        result.push(...entry);\n      }\n    }\n    return result;\n  }\n\n  private async getSubCategoriesInfo() {\n    this._subCategoriesInfo ??= (async () => {\n      const allSubCategories = new Map<SubCategoryId, SubCategoryInfo>();\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      const categoriesWithMoreThanOneSubCategory = new Array<CategoryId>();\n      for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n        categoriesWithMoreThanOneSubCategory.push(\n          ...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.subCategoryChildCount > 1).map((categoryInfo) => categoryInfo.id),\n        );\n      }\n\n      if (categoriesWithMoreThanOneSubCategory.length === 0) {\n        return allSubCategories;\n      }\n\n      for await (const queriedSubCategory of this.queryVisibleSubCategories(categoriesWithMoreThanOneSubCategory)) {\n        allSubCategories.set(queriedSubCategory.id, { categoryId: queriedSubCategory.parentId });\n      }\n      return allSubCategories;\n    })();\n    return this._subCategoriesInfo;\n  }\n\n  private async getDefinitionContainersInfo() {\n    this._definitionContainersInfo ??= (async () => {\n      const definitionContainersInfo = new Map<DefinitionContainerId, DefinitionContainerInfo>();\n      const [isDefinitionContainerSupported, modelsCategoriesInfo] = await Promise.all([\n        this.getIsDefinitionContainerSupported(),\n        this.getModelsCategoriesInfo(),\n      ]);\n      if (!isDefinitionContainerSupported || modelsCategoriesInfo.size === 0) {\n        return definitionContainersInfo;\n      }\n\n      for await (const queriedDefinitionContainer of this.queryDefinitionContainers()) {\n        const modelCategoriesInfo = modelsCategoriesInfo.get(queriedDefinitionContainer.id);\n\n        definitionContainersInfo.set(queriedDefinitionContainer.id, {\n          childCategories: modelCategoriesInfo?.childCategories ?? [],\n          modelId: queriedDefinitionContainer.modelId,\n          childDefinitionContainerIds: [],\n          parentDefinitionContainerExists: false,\n        });\n      }\n\n      for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {\n        const parentDefinitionContainer = definitionContainersInfo.get(definitionContainerInfo.modelId);\n        if (parentDefinitionContainer !== undefined) {\n          parentDefinitionContainer.childDefinitionContainerIds.push(definitionContainerId);\n          definitionContainerInfo.parentDefinitionContainerExists = true;\n        }\n      }\n\n      return definitionContainersInfo;\n    })();\n    return this._definitionContainersInfo;\n  }\n\n  public async getDirectChildDefinitionContainersAndCategories(\n    parentDefinitionContainerIds: Id64Array,\n  ): Promise<{ categories: CategoryInfo[]; definitionContainers: Array<DefinitionContainerId> }> {\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n\n    const result = { definitionContainers: new Array<DefinitionContainerId>(), categories: new Array<CategoryInfo>() };\n\n    parentDefinitionContainerIds.forEach((parentDefinitionContainerId) => {\n      const parentDefinitionContainerInfo = definitionContainersInfo.get(parentDefinitionContainerId);\n      if (parentDefinitionContainerInfo !== undefined) {\n        result.definitionContainers.push(...parentDefinitionContainerInfo.childDefinitionContainerIds);\n        result.categories.push(...parentDefinitionContainerInfo.childCategories);\n      }\n    });\n    return result;\n  }\n\n  public async getCategoriesElementModels(categoryIds: Id64Array, includeSubModels?: boolean): Promise<Map<CategoryId, Set<ModelId>>> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    const result = new Map<CategoryId, Set<ModelId>>();\n    for (const categoryId of categoryIds) {\n      for (const [modelId, { categoryIds: categories, isSubModel }] of elementModelsCategories) {\n        if ((includeSubModels || !isSubModel) && categories.has(categoryId)) {\n          let categoryModels = result.get(categoryId);\n          if (!categoryModels) {\n            categoryModels = new Set<ModelId>();\n            result.set(categoryId, categoryModels);\n          }\n          categoryModels.add(modelId);\n        }\n      }\n    }\n    return result;\n  }\n\n  public async getModelCategoryIds(modelId: Id64String): Promise<Id64Array> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    return [...(elementModelsCategories.get(modelId)?.categoryIds ?? [])];\n  }\n\n  public async hasSubModel(elementId: Id64String): Promise<boolean> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    return elementModelsCategories.has(elementId);\n  }\n\n  public async getAllContainedCategories(definitionContainerIds: Id64Array): Promise<Id64Array> {\n    const result = new Array<CategoryId>();\n\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n    const indirectCategories = await Promise.all(\n      definitionContainerIds.map(async (definitionContainerId) => {\n        const definitionContainerInfo = definitionContainersInfo.get(definitionContainerId);\n        if (definitionContainerInfo === undefined) {\n          return [];\n        }\n        result.push(...definitionContainerInfo.childCategories.map((category) => category.id));\n        return this.getAllContainedCategories(definitionContainerInfo.childDefinitionContainerIds);\n      }),\n    );\n    for (const categories of indirectCategories) {\n      result.push(...categories);\n    }\n\n    return result;\n  }\n\n  public async getInstanceKeyPaths(\n    props: { categoryId: Id64String } | { definitionContainerId: Id64String } | { subCategoryId: Id64String },\n  ): Promise<InstanceKey[]> {\n    if (\"subCategoryId\" in props) {\n      const subCategoriesInfo = await this.getSubCategoriesInfo();\n      const subCategoryInfo = subCategoriesInfo.get(props.subCategoryId);\n      if (subCategoryInfo === undefined) {\n        return [];\n      }\n      return [...(await this.getInstanceKeyPaths({ categoryId: subCategoryInfo.categoryId })), { id: props.subCategoryId, className: CLASS_NAME_SubCategory }];\n    }\n\n    if (\"categoryId\" in props) {\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      for (const [modelId, modelCategoriesInfo] of modelsCategoriesInfo) {\n        if (modelCategoriesInfo.childCategories.find((childCategory) => childCategory.id === props.categoryId)) {\n          if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n            return [{ id: props.categoryId, className: this._categoryClass }];\n          }\n\n          return [...(await this.getInstanceKeyPaths({ definitionContainerId: modelId })), { id: props.categoryId, className: this._categoryClass }];\n        }\n      }\n      return [];\n    }\n\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n    const definitionContainerInfo = definitionContainersInfo.get(props.definitionContainerId);\n    if (definitionContainerInfo === undefined) {\n      return [];\n    }\n\n    if (!definitionContainerInfo.parentDefinitionContainerExists) {\n      return [{ id: props.definitionContainerId, className: CLASS_NAME_DefinitionContainer }];\n    }\n\n    return [\n      ...(await this.getInstanceKeyPaths({ definitionContainerId: definitionContainerInfo.modelId })),\n      { id: props.definitionContainerId, className: CLASS_NAME_DefinitionContainer },\n    ];\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  public async getAllDefinitionContainersAndCategories(): Promise<{ categories: Array<CategoryId>; definitionContainers: Array<DefinitionContainerId> }> {\n    const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);\n    const result = { definitionContainers: [...definitionContainersInfo.keys()], categories: new Array<Id64String>() };\n    for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n      result.categories.push(...modelCategoriesInfo.childCategories.map((childCategory) => childCategory.id));\n    }\n\n    return result;\n  }\n\n  public async getRootDefinitionContainersAndCategories(): Promise<{ categories: CategoryInfo[]; definitionContainers: Array<DefinitionContainerId> }> {\n    const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);\n    const result = { definitionContainers: new Array<Id64String>(), categories: new Array<CategoryInfo>() };\n    for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n      if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n        result.categories.push(...modelCategoriesInfo.childCategories);\n      }\n    }\n\n    for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {\n      if (!definitionContainerInfo.parentDefinitionContainerExists) {\n        result.definitionContainers.push(definitionContainerId);\n      }\n    }\n    return result;\n  }\n\n  public async getSubCategories(categoryIds: Id64Array): Promise<Map<CategoryId, Array<SubCategoryId>>> {\n    const subCategoriesInfo = await this.getSubCategoriesInfo();\n    const result = new Map<CategoryId, Array<SubCategoryId>>();\n    for (const categoryId of categoryIds) {\n      for (const [subCategoryId, subCategoryInfo] of subCategoriesInfo) {\n        if (subCategoryInfo.categoryId === categoryId) {\n          let categoryEntry = result.get(categoryId);\n          if (!categoryEntry) {\n            categoryEntry = [];\n            result.set(categoryId, categoryEntry);\n          }\n          categoryEntry.push(subCategoryId);\n        }\n      }\n    }\n\n    return result;\n  }\n\n  public async getIsDefinitionContainerSupported(): Promise<boolean> {\n    this._isDefinitionContainerSupported ??= this.queryIsDefinitionContainersSupported();\n    return this._isDefinitionContainerSupported;\n  }\n}\n"]}

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

@@ -9,7 +9,7 @@ import { HierarchyNode, HierarchyNodeKey } from "@itwin/presentation-hierarchies
 import { AlwaysAndNeverDrawnElementInfo } from "../../common/internal/AlwaysAndNeverDrawnElementInfo.js";
 import { toVoidPromise } from "../../common/internal/Rxjs.js";
 import { createVisibilityStatus } from "../../common/internal/Tooltip.js";
-import { getClassesByView, releaseMainThreadOnItemsCount, setDifference, setIntersection } from "../../common/internal/Utils.js";
+import { getClassesByView, getDistinctMapValues, releaseMainThreadOnItemsCount, setDifference, setIntersection } from "../../common/internal/Utils.js";
 import { createVisibilityChangeEventListener } from "../../common/internal/VisibilityChangeEventListener.js";
 import { changeElementStateNoChildrenOperator, enableCategoryDisplay, enableSubCategoryDisplay, filterSubModeledElementIds, getElementOverriddenVisibility, getElementVisibility, getSubModeledElementsVisibilityStatus, getVisibilityFromAlwaysAndNeverDrawnElementsImpl, mergeVisibilityStatuses, } from "../../common/internal/VisibilityUtils.js";
 import { createVisibilityHandlerResult } from "../../common/UseHierarchyVisibility.js";
@@ -199,7 +199,7 @@ class CategoriesTreeVisibilityHandlerImpl {
         const result = defer(async () => {
             const { categoryId, subCategoryIds } = props;
             let visibility = "unknown";
-            const categoryModels = [...(await this._idsCache.getCategoriesElementModels([categoryId], true)).values()].flat();
+            const categoryModels = getDistinctMapValues(await this._idsCache.getCategoriesElementModels([categoryId], true));
             let nonDefaultModelDisplayStatesCount = 0;
             for (const modelId of categoryModels) {
                 if (!this._props.viewport.view.viewsModel(modelId)) {
@@ -228,7 +228,7 @@ class CategoriesTreeVisibilityHandlerImpl {
                     continue;
                 }
             }
-            if (categoryModels.length > 0 && nonDefaultModelDisplayStatesCount === categoryModels.length) {
+            if (categoryModels.size > 0 && nonDefaultModelDisplayStatesCount === categoryModels.size) {
                 assert(visibility === "visible" || visibility === "hidden");
                 return createVisibilityStatus(visibility);
             }
@@ -300,7 +300,7 @@ class CategoriesTreeVisibilityHandlerImpl {
                 return from(this.getDefaultCategoryVisibilityStatus({ categoryIds: props.categoryIds }));
             }
             const modelsObservable = props.modelId
-                ? of(new Map(props.categoryIds.map((id) => [id, [props.modelId]])))
+                ? of(new Map(props.categoryIds.map((id) => [id, new Set([props.modelId])])))
                 : from(this._idsCache.getCategoriesElementModels(props.categoryIds));
             return merge(
             // get visibility status from always and never drawn elements
@@ -572,8 +572,13 @@ class CategoriesTreeVisibilityHandlerImpl {
     }
     async enableCategoriesElementModelsVisibility(categoryIds) {
         const categoriesModelsMap = await this._idsCache.getCategoriesElementModels(categoryIds, true);
-        const modelIds = [...categoriesModelsMap.values()].flat();
-        const hiddenModels = modelIds.filter((modelId) => !this._props.viewport.view.viewsModel(modelId));
+        const modelIds = getDistinctMapValues(categoriesModelsMap);
+        const hiddenModels = new Array();
+        modelIds.forEach((modelId) => {
+            if (!this._props.viewport.view.viewsModel(modelId)) {
+                hiddenModels.push(modelId);
+            }
+        });
         if (hiddenModels.length > 0) {
             this._props.viewport.changeModelDisplay(hiddenModels, true);
         }
@@ -594,7 +599,7 @@ class CategoriesTreeVisibilityHandlerImpl {
                 return concat(from(enableCategoryDisplay(viewport, categoryIds, on, on)), on ? from(this.enableCategoriesElementModelsVisibility(categoryIds)) : EMPTY);
             }
             const modelIdsObservable = modelId
-                ? of(new Map(categoryIds.map((id) => [id, [modelId]])))
+                ? of(new Map(categoryIds.map((id) => [id, new Set([modelId])])))
                 : from(this._idsCache.getCategoriesElementModels(categoryIds, true));
             return concat(modelId ? EMPTY : from(enableCategoryDisplay(viewport, categoryIds, on, ignoreSubCategories ? false : on)), on
                 ? modelIdsObservable.pipe(mergeMap((categoriesMap) => from(categoriesMap.values())), mergeAll(), filter((modelIdToCheck) => !viewport.view.viewsModel(modelIdToCheck)), mergeMap((modelIdToCheck) => this.showModelWithoutAnyCategoriesOrElements(modelIdToCheck)))
@@ -681,7 +686,7 @@ class CategoriesTreeVisibilityHandlerImpl {
             }));
         }
         const { modelId, categoryIds } = props.queryProps;
-        const totalCount = (modelId ? of(new Map(categoryIds.map((categoryId) => [categoryId, [modelId]]))) : from(this._idsCache.getCategoriesElementModels(categoryIds))).pipe(mergeMap((categoriesMap) => from(categoriesMap)), mergeMap(([categoryId, modelIds]) => {
+        const totalCount = (modelId ? of(new Map(categoryIds.map((categoryId) => [categoryId, new Set([modelId])]))) : from(this._idsCache.getCategoriesElementModels(categoryIds))).pipe(mergeMap((categoriesMap) => from(categoriesMap)), mergeMap(([categoryId, modelIds]) => {
             return from(modelIds).pipe(mergeMap((modelOfCategory) => from(this._idsCache.getCategoryElementsCount(modelOfCategory, categoryId))), reduce((acc, specificModelCategoryCount) => {
                 return acc + specificModelCategoryCount;
             }, 0));