@itwin/tree-widget-react 4.0.0-alpha.6 → 4.0.0-alpha.7

package/lib/esm/tree-widget-react/components/trees/common/components/Tree.css

@@ -4,7 +4,8 @@
  *--------------------------------------------------------------------------------------------*/
 
 .tw-tree-renderer-container {
-  overflow: auto;
   height: 100%;
   padding-top: var(--iui-size-xs);
+  display: flex;
+  flex-direction: column;
 }

package/lib/esm/tree-widget-react/components/trees/common/components/Tree.js

@@ -8,12 +8,12 @@ import { useCallback, useMemo, useState } from "react";
 import { BeEvent } from "@itwin/core-bentley";
 import { SchemaMetadataContextProvider } from "@itwin/presentation-components";
 import { useIModelUnifiedSelectionTree } from "@itwin/presentation-hierarchies-react";
-import { useHierarchiesLocalization } from "../UseHierarchiesLocalization.js";
-import { useHierarchyLevelFiltering } from "../UseHierarchyFiltering.js";
-import { useIModelChangeListener } from "../UseIModelChangeListener.js";
+import { useHierarchiesLocalization } from "../internal/UseHierarchiesLocalization.js";
+import { useHierarchyLevelFiltering } from "../internal/UseHierarchyFiltering.js";
+import { useIModelChangeListener } from "../internal/UseIModelChangeListener.js";
+import { createIModelAccess } from "../internal/Utils.js";
 import { useNodeHighlighting } from "../UseNodeHighlighting.js";
 import { useReportingAction, useTelemetryContext } from "../UseTelemetryContext.js";
-import { createIModelAccess } from "../Utils.js";
 import { Delayed } from "./Delayed.js";
 import { EmptyTreeContent } from "./EmptyTree.js";
 import { ProgressOverlay } from "./ProgressOverlay.js";

package/lib/esm/tree-widget-react/components/trees/common/components/Tree.js.map

@@ -1 +1 @@
-{"version":3,"file":"Tree.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/Tree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,YAAY,CAAC;AACpB,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACvD,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,6BAA6B,EAAE,MAAM,gCAAgC,CAAC;AAC/E,OAAO,EAAE,6BAA6B,EAAE,MAAM,uCAAuC,CAAC;AACtF,OAAO,EAAE,0BAA0B,EAAE,MAAM,kCAAkC,CAAC;AAC9E,OAAO,EAAE,0BAA0B,EAAE,MAAM,6BAA6B,CAAC;AACzE,OAAO,EAAE,uBAAuB,EAAE,MAAM,+BAA+B,CAAC;AACxE,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AACpF,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAuDjD;;;GAGG;AACH,MAAM,UAAU,IAAI,CAAC,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,YAAY,EAAE,oBAAoB,EAAE,GAAG,KAAK,EAAa;IAC3I,MAAM,8BAA8B,GAAG,uBAAuB,IAAI,IAAI,CAAC;IAEvE,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,EAAE;QAChC,OAAO,oBAAoB,IAAI,kBAAkB,CAAC,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAChG,CAAC,EAAE,CAAC,oBAAoB,EAAE,gBAAgB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAE3D,OAAO,CACL,KAAC,6BAA6B,IAAC,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,qBAAqB,EAAE,gBAAgB,YAC1F,KAAC,QAAQ,OAAK,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,YAAY,EAAE,YAAY,EAAE,8BAA8B,EAAE,8BAA8B,GAAI,GACzH,CACjC,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,EAChB,MAAM,EACN,YAAY,EACZ,QAAQ,EACR,gBAAgB,EAChB,gBAAgB,EAChB,8BAA8B,EAC9B,sBAAsB,EACtB,kBAAkB,EAClB,aAAa,EACb,QAAQ,EACR,YAAY,EACZ,gBAAgB,EAChB,SAAS,GACsG;IAC/G,MAAM,gBAAgB,GAAG,0BAA0B,EAAE,CAAC;IACtD,MAAM,EAAE,aAAa,EAAE,qBAAqB,EAAE,GAAG,mBAAmB,EAAE,CAAC;IACvE,MAAM,CAAC,aAAa,CAAC,GAAG,QAAQ,CAAC,IAAI,OAAO,EAAc,CAAC,CAAC;IAC5D,MAAM,EACJ,SAAS,EACT,OAAO,EACP,SAAS,EACT,WAAW,EAAE,iBAAiB,EAC9B,YAAY,EAAE,aAAa,EAC3B,UAAU,EACV,GAAG,SAAS,EACb,GAAG,6BAA6B,CAAC;QAChC,YAAY;QACZ,aAAa;QACb,sBAAsB;QACtB,gBAAgB;QAChB,UAAU,EAAE,QAAQ;QACpB,gBAAgB;QAChB,gBAAgB;QAChB,qBAAqB,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE;YAC1C,IAAI,MAAM,KAAK,QAAQ,EAAE;gBACvB,QAAQ,EAAE,EAAE,CAAC;aACd;YACD,qBAAqB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QACD,wBAAwB,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,SAAS,EAAE,gCAAgC,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;QACxH,oBAAoB,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE;YACxC,sCAAsC;YACtC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACrB,aAAa,CAAC,EAAE,SAAS,EAAE,SAAS,IAAI,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1E,CAAC;KACF,CAAC,CAAC;IACH,uBAAuB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC;IAE5G,MAAM,WAAW,GAAG,qBAAqB,CAAC;QACxC,MAAM,EAAE,kBAAkB,CAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;QACzD,SAAS,EAAE,kBAAkB;QAC7B,OAAO;KACR,CAAC,CAAC;IACH,MAAM,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,0BAA0B,CAAC;QACpE,MAAM;QACN,8BAA8B;KAC/B,CAAC,CAAC;IACH,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;IACvE,MAAM,wBAAwB,GAAG,kBAAkB,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;IAC/E,MAAM,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;IAEnE,IAAI,SAAS,KAAK,SAAS,EAAE;QAC3B,OAAO,KAAC,YAAY,KAAG,CAAC;KACzB;IAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE;QACxC,OAAO,4BAAG,gBAAgB,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAC,gBAAgB,KAAG,GAAI,CAAC;KAC1E;IAED,MAAM,iBAAiB,GAA6C;QAClE,GAAG,SAAS;QACZ,SAAS;QACT,aAAa,EAAE,aAAa,IAAI,QAAQ;QACxC,WAAW;QACX,UAAU,EAAE,mBAAmB;QAC/B,aAAa,EAAE,wBAAwB;QACvC,QAAQ;KACT,CAAC;IAEF,OAAO,CACL,eAAK,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,aACtE,eAAK,SAAS,EAAE,4BAA4B,EAAE,EAAE,EAAC,4BAA4B,aAC1E,YAAY,CAAC,iBAAiB,CAAC,EAC/B,eAAe,IACZ,EACN,KAAC,OAAO,IAAC,IAAI,EAAE,SAAS,YACtB,KAAC,eAAe,KAAG,GACX,IACN,CACP,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,EAC7B,MAAM,EACN,SAAS,EACT,OAAO,GAKR;IACC,OAAO,WAAW,CAChB,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,CACtB,MAAM,CACJ,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;QACxB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7B,OAAO,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,EACF,UAAU,CACX,EACH,CAAC,MAAM,EAAE,OAAO,EAAE,SAAS,CAAC,CAC7B,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport \"./Tree.css\";\nimport { useCallback, useMemo, useState } from \"react\";\nimport { BeEvent } from \"@itwin/core-bentley\";\nimport { SchemaMetadataContextProvider } from \"@itwin/presentation-components\";\nimport { useIModelUnifiedSelectionTree } from \"@itwin/presentation-hierarchies-react\";\nimport { useHierarchiesLocalization } from \"../UseHierarchiesLocalization.js\";\nimport { useHierarchyLevelFiltering } from \"../UseHierarchyFiltering.js\";\nimport { useIModelChangeListener } from \"../UseIModelChangeListener.js\";\nimport { useNodeHighlighting } from \"../UseNodeHighlighting.js\";\nimport { useReportingAction, useTelemetryContext } from \"../UseTelemetryContext.js\";\nimport { createIModelAccess } from \"../Utils.js\";\nimport { Delayed } from \"./Delayed.js\";\nimport { EmptyTreeContent } from \"./EmptyTree.js\";\nimport { ProgressOverlay } from \"./ProgressOverlay.js\";\nimport { SkeletonTree } from \"./SkeletonTree.js\";\n\nimport type { BaseTreeRendererProps } from \"./BaseTreeRenderer.js\";\nimport type { MarkRequired } from \"@itwin/core-bentley\";\nimport type { FunctionProps } from \"../Utils.js\";\nimport type { ReactNode } from \"react\";\nimport type { IModelConnection } from \"@itwin/core-frontend\";\nimport type { SchemaContext } from \"@itwin/ecschema-metadata\";\nimport type { PresentationHierarchyNode, SelectionStorage, useIModelTree, useSelectionHandler } from \"@itwin/presentation-hierarchies-react\";\nimport type { HighlightInfo } from \"../UseNodeHighlighting.js\";\n/** @beta */\nexport type TreeProps = Pick<FunctionProps<typeof useIModelTree>, \"getFilteredPaths\" | \"getHierarchyDefinition\"> &\n  Partial<Pick<FunctionProps<typeof useSelectionHandler>, \"selectionMode\">> & {\n    /** iModel connection that should be used to pull data from. */\n    imodel: IModelConnection;\n    /** Callback for getting `SchemaContext` for specific iModel. */\n    getSchemaContext: (imodel: IModelConnection) => SchemaContext;\n    /** Unique tree component name that will be used as unified selection change event source when selecting node. */\n    treeName: string;\n    /** Unified selection storage that should be used by tree to handle tree selection changes. */\n    selectionStorage: SelectionStorage;\n    /**\n     * An optional predicate to allow or prohibit selection of a node.\n     * When not supplied, all nodes are selectable.\n     */\n    selectionPredicate?: (node: PresentationHierarchyNode) => boolean;\n    /** Tree renderer that should be used to render tree data. */\n    treeRenderer: (\n      treeProps: Required<\n        Pick<\n          BaseTreeRendererProps,\n          | \"rootNodes\"\n          | \"expandNode\"\n          | \"getLabel\"\n          | \"onFilterClick\"\n          | \"selectNodes\"\n          | \"selectionMode\"\n          | \"isNodeSelected\"\n          | \"getHierarchyLevelDetails\"\n          | \"getLabel\"\n        >\n      >,\n    ) => ReactNode;\n    /** Custom iModel access that is stored outside tree component. If not provided it new iModel access will be created using `imodel` prop. */\n    imodelAccess?: FunctionProps<typeof useIModelTree>[\"imodelAccess\"];\n    /** Size limit that should be applied on each hierarchy level. Default to `1000`. */\n    hierarchyLevelSizeLimit?: number;\n    /** Component that should be renderer if there are no tree nodes. */\n    emptyTreeContent?: ReactNode;\n    /** Callback that this invoked when tree reloads. */\n    onReload?: () => void;\n    /** Options for highlighting node labels. */\n    highlight?: HighlightInfo;\n  };\n\n/**\n * Default tree component that manages tree state and renders using supplied `treeRenderer`.\n * @beta\n */\nexport function Tree({ getSchemaContext, hierarchyLevelSizeLimit, selectionStorage, imodelAccess: providedIModelAccess, ...props }: TreeProps) {\n  const defaultHierarchyLevelSizeLimit = hierarchyLevelSizeLimit ?? 1000;\n\n  const imodelAccess = useMemo(() => {\n    return providedIModelAccess ?? createIModelAccess({ getSchemaContext, imodel: props.imodel });\n  }, [providedIModelAccess, getSchemaContext, props.imodel]);\n\n  return (\n    <SchemaMetadataContextProvider imodel={props.imodel} schemaContextProvider={getSchemaContext}>\n      <TreeImpl {...props} selectionStorage={selectionStorage} imodelAccess={imodelAccess} defaultHierarchyLevelSizeLimit={defaultHierarchyLevelSizeLimit} />\n    </SchemaMetadataContextProvider>\n  );\n}\n\nfunction TreeImpl({\n  imodel,\n  imodelAccess,\n  treeName,\n  emptyTreeContent,\n  getFilteredPaths,\n  defaultHierarchyLevelSizeLimit,\n  getHierarchyDefinition,\n  selectionPredicate,\n  selectionMode,\n  onReload,\n  treeRenderer,\n  selectionStorage,\n  highlight,\n}: MarkRequired<Omit<TreeProps, \"getSchemaContext\">, \"imodelAccess\"> & { defaultHierarchyLevelSizeLimit: number }) {\n  const localizedStrings = useHierarchiesLocalization();\n  const { onFeatureUsed, onPerformanceMeasured } = useTelemetryContext();\n  const [imodelChanged] = useState(new BeEvent<() => void>());\n  const {\n    rootNodes,\n    getNode,\n    isLoading,\n    selectNodes: selectNodesAction,\n    setFormatter: _setFormatter,\n    expandNode,\n    ...treeProps\n  } = useIModelUnifiedSelectionTree({\n    imodelAccess,\n    imodelChanged,\n    getHierarchyDefinition,\n    getFilteredPaths,\n    sourceName: treeName,\n    localizedStrings,\n    selectionStorage,\n    onPerformanceMeasured: (action, duration) => {\n      if (action === \"reload\") {\n        onReload?.();\n      }\n      onPerformanceMeasured(action, duration);\n    },\n    onHierarchyLimitExceeded: () => onFeatureUsed({ featureId: \"hierarchy-level-size-limit-hit\", reportInteraction: false }),\n    onHierarchyLoadError: ({ type, error }) => {\n      // eslint-disable-next-line no-console\n      console.error(error);\n      onFeatureUsed({ featureId: `error-${type}`, reportInteraction: false });\n    },\n  });\n  useIModelChangeListener({ imodel, action: useCallback(() => imodelChanged.raiseEvent(), [imodelChanged]) });\n\n  const selectNodes = useSelectionPredicate({\n    action: useReportingAction({ action: selectNodesAction }),\n    predicate: selectionPredicate,\n    getNode,\n  });\n  const { filteringDialog, onFilterClick } = useHierarchyLevelFiltering({\n    imodel,\n    defaultHierarchyLevelSizeLimit,\n  });\n  const reportingExpandNode = useReportingAction({ action: expandNode });\n  const reportingOnFilterClicked = useReportingAction({ action: onFilterClick });\n  const { getLabel } = useNodeHighlighting({ rootNodes, highlight });\n\n  if (rootNodes === undefined) {\n    return <SkeletonTree />;\n  }\n\n  if (rootNodes.length === 0 && !isLoading) {\n    return <>{emptyTreeContent ? emptyTreeContent : <EmptyTreeContent />}</>;\n  }\n\n  const treeRendererProps: FunctionProps<TreeProps[\"treeRenderer\"]> = {\n    ...treeProps,\n    rootNodes,\n    selectionMode: selectionMode ?? \"single\",\n    selectNodes,\n    expandNode: reportingExpandNode,\n    onFilterClick: reportingOnFilterClicked,\n    getLabel,\n  };\n\n  return (\n    <div style={{ position: \"relative\", height: \"100%\", overflow: \"hidden\" }}>\n      <div className={\"tw-tree-renderer-container\"} id=\"tw-tree-renderer-container\">\n        {treeRenderer(treeRendererProps)}\n        {filteringDialog}\n      </div>\n      <Delayed show={isLoading}>\n        <ProgressOverlay />\n      </Delayed>\n    </div>\n  );\n}\n\nfunction useSelectionPredicate({\n  action,\n  predicate,\n  getNode,\n}: {\n  action: (...args: any[]) => void;\n  predicate?: (node: PresentationHierarchyNode) => boolean;\n  getNode: (nodeId: string) => PresentationHierarchyNode | undefined;\n}): ReturnType<typeof useIModelUnifiedSelectionTree>[\"selectNodes\"] {\n  return useCallback(\n    (nodeIds, changeType) =>\n      action(\n        nodeIds.filter((nodeId) => {\n          const node = getNode(nodeId);\n          return node && (!predicate || predicate(node));\n        }),\n        changeType,\n      ),\n    [action, getNode, predicate],\n  );\n}\n"]}
+{"version":3,"file":"Tree.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/Tree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,YAAY,CAAC;AACpB,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACvD,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,6BAA6B,EAAE,MAAM,gCAAgC,CAAC;AAC/E,OAAO,EAAE,6BAA6B,EAAE,MAAM,uCAAuC,CAAC;AACtF,OAAO,EAAE,0BAA0B,EAAE,MAAM,2CAA2C,CAAC;AACvF,OAAO,EAAE,0BAA0B,EAAE,MAAM,sCAAsC,CAAC;AAClF,OAAO,EAAE,uBAAuB,EAAE,MAAM,wCAAwC,CAAC;AACjF,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AACpF,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAwDjD;;;GAGG;AACH,MAAM,UAAU,IAAI,CAAC,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,YAAY,EAAE,oBAAoB,EAAE,GAAG,KAAK,EAAa;IAC3I,MAAM,8BAA8B,GAAG,uBAAuB,IAAI,IAAI,CAAC;IAEvE,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,EAAE;QAChC,OAAO,oBAAoB,IAAI,kBAAkB,CAAC,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAChG,CAAC,EAAE,CAAC,oBAAoB,EAAE,gBAAgB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAE3D,OAAO,CACL,KAAC,6BAA6B,IAAC,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,qBAAqB,EAAE,gBAAgB,YAC1F,KAAC,QAAQ,OAAK,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,YAAY,EAAE,YAAY,EAAE,8BAA8B,EAAE,8BAA8B,GAAI,GACzH,CACjC,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,EAChB,MAAM,EACN,YAAY,EACZ,QAAQ,EACR,gBAAgB,EAChB,gBAAgB,EAChB,8BAA8B,EAC9B,sBAAsB,EACtB,kBAAkB,EAClB,aAAa,EACb,QAAQ,EACR,YAAY,EACZ,gBAAgB,EAChB,SAAS,GACsG;IAC/G,MAAM,gBAAgB,GAAG,0BAA0B,EAAE,CAAC;IACtD,MAAM,EAAE,aAAa,EAAE,qBAAqB,EAAE,GAAG,mBAAmB,EAAE,CAAC;IACvE,MAAM,CAAC,aAAa,CAAC,GAAG,QAAQ,CAAC,IAAI,OAAO,EAAc,CAAC,CAAC;IAC5D,MAAM,EACJ,SAAS,EACT,OAAO,EACP,SAAS,EACT,WAAW,EAAE,iBAAiB,EAC9B,YAAY,EAAE,aAAa,EAC3B,UAAU,EACV,GAAG,SAAS,EACb,GAAG,6BAA6B,CAAC;QAChC,YAAY;QACZ,aAAa;QACb,sBAAsB;QACtB,gBAAgB;QAChB,UAAU,EAAE,QAAQ;QACpB,gBAAgB;QAChB,gBAAgB;QAChB,qBAAqB,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE;YAC1C,IAAI,MAAM,KAAK,QAAQ,EAAE;gBACvB,QAAQ,EAAE,EAAE,CAAC;aACd;YACD,qBAAqB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QACD,wBAAwB,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,SAAS,EAAE,gCAAgC,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;QACxH,oBAAoB,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE;YACxC,sCAAsC;YACtC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACrB,aAAa,CAAC,EAAE,SAAS,EAAE,SAAS,IAAI,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1E,CAAC;KACF,CAAC,CAAC;IACH,uBAAuB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC;IAE5G,MAAM,WAAW,GAAG,qBAAqB,CAAC;QACxC,MAAM,EAAE,kBAAkB,CAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;QACzD,SAAS,EAAE,kBAAkB;QAC7B,OAAO;KACR,CAAC,CAAC;IACH,MAAM,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,0BAA0B,CAAC;QACpE,MAAM;QACN,8BAA8B;KAC/B,CAAC,CAAC;IACH,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;IACvE,MAAM,wBAAwB,GAAG,kBAAkB,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;IAC/E,MAAM,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;IAEnE,IAAI,SAAS,KAAK,SAAS,EAAE;QAC3B,OAAO,KAAC,YAAY,KAAG,CAAC;KACzB;IAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE;QACxC,OAAO,4BAAG,gBAAgB,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAC,gBAAgB,KAAG,GAAI,CAAC;KAC1E;IAED,MAAM,iBAAiB,GAA6C;QAClE,GAAG,SAAS;QACZ,SAAS;QACT,aAAa,EAAE,aAAa,IAAI,QAAQ;QACxC,WAAW;QACX,UAAU,EAAE,mBAAmB;QAC/B,aAAa,EAAE,wBAAwB;QACvC,QAAQ;KACT,CAAC;IAEF,OAAO,CACL,eAAK,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,aACtE,eAAK,SAAS,EAAE,4BAA4B,EAAE,EAAE,EAAC,4BAA4B,aAC1E,YAAY,CAAC,iBAAiB,CAAC,EAC/B,eAAe,IACZ,EACN,KAAC,OAAO,IAAC,IAAI,EAAE,SAAS,YACtB,KAAC,eAAe,KAAG,GACX,IACN,CACP,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,EAC7B,MAAM,EACN,SAAS,EACT,OAAO,GAKR;IACC,OAAO,WAAW,CAChB,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,CACtB,MAAM,CACJ,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;QACxB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7B,OAAO,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,EACF,UAAU,CACX,EACH,CAAC,MAAM,EAAE,OAAO,EAAE,SAAS,CAAC,CAC7B,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport \"./Tree.css\";\nimport { useCallback, useMemo, useState } from \"react\";\nimport { BeEvent } from \"@itwin/core-bentley\";\nimport { SchemaMetadataContextProvider } from \"@itwin/presentation-components\";\nimport { useIModelUnifiedSelectionTree } from \"@itwin/presentation-hierarchies-react\";\nimport { useHierarchiesLocalization } from \"../internal/UseHierarchiesLocalization.js\";\nimport { useHierarchyLevelFiltering } from \"../internal/UseHierarchyFiltering.js\";\nimport { useIModelChangeListener } from \"../internal/UseIModelChangeListener.js\";\nimport { createIModelAccess } from \"../internal/Utils.js\";\nimport { useNodeHighlighting } from \"../UseNodeHighlighting.js\";\nimport { useReportingAction, useTelemetryContext } from \"../UseTelemetryContext.js\";\nimport { Delayed } from \"./Delayed.js\";\nimport { EmptyTreeContent } from \"./EmptyTree.js\";\nimport { ProgressOverlay } from \"./ProgressOverlay.js\";\nimport { SkeletonTree } from \"./SkeletonTree.js\";\n\nimport type { BaseTreeRendererProps } from \"./BaseTreeRenderer.js\";\nimport type { MarkRequired } from \"@itwin/core-bentley\";\nimport type { FunctionProps } from \"../Utils.js\";\nimport type { ReactNode } from \"react\";\nimport type { IModelConnection } from \"@itwin/core-frontend\";\nimport type { SchemaContext } from \"@itwin/ecschema-metadata\";\nimport type { PresentationHierarchyNode, SelectionStorage, useIModelTree, useSelectionHandler } from \"@itwin/presentation-hierarchies-react\";\nimport type { HighlightInfo } from \"../UseNodeHighlighting.js\";\n\n/** @beta */\nexport type TreeProps = Pick<FunctionProps<typeof useIModelTree>, \"getFilteredPaths\" | \"getHierarchyDefinition\"> &\n  Partial<Pick<FunctionProps<typeof useSelectionHandler>, \"selectionMode\">> & {\n    /** iModel connection that should be used to pull data from. */\n    imodel: IModelConnection;\n    /** Callback for getting `SchemaContext` for specific iModel. */\n    getSchemaContext: (imodel: IModelConnection) => SchemaContext;\n    /** Unique tree component name that will be used as unified selection change event source when selecting node. */\n    treeName: string;\n    /** Unified selection storage that should be used by tree to handle tree selection changes. */\n    selectionStorage: SelectionStorage;\n    /**\n     * An optional predicate to allow or prohibit selection of a node.\n     * When not supplied, all nodes are selectable.\n     */\n    selectionPredicate?: (node: PresentationHierarchyNode) => boolean;\n    /** Tree renderer that should be used to render tree data. */\n    treeRenderer: (\n      treeProps: Required<\n        Pick<\n          BaseTreeRendererProps,\n          | \"rootNodes\"\n          | \"expandNode\"\n          | \"getLabel\"\n          | \"onFilterClick\"\n          | \"selectNodes\"\n          | \"selectionMode\"\n          | \"isNodeSelected\"\n          | \"getHierarchyLevelDetails\"\n          | \"getLabel\"\n        >\n      >,\n    ) => ReactNode;\n    /** Custom iModel access that is stored outside tree component. If not provided it new iModel access will be created using `imodel` prop. */\n    imodelAccess?: FunctionProps<typeof useIModelTree>[\"imodelAccess\"];\n    /** Size limit that should be applied on each hierarchy level. Default to `1000`. */\n    hierarchyLevelSizeLimit?: number;\n    /** Component that should be renderer if there are no tree nodes. */\n    emptyTreeContent?: ReactNode;\n    /** Callback that this invoked when tree reloads. */\n    onReload?: () => void;\n    /** Options for highlighting node labels. */\n    highlight?: HighlightInfo;\n  };\n\n/**\n * Default tree component that manages tree state and renders using supplied `treeRenderer`.\n * @beta\n */\nexport function Tree({ getSchemaContext, hierarchyLevelSizeLimit, selectionStorage, imodelAccess: providedIModelAccess, ...props }: TreeProps) {\n  const defaultHierarchyLevelSizeLimit = hierarchyLevelSizeLimit ?? 1000;\n\n  const imodelAccess = useMemo(() => {\n    return providedIModelAccess ?? createIModelAccess({ getSchemaContext, imodel: props.imodel });\n  }, [providedIModelAccess, getSchemaContext, props.imodel]);\n\n  return (\n    <SchemaMetadataContextProvider imodel={props.imodel} schemaContextProvider={getSchemaContext}>\n      <TreeImpl {...props} selectionStorage={selectionStorage} imodelAccess={imodelAccess} defaultHierarchyLevelSizeLimit={defaultHierarchyLevelSizeLimit} />\n    </SchemaMetadataContextProvider>\n  );\n}\n\nfunction TreeImpl({\n  imodel,\n  imodelAccess,\n  treeName,\n  emptyTreeContent,\n  getFilteredPaths,\n  defaultHierarchyLevelSizeLimit,\n  getHierarchyDefinition,\n  selectionPredicate,\n  selectionMode,\n  onReload,\n  treeRenderer,\n  selectionStorage,\n  highlight,\n}: MarkRequired<Omit<TreeProps, \"getSchemaContext\">, \"imodelAccess\"> & { defaultHierarchyLevelSizeLimit: number }) {\n  const localizedStrings = useHierarchiesLocalization();\n  const { onFeatureUsed, onPerformanceMeasured } = useTelemetryContext();\n  const [imodelChanged] = useState(new BeEvent<() => void>());\n  const {\n    rootNodes,\n    getNode,\n    isLoading,\n    selectNodes: selectNodesAction,\n    setFormatter: _setFormatter,\n    expandNode,\n    ...treeProps\n  } = useIModelUnifiedSelectionTree({\n    imodelAccess,\n    imodelChanged,\n    getHierarchyDefinition,\n    getFilteredPaths,\n    sourceName: treeName,\n    localizedStrings,\n    selectionStorage,\n    onPerformanceMeasured: (action, duration) => {\n      if (action === \"reload\") {\n        onReload?.();\n      }\n      onPerformanceMeasured(action, duration);\n    },\n    onHierarchyLimitExceeded: () => onFeatureUsed({ featureId: \"hierarchy-level-size-limit-hit\", reportInteraction: false }),\n    onHierarchyLoadError: ({ type, error }) => {\n      // eslint-disable-next-line no-console\n      console.error(error);\n      onFeatureUsed({ featureId: `error-${type}`, reportInteraction: false });\n    },\n  });\n  useIModelChangeListener({ imodel, action: useCallback(() => imodelChanged.raiseEvent(), [imodelChanged]) });\n\n  const selectNodes = useSelectionPredicate({\n    action: useReportingAction({ action: selectNodesAction }),\n    predicate: selectionPredicate,\n    getNode,\n  });\n  const { filteringDialog, onFilterClick } = useHierarchyLevelFiltering({\n    imodel,\n    defaultHierarchyLevelSizeLimit,\n  });\n  const reportingExpandNode = useReportingAction({ action: expandNode });\n  const reportingOnFilterClicked = useReportingAction({ action: onFilterClick });\n  const { getLabel } = useNodeHighlighting({ rootNodes, highlight });\n\n  if (rootNodes === undefined) {\n    return <SkeletonTree />;\n  }\n\n  if (rootNodes.length === 0 && !isLoading) {\n    return <>{emptyTreeContent ? emptyTreeContent : <EmptyTreeContent />}</>;\n  }\n\n  const treeRendererProps: FunctionProps<TreeProps[\"treeRenderer\"]> = {\n    ...treeProps,\n    rootNodes,\n    selectionMode: selectionMode ?? \"single\",\n    selectNodes,\n    expandNode: reportingExpandNode,\n    onFilterClick: reportingOnFilterClicked,\n    getLabel,\n  };\n\n  return (\n    <div style={{ position: \"relative\", height: \"100%\", overflow: \"hidden\" }}>\n      <div className={\"tw-tree-renderer-container\"} id=\"tw-tree-renderer-container\">\n        {treeRenderer(treeRendererProps)}\n        {filteringDialog}\n      </div>\n      <Delayed show={isLoading}>\n        <ProgressOverlay />\n      </Delayed>\n    </div>\n  );\n}\n\nfunction useSelectionPredicate({\n  action,\n  predicate,\n  getNode,\n}: {\n  action: (...args: any[]) => void;\n  predicate?: (node: PresentationHierarchyNode) => boolean;\n  getNode: (nodeId: string) => PresentationHierarchyNode | undefined;\n}): ReturnType<typeof useIModelUnifiedSelectionTree>[\"selectNodes\"] {\n  return useCallback(\n    (nodeIds, changeType) =>\n      action(\n        nodeIds.filter((nodeId) => {\n          const node = getNode(nodeId);\n          return node && (!predicate || predicate(node));\n        }),\n        changeType,\n      ),\n    [action, getNode, predicate],\n  );\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/components/UseVisibilityButtonHandler.d.ts

@@ -8,6 +8,7 @@ interface UseVisibilityButtonHandlerProps {
 interface UseVisibilityButtonHandlerResult {
     onVisibilityButtonClick: TreeItemVisibilityButtonProps["onVisibilityButtonClick"];
 }
+/** @internal */
 export declare function useVisibilityButtonHandler({ rootNodes, isNodeSelected, onClick }: UseVisibilityButtonHandlerProps): UseVisibilityButtonHandlerResult;
 export {};
 //# sourceMappingURL=UseVisibilityButtonHandler.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/components/UseVisibilityButtonHandler.js

@@ -4,6 +4,7 @@
  *--------------------------------------------------------------------------------------------*/
 import { useCallback } from "react";
 import { isPresentationHierarchyNode } from "@itwin/presentation-hierarchies-react";
+/** @internal */
 export function useVisibilityButtonHandler({ rootNodes, isNodeSelected, onClick }) {
     const onVisibilityButtonClick = useCallback((clickedNode, state) => {
         if (!isNodeSelected?.(clickedNode.id)) {

package/lib/esm/tree-widget-react/components/trees/common/components/UseVisibilityButtonHandler.js.map

@@ -1 +1 @@
-{"version":3,"file":"UseVisibilityButtonHandler.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/UseVisibilityButtonHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,2BAA2B,EAAE,MAAM,uCAAuC,CAAC;AAepF,MAAM,UAAU,0BAA0B,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,OAAO,EAAmC;IAChH,MAAM,uBAAuB,GAAG,WAAW,CACzC,CAAC,WAAW,EAAE,KAAK,EAAE,EAAE;QACrB,IAAI,CAAC,cAAc,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE;YACrC,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YAC5B,OAAO;SACR;QACD,SAAS,IAAI,mBAAmB,CAAC,SAAS,EAAE,cAAc,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;IAC9F,CAAC,EACD,CAAC,SAAS,EAAE,cAAc,EAAE,OAAO,CAAC,CACrC,CAAC;IAEF,OAAO,EAAE,uBAAuB,EAAE,CAAC;AACrC,CAAC;AAED,SAAS,mBAAmB,CAC1B,KAAkC,EAClC,cAA2C,EAC3C,QAAmD;IAEnD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QACrB,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,EAAE;YACtC,OAAO;SACR;QACD,IAAI,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE;YAC3B,QAAQ,CAAC,IAAI,CAAC,CAAC;SAChB;QACD,IAAI,IAAI,CAAC,UAAU,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE;YACzD,mBAAmB,CAAC,IAAI,CAAC,QAAQ,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;SAC9D;IACH,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback } from \"react\";\nimport { isPresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\n\nimport type { PresentationHierarchyNode, PresentationTreeNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { TreeItemVisibilityButtonProps } from \"./TreeNodeVisibilityButton.js\";\n\ninterface UseVisibilityButtonHandlerProps {\n  rootNodes: PresentationTreeNode[] | undefined;\n  isNodeSelected?: (nodeId: string) => boolean;\n  onClick: TreeItemVisibilityButtonProps[\"onVisibilityButtonClick\"];\n}\n\ninterface UseVisibilityButtonHandlerResult {\n  onVisibilityButtonClick: TreeItemVisibilityButtonProps[\"onVisibilityButtonClick\"];\n}\n\nexport function useVisibilityButtonHandler({ rootNodes, isNodeSelected, onClick }: UseVisibilityButtonHandlerProps): UseVisibilityButtonHandlerResult {\n  const onVisibilityButtonClick = useCallback<TreeItemVisibilityButtonProps[\"onVisibilityButtonClick\"]>(\n    (clickedNode, state) => {\n      if (!isNodeSelected?.(clickedNode.id)) {\n        onClick(clickedNode, state);\n        return;\n      }\n      rootNodes && forEachSelectedNode(rootNodes, isNodeSelected, (node) => onClick(node, state));\n    },\n    [rootNodes, isNodeSelected, onClick],\n  );\n\n  return { onVisibilityButtonClick };\n}\n\nfunction forEachSelectedNode(\n  nodes: Array<PresentationTreeNode>,\n  isNodeSelected: (nodeId: string) => boolean,\n  callback: (node: PresentationHierarchyNode) => void,\n) {\n  nodes.forEach((node) => {\n    if (!isPresentationHierarchyNode(node)) {\n      return;\n    }\n    if (isNodeSelected(node.id)) {\n      callback(node);\n    }\n    if (node.isExpanded && typeof node.children !== \"boolean\") {\n      forEachSelectedNode(node.children, isNodeSelected, callback);\n    }\n  });\n}\n"]}
+{"version":3,"file":"UseVisibilityButtonHandler.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/UseVisibilityButtonHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,2BAA2B,EAAE,MAAM,uCAAuC,CAAC;AAepF,gBAAgB;AAChB,MAAM,UAAU,0BAA0B,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,OAAO,EAAmC;IAChH,MAAM,uBAAuB,GAAG,WAAW,CACzC,CAAC,WAAW,EAAE,KAAK,EAAE,EAAE;QACrB,IAAI,CAAC,cAAc,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE;YACrC,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YAC5B,OAAO;SACR;QACD,SAAS,IAAI,mBAAmB,CAAC,SAAS,EAAE,cAAc,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;IAC9F,CAAC,EACD,CAAC,SAAS,EAAE,cAAc,EAAE,OAAO,CAAC,CACrC,CAAC;IAEF,OAAO,EAAE,uBAAuB,EAAE,CAAC;AACrC,CAAC;AAED,SAAS,mBAAmB,CAC1B,KAAkC,EAClC,cAA2C,EAC3C,QAAmD;IAEnD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QACrB,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,EAAE;YACtC,OAAO;SACR;QACD,IAAI,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE;YAC3B,QAAQ,CAAC,IAAI,CAAC,CAAC;SAChB;QACD,IAAI,IAAI,CAAC,UAAU,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE;YACzD,mBAAmB,CAAC,IAAI,CAAC,QAAQ,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;SAC9D;IACH,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback } from \"react\";\nimport { isPresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\n\nimport type { PresentationHierarchyNode, PresentationTreeNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { TreeItemVisibilityButtonProps } from \"./TreeNodeVisibilityButton.js\";\n\ninterface UseVisibilityButtonHandlerProps {\n  rootNodes: PresentationTreeNode[] | undefined;\n  isNodeSelected?: (nodeId: string) => boolean;\n  onClick: TreeItemVisibilityButtonProps[\"onVisibilityButtonClick\"];\n}\n\ninterface UseVisibilityButtonHandlerResult {\n  onVisibilityButtonClick: TreeItemVisibilityButtonProps[\"onVisibilityButtonClick\"];\n}\n\n/** @internal */\nexport function useVisibilityButtonHandler({ rootNodes, isNodeSelected, onClick }: UseVisibilityButtonHandlerProps): UseVisibilityButtonHandlerResult {\n  const onVisibilityButtonClick = useCallback<TreeItemVisibilityButtonProps[\"onVisibilityButtonClick\"]>(\n    (clickedNode, state) => {\n      if (!isNodeSelected?.(clickedNode.id)) {\n        onClick(clickedNode, state);\n        return;\n      }\n      rootNodes && forEachSelectedNode(rootNodes, isNodeSelected, (node) => onClick(node, state));\n    },\n    [rootNodes, isNodeSelected, onClick],\n  );\n\n  return { onVisibilityButtonClick };\n}\n\nfunction forEachSelectedNode(\n  nodes: Array<PresentationTreeNode>,\n  isNodeSelected: (nodeId: string) => boolean,\n  callback: (node: PresentationHierarchyNode) => void,\n) {\n  nodes.forEach((node) => {\n    if (!isPresentationHierarchyNode(node)) {\n      return;\n    }\n    if (isNodeSelected(node.id)) {\n      callback(node);\n    }\n    if (node.isExpanded && typeof node.children !== \"boolean\") {\n      forEachSelectedNode(node.children, isNodeSelected, callback);\n    }\n  });\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/components/VisibilityTree.js

@@ -4,8 +4,8 @@ import { jsx as _jsx } from "react/jsx-runtime";
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
 import { useCallback, useMemo } from "react";
+import { createIModelAccess } from "../internal/Utils.js";
 import { useHierarchyVisibility } from "../UseHierarchyVisibility.js";
-import { createIModelAccess } from "../Utils.js";
 import { Tree } from "./Tree.js";
 /**
  * Tree component that can control visibility of instances represented by tree nodes.

package/lib/esm/tree-widget-react/components/trees/common/components/VisibilityTree.js.map

@@ -1 +1 @@
-{"version":3,"file":"VisibilityTree.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/VisibilityTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAC7C,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAmBjC;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,EAAE,wBAAwB,EAAE,YAAY,EAAE,GAAG,KAAK,EAAuB;IACtG,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,GAAG,KAAK,CAAC;IAC3C,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC,CAAC;IACjH,MAAM,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,cAAc,EAAE,GAAG,sBAAsB,CAAC;QACnG,wBAAwB,EAAE,WAAW,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,wBAAwB,EAAE,YAAY,CAAC,CAAC;KAClI,CAAC,CAAC;IAEH,OAAO,CACL,KAAC,IAAI,OACC,KAAK,EACT,QAAQ,EAAE,cAAc,EACxB,YAAY,EAAE,YAAY,EAC1B,YAAY,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,GAAG,SAAS,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,CAAC,GAC9G,CACH,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useMemo } from \"react\";\nimport { useHierarchyVisibility } from \"../UseHierarchyVisibility.js\";\nimport { createIModelAccess } from \"../Utils.js\";\nimport { Tree } from \"./Tree.js\";\n\nimport type { FunctionProps } from \"../Utils.js\";\nimport type { TreeProps } from \"./Tree.js\";\nimport type { ReactNode } from \"react\";\nimport type { VisibilityTreeRendererProps } from \"./VisibilityTreeRenderer.js\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { HierarchyVisibilityHandler } from \"../UseHierarchyVisibility.js\";\n\n/** @beta */\nexport type VisibilityTreeProps = Omit<TreeProps, \"treeRenderer\" | \"imodelAccess\"> & {\n  /** Callback for creating visibility handler used to control visibility of instances represented by tree nodes. */\n  visibilityHandlerFactory: (props: { imodelAccess: ECClassHierarchyInspector }) => HierarchyVisibilityHandler;\n  /** Tree renderer that should be used to render tree data. */\n  treeRenderer: (\n    treeProps: FunctionProps<TreeProps[\"treeRenderer\"]> & Pick<VisibilityTreeRendererProps, \"getVisibilityButtonState\" | \"onVisibilityButtonClick\">,\n  ) => ReactNode;\n};\n\n/**\n * Tree component that can control visibility of instances represented by tree nodes.\n * @beta\n */\nexport function VisibilityTree({ visibilityHandlerFactory, treeRenderer, ...props }: VisibilityTreeProps) {\n  const { imodel, getSchemaContext } = props;\n  const imodelAccess = useMemo(() => createIModelAccess({ imodel, getSchemaContext }), [imodel, getSchemaContext]);\n  const { getVisibilityButtonState, onVisibilityButtonClick, triggerRefresh } = useHierarchyVisibility({\n    visibilityHandlerFactory: useCallback(() => visibilityHandlerFactory({ imodelAccess }), [visibilityHandlerFactory, imodelAccess]),\n  });\n\n  return (\n    <Tree\n      {...props}\n      onReload={triggerRefresh}\n      imodelAccess={imodelAccess}\n      treeRenderer={(treeProps) => treeRenderer({ ...treeProps, getVisibilityButtonState, onVisibilityButtonClick })}\n    />\n  );\n}\n"]}
+{"version":3,"file":"VisibilityTree.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/VisibilityTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAC7C,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAmBjC;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,EAAE,wBAAwB,EAAE,YAAY,EAAE,GAAG,KAAK,EAAuB;IACtG,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,GAAG,KAAK,CAAC;IAC3C,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC,CAAC;IACjH,MAAM,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,cAAc,EAAE,GAAG,sBAAsB,CAAC;QACnG,wBAAwB,EAAE,WAAW,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,wBAAwB,EAAE,YAAY,CAAC,CAAC;KAClI,CAAC,CAAC;IAEH,OAAO,CACL,KAAC,IAAI,OACC,KAAK,EACT,QAAQ,EAAE,cAAc,EACxB,YAAY,EAAE,YAAY,EAC1B,YAAY,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,GAAG,SAAS,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,CAAC,GAC9G,CACH,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useMemo } from \"react\";\nimport { createIModelAccess } from \"../internal/Utils.js\";\nimport { useHierarchyVisibility } from \"../UseHierarchyVisibility.js\";\nimport { Tree } from \"./Tree.js\";\n\nimport type { FunctionProps } from \"../Utils.js\";\nimport type { TreeProps } from \"./Tree.js\";\nimport type { ReactNode } from \"react\";\nimport type { VisibilityTreeRendererProps } from \"./VisibilityTreeRenderer.js\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { HierarchyVisibilityHandler } from \"../UseHierarchyVisibility.js\";\n\n/** @beta */\nexport type VisibilityTreeProps = Omit<TreeProps, \"treeRenderer\" | \"imodelAccess\"> & {\n  /** Callback for creating visibility handler used to control visibility of instances represented by tree nodes. */\n  visibilityHandlerFactory: (props: { imodelAccess: ECClassHierarchyInspector }) => HierarchyVisibilityHandler;\n  /** Tree renderer that should be used to render tree data. */\n  treeRenderer: (\n    treeProps: FunctionProps<TreeProps[\"treeRenderer\"]> & Pick<VisibilityTreeRendererProps, \"getVisibilityButtonState\" | \"onVisibilityButtonClick\">,\n  ) => ReactNode;\n};\n\n/**\n * Tree component that can control visibility of instances represented by tree nodes.\n * @beta\n */\nexport function VisibilityTree({ visibilityHandlerFactory, treeRenderer, ...props }: VisibilityTreeProps) {\n  const { imodel, getSchemaContext } = props;\n  const imodelAccess = useMemo(() => createIModelAccess({ imodel, getSchemaContext }), [imodel, getSchemaContext]);\n  const { getVisibilityButtonState, onVisibilityButtonClick, triggerRefresh } = useHierarchyVisibility({\n    visibilityHandlerFactory: useCallback(() => visibilityHandlerFactory({ imodelAccess }), [visibilityHandlerFactory, imodelAccess]),\n  });\n\n  return (\n    <Tree\n      {...props}\n      onReload={triggerRefresh}\n      imodelAccess={imodelAccess}\n      treeRenderer={(treeProps) => treeRenderer({ ...treeProps, getVisibilityButtonState, onVisibilityButtonClick })}\n    />\n  );\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/components/VisibilityTreeRenderer.js.map

@@ -1 +1 @@
-{"version":3,"file":"VisibilityTreeRenderer.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/VisibilityTreeRenderer.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG,uCAAuC;AAEvC,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAChC,OAAO,EAAE,eAAe,EAAE,MAAM,uCAAuC,CAAC;AACxE,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAO7E;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CAAC,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,KAAK,EAA+B;IACnJ,MAAM,EAAE,uBAAuB,EAAE,GAAG,0BAA0B,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,OAAO,EAAE,CAAC,CAAC;IAC9I,MAAM,qBAAqB,GAAkC,OAAO,CAClE,GAAG,EAAE,CAAC,CAAC;QACL,OAAO,EAAE,SAAS;QAClB,wBAAwB;QACxB,uBAAuB;KACxB,CAAC,EACF,CAAC,wBAAwB,EAAE,uBAAuB,CAAC,CACpD,CAAC;IAEF,MAAM,YAAY,GAAG,eAAe,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,aAAa,EAAE,wBAAwB,EAAE,KAAK,CAAC,wBAAwB,EAAE,CAAC,CAAC;IAElI,OAAO,CACL,KAAC,gBAAgB,OACX,KAAK,EACT,OAAO,EAAE,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GACvI,CACH,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n/* eslint-disable @itwin/no-internal */\n\nimport { useMemo } from \"react\";\nimport { useFilterAction } from \"@itwin/presentation-hierarchies-react\";\nimport { BaseTreeRenderer } from \"./BaseTreeRenderer.js\";\nimport { createVisibilityAction } from \"./TreeNodeVisibilityButton.js\";\nimport { useVisibilityButtonHandler } from \"./UseVisibilityButtonHandler.js\";\n\nimport type { BaseTreeRendererProps } from \"./BaseTreeRenderer.js\";\nimport type { TreeItemVisibilityButtonProps } from \"./TreeNodeVisibilityButton.js\";\n/** @beta */\nexport type VisibilityTreeRendererProps = BaseTreeRendererProps & TreeItemVisibilityButtonProps;\n\n/**\n * Tree renderer that renders tree nodes with eye checkboxes for controlling visibility of instances represented by tree nodes.\n * @beta\n */\nexport function VisibilityTreeRenderer({ getVisibilityButtonState, onVisibilityButtonClick: onClick, actions, ...props }: VisibilityTreeRendererProps) {\n  const { onVisibilityButtonClick } = useVisibilityButtonHandler({ rootNodes: props.rootNodes, isNodeSelected: props.isNodeSelected, onClick });\n  const visibilityButtonProps: TreeItemVisibilityButtonProps = useMemo(\n    () => ({\n      variant: \"eyeball\",\n      getVisibilityButtonState,\n      onVisibilityButtonClick,\n    }),\n    [getVisibilityButtonState, onVisibilityButtonClick],\n  );\n\n  const filterAction = useFilterAction({ onFilter: props.onFilterClick, getHierarchyLevelDetails: props.getHierarchyLevelDetails });\n\n  return (\n    <BaseTreeRenderer\n      {...props}\n      actions={[...(visibilityButtonProps ? [createVisibilityAction(visibilityButtonProps)] : []), filterAction, ...(actions ? actions : [])]}\n    />\n  );\n}\n"]}
+{"version":3,"file":"VisibilityTreeRenderer.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/VisibilityTreeRenderer.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG,uCAAuC;AAEvC,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAChC,OAAO,EAAE,eAAe,EAAE,MAAM,uCAAuC,CAAC;AACxE,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAQ7E;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CAAC,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,KAAK,EAA+B;IACnJ,MAAM,EAAE,uBAAuB,EAAE,GAAG,0BAA0B,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,OAAO,EAAE,CAAC,CAAC;IAC9I,MAAM,qBAAqB,GAAkC,OAAO,CAClE,GAAG,EAAE,CAAC,CAAC;QACL,OAAO,EAAE,SAAS;QAClB,wBAAwB;QACxB,uBAAuB;KACxB,CAAC,EACF,CAAC,wBAAwB,EAAE,uBAAuB,CAAC,CACpD,CAAC;IAEF,MAAM,YAAY,GAAG,eAAe,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,aAAa,EAAE,wBAAwB,EAAE,KAAK,CAAC,wBAAwB,EAAE,CAAC,CAAC;IAElI,OAAO,CACL,KAAC,gBAAgB,OACX,KAAK,EACT,OAAO,EAAE,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GACvI,CACH,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n/* eslint-disable @itwin/no-internal */\n\nimport { useMemo } from \"react\";\nimport { useFilterAction } from \"@itwin/presentation-hierarchies-react\";\nimport { BaseTreeRenderer } from \"./BaseTreeRenderer.js\";\nimport { createVisibilityAction } from \"./TreeNodeVisibilityButton.js\";\nimport { useVisibilityButtonHandler } from \"./UseVisibilityButtonHandler.js\";\n\nimport type { BaseTreeRendererProps } from \"./BaseTreeRenderer.js\";\nimport type { TreeItemVisibilityButtonProps } from \"./TreeNodeVisibilityButton.js\";\n\n/** @beta */\nexport type VisibilityTreeRendererProps = BaseTreeRendererProps & TreeItemVisibilityButtonProps;\n\n/**\n * Tree renderer that renders tree nodes with eye checkboxes for controlling visibility of instances represented by tree nodes.\n * @beta\n */\nexport function VisibilityTreeRenderer({ getVisibilityButtonState, onVisibilityButtonClick: onClick, actions, ...props }: VisibilityTreeRendererProps) {\n  const { onVisibilityButtonClick } = useVisibilityButtonHandler({ rootNodes: props.rootNodes, isNodeSelected: props.isNodeSelected, onClick });\n  const visibilityButtonProps: TreeItemVisibilityButtonProps = useMemo(\n    () => ({\n      variant: \"eyeball\",\n      getVisibilityButtonState,\n      onVisibilityButtonClick,\n    }),\n    [getVisibilityButtonState, onVisibilityButtonClick],\n  );\n\n  const filterAction = useFilterAction({ onFilter: props.onFilterClick, getHierarchyLevelDetails: props.getHierarchyLevelDetails });\n\n  return (\n    <BaseTreeRenderer\n      {...props}\n      actions={[...(visibilityButtonProps ? [createVisibilityAction(visibilityButtonProps)] : []), filterAction, ...(actions ? actions : [])]}\n    />\n  );\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/internal/AlwaysAndNeverDrawnElementInfo.d.ts

@@ -0,0 +1,44 @@
+/// <reference types="node" resolution-mode="require"/>
+import type { Observable } from "rxjs";
+import type { Id64Array, Id64String } from "@itwin/core-bentley";
+import type { Viewport } from "@itwin/core-frontend";
+import type { ElementId } from "./Types.js";
+/** @internal */
+export interface ModelAlwaysOrNeverDrawnElementsQueryProps {
+    modelId: Id64String;
+}
+/** @internal */
+export interface CategoryAlwaysOrNeverDrawnElementsQueryProps {
+    modelId?: Id64String;
+    categoryIds: Id64Array;
+}
+/** @internal */
+export type AlwaysOrNeverDrawnElementsQueryProps = ModelAlwaysOrNeverDrawnElementsQueryProps | CategoryAlwaysOrNeverDrawnElementsQueryProps;
+/** @internal */
+export declare const SET_CHANGE_DEBOUNCE_TIME = 20;
+/** @internal */
+export declare class AlwaysAndNeverDrawnElementInfo implements Disposable {
+    private readonly _viewport;
+    private readonly _elementClassName?;
+    private _subscriptions;
+    private _alwaysDrawn;
+    private _neverDrawn;
+    private _disposeSubject;
+    private _suppressors;
+    private _suppress;
+    private _forceUpdate;
+    constructor(_viewport: Viewport, _elementClassName?: string | undefined);
+    suppressChangeEvents(): void;
+    resumeChangeEvents(): void;
+    getElements(props: {
+        setType: "always" | "never";
+    } & AlwaysOrNeverDrawnElementsQueryProps): Observable<Set<ElementId>>;
+    private createCacheEntryObservable;
+    [Symbol.dispose](): void;
+    private queryAlwaysOrNeverDrawnElementInfo;
+    private queryElementInfo;
+    getAlwaysDrawnElements(props: AlwaysOrNeverDrawnElementsQueryProps): Observable<Set<string>>;
+    getNeverDrawnElements(props: AlwaysOrNeverDrawnElementsQueryProps): Observable<Set<string>>;
+    clearAlwaysAndNeverDrawnElements(props: AlwaysOrNeverDrawnElementsQueryProps): Observable<void>;
+}
+//# sourceMappingURL=AlwaysAndNeverDrawnElementInfo.d.ts.map

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

@@ -2,13 +2,16 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { BehaviorSubject, debounceTime, EMPTY, filter, first, from, fromEventPattern, map, merge, reduce, scan, share, shareReplay, startWith, Subject, switchMap, take, takeUntil, tap, } from "rxjs";
+import { BehaviorSubject, debounceTime, EMPTY, filter, first, forkJoin, from, fromEventPattern, map, merge, reduce, scan, share, shareReplay, startWith, Subject, switchMap, take, takeUntil, tap, } from "rxjs";
 import { createECSqlQueryExecutor } from "@itwin/presentation-core-interop";
-import { pushToMap } from "../../common/Utils.js";
+import { getClassesByView, pushToMap, setDifference } from "./Utils.js";
+/** @internal */
 export const SET_CHANGE_DEBOUNCE_TIME = 20;
+/** @internal */
 export class AlwaysAndNeverDrawnElementInfo {
-    constructor(_viewport) {
+    constructor(_viewport, _elementClassName) {
         this._viewport = _viewport;
+        this._elementClassName = _elementClassName;
         this._disposeSubject = new Subject();
         this._suppress = new Subject();
         this._forceUpdate = new Subject();
@@ -37,14 +40,40 @@ export class AlwaysAndNeverDrawnElementInfo {
     resumeChangeEvents() {
         this._suppress.next(false);
     }
-    getElements({ setType, modelId, categoryId }) {
-        const cache = setType === "always" ? this._alwaysDrawn : this._neverDrawn;
-        const getElements = categoryId
+    getElements(props) {
+        const cache = props.setType === "always" ? this._alwaysDrawn : this._neverDrawn;
+        const getElements = "categoryIds" in props
             ? (entry) => {
-                return entry?.get(modelId)?.get(categoryId) ?? new Set();
+                const result = new Set();
+                if (props.modelId) {
+                    const categoryMap = entry?.get(props.modelId);
+                    if (!categoryMap) {
+                        return result;
+                    }
+                    for (const categoryId of props.categoryIds) {
+                        const elements = categoryMap.get(categoryId);
+                        if (elements) {
+                            elements.forEach((element) => {
+                                result.add(element);
+                            });
+                        }
+                    }
+                    return result;
+                }
+                for (const [, categoryMap] of entry ?? []) {
+                    for (const categoryId of props.categoryIds) {
+                        const elements = categoryMap.get(categoryId);
+                        if (elements) {
+                            elements.forEach((element) => {
+                                result.add(element);
+                            });
+                        }
+                    }
+                }
+                return result;
             }
             : (entry) => {
-                const modelEntry = entry?.get(modelId);
+                const modelEntry = entry?.get(props.modelId);
                 const elements = new Set();
                 for (const set of modelEntry?.values() ?? []) {
                     set.forEach((id) => elements.add(id));
@@ -80,7 +109,7 @@ export class AlwaysAndNeverDrawnElementInfo {
         first((x) => !!x));
         return obs;
     }
-    dispose() {
+    [Symbol.dispose]() {
         this._subscriptions.forEach((x) => x.unsubscribe());
         this._subscriptions = [];
         this._disposeSubject.next();
@@ -99,40 +128,61 @@ export class AlwaysAndNeverDrawnElementInfo {
     }
     queryElementInfo(elementIds, requestId) {
         const executor = createECSqlQueryExecutor(this._viewport.iModel);
+        const { elementClass } = this._elementClassName ? { elementClass: this._elementClassName } : getClassesByView(this._viewport.view.is2d() ? "2d" : "3d");
         const reader = executor.createQueryReader({
             ctes: [
                 `
-        ElementInfo(elementId, modelId, categoryId, parentId) AS (
-          SELECT
-            ECInstanceId elementId,
-            Model.Id modelId,
-            Category.Id categoryId,
-            Parent.Id parentId
-          FROM bis.GeometricElement3d
-          WHERE InVirtualSet(?, ECInstanceId)
+            ElementInfo(elementId, modelId, categoryId, parentId) AS (
+              SELECT
+                ECInstanceId elementId,
+                Model.Id modelId,
+                Category.Id categoryId,
+                Parent.Id parentId
+              FROM ${elementClass}
+              WHERE InVirtualSet(?, ECInstanceId)
 
-          UNION ALL
+              UNION ALL
 
-          SELECT
-            e.elementId,
-            e.modelId,
-            p.Category.Id categoryId,
-            p.Parent.Id parentId
-          FROM bis.GeometricElement3d p
-          JOIN ElementInfo e ON p.ECInstanceId = e.parentId
-        )
-        `,
+              SELECT
+                e.elementId,
+                e.modelId,
+                p.Category.Id categoryId,
+                p.Parent.Id parentId
+              FROM ${elementClass} p
+              JOIN ElementInfo e ON p.ECInstanceId = e.parentId
+            )
+          `,
             ],
             ecsql: `
-        SELECT elementId, modelId, categoryId
-        FROM ElementInfo
-        WHERE parentId IS NULL
-      `,
+          SELECT elementId, modelId, categoryId
+          FROM ElementInfo
+          WHERE parentId IS NULL
+        `,
             bindings: [{ type: "idset", value: elementIds }],
         }, {
             restartToken: `ModelsTreeVisibilityHandler/${requestId}`,
         });
         return from(reader).pipe(map((row) => ({ elementId: row.elementId, modelId: row.modelId, categoryId: row.categoryId })));
     }
+    getAlwaysDrawnElements(props) {
+        return this.getElements({ ...props, setType: "always" });
+    }
+    getNeverDrawnElements(props) {
+        return this.getElements({ ...props, setType: "never" });
+    }
+    clearAlwaysAndNeverDrawnElements(props) {
+        return forkJoin({
+            alwaysDrawn: this.getAlwaysDrawnElements(props),
+            neverDrawn: this.getNeverDrawnElements(props),
+        }).pipe(map(({ alwaysDrawn, neverDrawn }) => {
+            const viewport = this._viewport;
+            if (viewport.alwaysDrawn?.size && alwaysDrawn.size) {
+                viewport.setAlwaysDrawn(setDifference(viewport.alwaysDrawn, alwaysDrawn));
+            }
+            if (viewport.neverDrawn?.size && neverDrawn.size) {
+                viewport.setNeverDrawn(setDifference(viewport.neverDrawn, neverDrawn));
+            }
+        }));
+    }
 }
 //# sourceMappingURL=AlwaysAndNeverDrawnElementInfo.js.map

package/lib/esm/tree-widget-react/components/trees/common/internal/AlwaysAndNeverDrawnElementInfo.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AlwaysAndNeverDrawnElementInfo.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/AlwaysAndNeverDrawnElementInfo.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EACL,eAAe,EACf,YAAY,EACZ,KAAK,EACL,MAAM,EACN,KAAK,EACL,QAAQ,EACR,IAAI,EACJ,gBAAgB,EAChB,GAAG,EACH,KAAK,EACL,MAAM,EACN,IAAI,EACJ,KAAK,EACL,WAAW,EACX,SAAS,EACT,OAAO,EACP,SAAS,EACT,IAAI,EACJ,SAAS,EACT,GAAG,GACJ,MAAM,MAAM,CAAC;AACd,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,gBAAgB,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AA6BxE,gBAAgB;AAChB,MAAM,CAAC,MAAM,wBAAwB,GAAG,EAAE,CAAC;AAE3C,gBAAgB;AAChB,MAAM,OAAO,8BAA8B;IAUzC,YACmB,SAAmB,EACnB,iBAA0B;QAD1B,cAAS,GAAT,SAAS,CAAU;QACnB,sBAAiB,GAAjB,iBAAiB,CAAS;QARrC,oBAAe,GAAG,IAAI,OAAO,EAAQ,CAAC;QAGtC,cAAS,GAAG,IAAI,OAAO,EAAW,CAAC;QACnC,iBAAY,GAAG,IAAI,OAAO,EAAQ,CAAC;QAMzC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC;YAClD,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,oBAAoB;YAC1C,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW;YACxC,EAAE,EAAE,aAAa;SAClB,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,0BAA0B,CAAC;YACjD,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,mBAAmB;YACzC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU;YACvC,EAAE,EAAE,YAAY;SACjB,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CACrC,IAAI,CAAC,CAAC,GAAG,EAAE,QAAQ,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EACrD,SAAS,CAAC,CAAC,CAAC,EACZ,WAAW,CAAC,CAAC,CAAC,CACf,CAAC;QACF,IAAI,CAAC,cAAc,GAAG;YACpB,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;YAC7B,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE;YAC5B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC3E,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE;aACrC,CAAC;SACH,CAAC;IACJ,CAAC;IAEM,oBAAoB;QACzB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEM,kBAAkB;QACvB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IAEM,WAAW,CAAC,KAA6E;QAC9F,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;QAChF,MAAM,WAAW,GACf,aAAa,IAAI,KAAK;YACpB,CAAC,CAAC,CAAC,KAA6B,EAAE,EAAE;gBAChC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAa,CAAC;gBACpC,IAAI,KAAK,CAAC,OAAO,EAAE;oBACjB,MAAM,WAAW,GAAG,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBAC9C,IAAI,CAAC,WAAW,EAAE;wBAChB,OAAO,MAAM,CAAC;qBACf;oBACD,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,WAAW,EAAE;wBAC1C,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;wBAC7C,IAAI,QAAQ,EAAE;4BACZ,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gCAC3B,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;4BACtB,CAAC,CAAC,CAAC;yBACJ;qBACF;oBACD,OAAO,MAAM,CAAC;iBACf;gBACD,KAAK,MAAM,CAAC,EAAE,WAAW,CAAC,IAAI,KAAK,IAAI,EAAE,EAAE;oBACzC,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,WAAW,EAAE;wBAC1C,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;wBAC7C,IAAI,QAAQ,EAAE;4BACZ,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gCAC3B,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;4BACtB,CAAC,CAAC,CAAC;yBACJ;qBACF;iBACF;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;YACH,CAAC,CAAC,CAAC,KAA6B,EAAE,EAAE;gBAChC,MAAM,UAAU,GAAG,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC7C,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAa,CAAC;gBACtC,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE;oBAC5C,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;iBACvC;gBACD,OAAO,QAAQ,CAAC;YAClB,CAAC,CAAC;QAER,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;IACtC,CAAC;IAEO,0BAA0B,CAAC,KAAuF;QACxH,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QAC1B,MAAM,aAAa,GAAG,IAAI,eAAe,CAAyB,SAAS,CAAC,CAAC;QAE7E,MAAM,GAAG,GAAG,KAAK,CACf,gBAAgB,CACd,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,EACvC,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAC3C,EACD,IAAI,CAAC,YAAY,CAClB,CAAC,IAAI;QACJ,4CAA4C;QAC5C,SAAS,CAAC,SAAS,CAAC;QACpB,oDAAoD;QACpD,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC;QAC/B,0DAA0D;QAC1D,iFAAiF;QACjF,6EAA6E;QAC7E,GAAG,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,6CAA6C;QAC7C,SAAS,CAAC,GAAG,EAAE,CACb,IAAI,CAAC,YAAY,CAAC,IAAI,CACpB,MAAM,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,KAAK,CAAC,CAAC,EAC1C,IAAI,CAAC,CAAC,CAAC,CACR,CACF,EACD,YAAY,CAAC,wBAAwB,CAAC;QACtC,iDAAiD;QACjD,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC;QAC/B,iFAAiF;QACjF,SAAS,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kCAAkC,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;QAClF,2EAA2E;QAC3E,KAAK,CAAC;YACJ,SAAS,EAAE,GAAG,EAAE,CAAC,aAAa;YAC9B,mBAAmB,EAAE,KAAK;SAC3B,CAAC;QACF,sCAAsC;QACtC,KAAK,CAAC,CAAC,CAAC,EAAmB,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CACnC,CAAC;QACF,OAAO,GAAG,CAAC;IACb,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;IAC9B,CAAC;IAEO,kCAAkC,CAAC,GAA+B,EAAE,SAAiB;QAC3F,MAAM,WAAW,GAAG,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QACnF,OAAO,WAAW,CAAC,IAAI,CACrB,MAAM,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YACpB,IAAI,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACnC,IAAI,CAAC,KAAK,EAAE;gBACV,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;gBAClB,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;aAC/B;YACD,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,OAAO,KAAK,CAAC;QACf,CAAC,EAAE,IAAI,GAAG,EAA4C,CAAC,CACxD,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,UAAqB,EAAE,SAAiB;QAC/D,MAAM,QAAQ,GAAG,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACjE,MAAM,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACxJ,MAAM,MAAM,GAAG,QAAQ,CAAC,iBAAiB,CACvC;YACE,IAAI,EAAE;gBACJ;;;;;;;qBAOW,YAAY;;;;;;;;;;qBAUZ,YAAY;;;WAGtB;aACF;YACD,KAAK,EAAE;;;;SAIN;YACD,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;SACjD,EACD;YACE,YAAY,EAAE,+BAA+B,SAAS,EAAE;SACzD,CACF,CAAC;QAEF,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IAC3H,CAAC;IAEM,sBAAsB,CAAC,KAA2C;QACvE,OAAO,IAAI,CAAC,WAAW,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC3D,CAAC;IAEM,qBAAqB,CAAC,KAA2C;QACtE,OAAO,IAAI,CAAC,WAAW,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;IAC1D,CAAC;IAEM,gCAAgC,CAAC,KAA2C;QACjF,OAAO,QAAQ,CAAC;YACd,WAAW,EAAE,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC;YAC/C,UAAU,EAAE,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC;SAC9C,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE;YAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;YAChC,IAAI,QAAQ,CAAC,WAAW,EAAE,IAAI,IAAI,WAAW,CAAC,IAAI,EAAE;gBAClD,QAAQ,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC;aAC3E;YACD,IAAI,QAAQ,CAAC,UAAU,EAAE,IAAI,IAAI,UAAU,CAAC,IAAI,EAAE;gBAChD,QAAQ,CAAC,aAAa,CAAC,aAAa,CAAC,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC;aACxE;QACH,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport {\n  BehaviorSubject,\n  debounceTime,\n  EMPTY,\n  filter,\n  first,\n  forkJoin,\n  from,\n  fromEventPattern,\n  map,\n  merge,\n  reduce,\n  scan,\n  share,\n  shareReplay,\n  startWith,\n  Subject,\n  switchMap,\n  take,\n  takeUntil,\n  tap,\n} from \"rxjs\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { getClassesByView, pushToMap, setDifference } from \"./Utils.js\";\n\nimport type { Observable, Subscription } from \"rxjs\";\nimport type { BeEvent, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { CategoryId, ElementId, ModelId } from \"./Types.js\";\n\ninterface ElementInfo {\n  elementId: Id64String;\n  modelId: Id64String;\n  categoryId: Id64String;\n}\n\ntype CacheEntry = Map<ModelId, Map<CategoryId, Set<ElementId>>>;\n\n/** @internal */\nexport interface ModelAlwaysOrNeverDrawnElementsQueryProps {\n  modelId: Id64String;\n}\n\n/** @internal */\nexport interface CategoryAlwaysOrNeverDrawnElementsQueryProps {\n  modelId?: Id64String;\n  categoryIds: Id64Array;\n}\n\n/** @internal */\nexport type AlwaysOrNeverDrawnElementsQueryProps = ModelAlwaysOrNeverDrawnElementsQueryProps | CategoryAlwaysOrNeverDrawnElementsQueryProps;\n\n/** @internal */\nexport const SET_CHANGE_DEBOUNCE_TIME = 20;\n\n/** @internal */\nexport class AlwaysAndNeverDrawnElementInfo implements Disposable {\n  private _subscriptions: Subscription[];\n  private _alwaysDrawn: Observable<CacheEntry>;\n  private _neverDrawn: Observable<CacheEntry>;\n  private _disposeSubject = new Subject<void>();\n\n  private _suppressors: Observable<number>;\n  private _suppress = new Subject<boolean>();\n  private _forceUpdate = new Subject<void>();\n\n  constructor(\n    private readonly _viewport: Viewport,\n    private readonly _elementClassName?: string,\n  ) {\n    this._alwaysDrawn = this.createCacheEntryObservable({\n      event: this._viewport.onAlwaysDrawnChanged,\n      getSet: () => this._viewport.alwaysDrawn,\n      id: \"alwaysDrawn\",\n    });\n    this._neverDrawn = this.createCacheEntryObservable({\n      event: this._viewport.onNeverDrawnChanged,\n      getSet: () => this._viewport.neverDrawn,\n      id: \"neverDrawn\",\n    });\n    this._suppressors = this._suppress.pipe(\n      scan((acc, suppress) => acc + (suppress ? 1 : -1), 0),\n      startWith(0),\n      shareReplay(1),\n    );\n    this._subscriptions = [\n      this._alwaysDrawn.subscribe(),\n      this._neverDrawn.subscribe(),\n      this._suppressors.pipe(filter((suppressors) => suppressors === 0)).subscribe({\n        next: () => this._forceUpdate.next(),\n      }),\n    ];\n  }\n\n  public suppressChangeEvents() {\n    this._suppress.next(true);\n  }\n\n  public resumeChangeEvents() {\n    this._suppress.next(false);\n  }\n\n  public getElements(props: { setType: \"always\" | \"never\" } & AlwaysOrNeverDrawnElementsQueryProps): Observable<Set<ElementId>> {\n    const cache = props.setType === \"always\" ? this._alwaysDrawn : this._neverDrawn;\n    const getElements =\n      \"categoryIds\" in props\n        ? (entry: CacheEntry | undefined) => {\n            const result = new Set<ElementId>();\n            if (props.modelId) {\n              const categoryMap = entry?.get(props.modelId);\n              if (!categoryMap) {\n                return result;\n              }\n              for (const categoryId of props.categoryIds) {\n                const elements = categoryMap.get(categoryId);\n                if (elements) {\n                  elements.forEach((element) => {\n                    result.add(element);\n                  });\n                }\n              }\n              return result;\n            }\n            for (const [, categoryMap] of entry ?? []) {\n              for (const categoryId of props.categoryIds) {\n                const elements = categoryMap.get(categoryId);\n                if (elements) {\n                  elements.forEach((element) => {\n                    result.add(element);\n                  });\n                }\n              }\n            }\n            return result;\n          }\n        : (entry: CacheEntry | undefined) => {\n            const modelEntry = entry?.get(props.modelId);\n            const elements = new Set<ElementId>();\n            for (const set of modelEntry?.values() ?? []) {\n              set.forEach((id) => elements.add(id));\n            }\n            return elements;\n          };\n\n    return cache.pipe(map(getElements));\n  }\n\n  private createCacheEntryObservable(props: { event: BeEvent<() => void>; getSet(): Set<ElementId> | undefined; id: string }) {\n    const event = props.event;\n    const resultSubject = new BehaviorSubject<CacheEntry | undefined>(undefined);\n\n    const obs = merge(\n      fromEventPattern(\n        (handler) => event.addListener(handler),\n        (handler) => event.removeListener(handler),\n      ),\n      this._forceUpdate,\n    ).pipe(\n      // Fire the observable once at the beginning\n      startWith(undefined),\n      // Stop listening to events when dispose() is called\n      takeUntil(this._disposeSubject),\n      // Reset result subject as soon as a new event is emitted.\n      // This will make newly subscribed observers wait for the debounce period to pass\n      // instead of consuming the cached value which at this point becomes invalid.\n      tap(() => resultSubject.next(undefined)),\n      // Check if cache updates are not suppressed.\n      switchMap(() =>\n        this._suppressors.pipe(\n          filter((suppressors) => suppressors === 0),\n          take(1),\n        ),\n      ),\n      debounceTime(SET_CHANGE_DEBOUNCE_TIME),\n      // Cancel pending request if dispose() is called.\n      takeUntil(this._disposeSubject),\n      // If multiple requests are sent at once, preserve only the result of the newest.\n      switchMap(() => this.queryAlwaysOrNeverDrawnElementInfo(props.getSet(), props.id)),\n      // Share the result by using a subject which always emits the saved result.\n      share({\n        connector: () => resultSubject,\n        resetOnRefCountZero: false,\n      }),\n      // Wait until the result is available.\n      first((x): x is CacheEntry => !!x),\n    );\n    return obs;\n  }\n\n  public [Symbol.dispose](): void {\n    this._subscriptions.forEach((x) => x.unsubscribe());\n    this._subscriptions = [];\n    this._disposeSubject.next();\n  }\n\n  private queryAlwaysOrNeverDrawnElementInfo(set: Set<ElementId> | undefined, requestId: string): Observable<CacheEntry> {\n    const elementInfo = set?.size ? this.queryElementInfo([...set], requestId) : EMPTY;\n    return elementInfo.pipe(\n      reduce((state, val) => {\n        let entry = state.get(val.modelId);\n        if (!entry) {\n          entry = new Map();\n          state.set(val.modelId, entry);\n        }\n        pushToMap(entry, val.categoryId, val.elementId);\n        return state;\n      }, new Map<ModelId, Map<CategoryId, Set<ElementId>>>()),\n    );\n  }\n\n  private queryElementInfo(elementIds: Id64Array, requestId: string): Observable<ElementInfo> {\n    const executor = createECSqlQueryExecutor(this._viewport.iModel);\n    const { elementClass } = this._elementClassName ? { elementClass: this._elementClassName } : getClassesByView(this._viewport.view.is2d() ? \"2d\" : \"3d\");\n    const reader = executor.createQueryReader(\n      {\n        ctes: [\n          `\n            ElementInfo(elementId, modelId, categoryId, parentId) AS (\n              SELECT\n                ECInstanceId elementId,\n                Model.Id modelId,\n                Category.Id categoryId,\n                Parent.Id parentId\n              FROM ${elementClass}\n              WHERE InVirtualSet(?, ECInstanceId)\n\n              UNION ALL\n\n              SELECT\n                e.elementId,\n                e.modelId,\n                p.Category.Id categoryId,\n                p.Parent.Id parentId\n              FROM ${elementClass} p\n              JOIN ElementInfo e ON p.ECInstanceId = e.parentId\n            )\n          `,\n        ],\n        ecsql: `\n          SELECT elementId, modelId, categoryId\n          FROM ElementInfo\n          WHERE parentId IS NULL\n        `,\n        bindings: [{ type: \"idset\", value: elementIds }],\n      },\n      {\n        restartToken: `ModelsTreeVisibilityHandler/${requestId}`,\n      },\n    );\n\n    return from(reader).pipe(map((row) => ({ elementId: row.elementId, modelId: row.modelId, categoryId: row.categoryId })));\n  }\n\n  public getAlwaysDrawnElements(props: AlwaysOrNeverDrawnElementsQueryProps) {\n    return this.getElements({ ...props, setType: \"always\" });\n  }\n\n  public getNeverDrawnElements(props: AlwaysOrNeverDrawnElementsQueryProps) {\n    return this.getElements({ ...props, setType: \"never\" });\n  }\n\n  public clearAlwaysAndNeverDrawnElements(props: AlwaysOrNeverDrawnElementsQueryProps) {\n    return forkJoin({\n      alwaysDrawn: this.getAlwaysDrawnElements(props),\n      neverDrawn: this.getNeverDrawnElements(props),\n    }).pipe(\n      map(({ alwaysDrawn, neverDrawn }) => {\n        const viewport = this._viewport;\n        if (viewport.alwaysDrawn?.size && alwaysDrawn.size) {\n          viewport.setAlwaysDrawn(setDifference(viewport.alwaysDrawn, alwaysDrawn));\n        }\n        if (viewport.neverDrawn?.size && neverDrawn.size) {\n          viewport.setNeverDrawn(setDifference(viewport.neverDrawn, neverDrawn));\n        }\n      }),\n    );\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/internal/ClassNameDefinitions.d.ts

@@ -0,0 +1,33 @@
+/** @internal */
+export declare const CATEGORY_CLASS_NAME = "BisCore.Category";
+/** @internal */
+export declare const DEFINITION_CONTAINER_CLASS_NAME = "BisCore.DefinitionContainer";
+/** @internal */
+export declare const DEFINITION_ELEMENT_CLASS_NAME = "BisCore.DefinitionElement";
+/** @internal */
+export declare const DEFINITION_MODEL_CLASS_NAME = "BisCore.DefinitionModel";
+/** @internal */
+export declare const DRAWING_CATEGORY_CLASS_NAME = "BisCore.DrawingCategory";
+/** @internal */
+export declare const ELEMENT_CLASS_NAME = "BisCore.Element";
+/** @internal */
+export declare const GEOMETRIC_ELEMENT_2D_CLASS_NAME = "BisCore.GeometricElement2d";
+/** @internal */
+export declare const GEOMETRIC_ELEMENT_3D_CLASS_NAME = "BisCore.GeometricElement3d";
+/** @internal */
+export declare const GEOMETRIC_MODEL_2D_CLASS_NAME = "BisCore.GeometricModel2d";
+/** @internal */
+export declare const GEOMETRIC_MODEL_3D_CLASS_NAME = "BisCore.GeometricModel3d";
+/** @internal */
+export declare const INFORMATION_PARTITION_ELEMENT_CLASS_NAME = "BisCore.InformationPartitionElement";
+/** @internal */
+export declare const MODEL_CLASS_NAME = "BisCore.Model";
+/** @internal */
+export declare const SPATIAL_CATEGORY_CLASS_NAME = "BisCore.SpatialCategory";
+/** @internal */
+export declare const SUBJECT_CLASS_NAME = "BisCore.Subject";
+/** @internal */
+export declare const SUB_CATEGORY_CLASS_NAME = "BisCore.SubCategory";
+/** @internal */
+export declare const SUB_MODELED_ELEMENT_CLASS_NAME = "BisCore.ISubModeledElement";
+//# sourceMappingURL=ClassNameDefinitions.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/internal/ClassNameDefinitions.js

@@ -0,0 +1,37 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+/** @internal */
+export const CATEGORY_CLASS_NAME = "BisCore.Category";
+/** @internal */
+export const DEFINITION_CONTAINER_CLASS_NAME = "BisCore.DefinitionContainer";
+/** @internal */
+export const DEFINITION_ELEMENT_CLASS_NAME = "BisCore.DefinitionElement";
+/** @internal */
+export const DEFINITION_MODEL_CLASS_NAME = "BisCore.DefinitionModel";
+/** @internal */
+export const DRAWING_CATEGORY_CLASS_NAME = "BisCore.DrawingCategory";
+/** @internal */
+export const ELEMENT_CLASS_NAME = "BisCore.Element";
+/** @internal */
+export const GEOMETRIC_ELEMENT_2D_CLASS_NAME = "BisCore.GeometricElement2d";
+/** @internal */
+export const GEOMETRIC_ELEMENT_3D_CLASS_NAME = "BisCore.GeometricElement3d";
+/** @internal */
+export const GEOMETRIC_MODEL_2D_CLASS_NAME = "BisCore.GeometricModel2d";
+/** @internal */
+export const GEOMETRIC_MODEL_3D_CLASS_NAME = "BisCore.GeometricModel3d";
+/** @internal */
+export const INFORMATION_PARTITION_ELEMENT_CLASS_NAME = "BisCore.InformationPartitionElement";
+/** @internal */
+export const MODEL_CLASS_NAME = "BisCore.Model";
+/** @internal */
+export const SPATIAL_CATEGORY_CLASS_NAME = "BisCore.SpatialCategory";
+/** @internal */
+export const SUBJECT_CLASS_NAME = "BisCore.Subject";
+/** @internal */
+export const SUB_CATEGORY_CLASS_NAME = "BisCore.SubCategory";
+/** @internal */
+export const SUB_MODELED_ELEMENT_CLASS_NAME = "BisCore.ISubModeledElement";
+//# sourceMappingURL=ClassNameDefinitions.js.map

package/lib/esm/tree-widget-react/components/trees/common/internal/ClassNameDefinitions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ClassNameDefinitions.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/ClassNameDefinitions.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,gBAAgB;AAChB,MAAM,CAAC,MAAM,mBAAmB,GAAG,kBAAkB,CAAC;AAEtD,gBAAgB;AAChB,MAAM,CAAC,MAAM,+BAA+B,GAAG,6BAA6B,CAAC;AAE7E,gBAAgB;AAChB,MAAM,CAAC,MAAM,6BAA6B,GAAG,2BAA2B,CAAC;AAEzE,gBAAgB;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAAG,yBAAyB,CAAC;AAErE,gBAAgB;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAAG,yBAAyB,CAAC;AAErE,gBAAgB;AAChB,MAAM,CAAC,MAAM,kBAAkB,GAAG,iBAAiB,CAAC;AAEpD,gBAAgB;AAChB,MAAM,CAAC,MAAM,+BAA+B,GAAG,4BAA4B,CAAC;AAE5E,gBAAgB;AAChB,MAAM,CAAC,MAAM,+BAA+B,GAAG,4BAA4B,CAAC;AAE5E,gBAAgB;AAChB,MAAM,CAAC,MAAM,6BAA6B,GAAG,0BAA0B,CAAC;AAExE,gBAAgB;AAChB,MAAM,CAAC,MAAM,6BAA6B,GAAG,0BAA0B,CAAC;AAExE,gBAAgB;AAChB,MAAM,CAAC,MAAM,wCAAwC,GAAG,qCAAqC,CAAC;AAE9F,gBAAgB;AAChB,MAAM,CAAC,MAAM,gBAAgB,GAAG,eAAe,CAAC;AAEhD,gBAAgB;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAAG,yBAAyB,CAAC;AAErE,gBAAgB;AAChB,MAAM,CAAC,MAAM,kBAAkB,GAAG,iBAAiB,CAAC;AAEpD,gBAAgB;AAChB,MAAM,CAAC,MAAM,uBAAuB,GAAG,qBAAqB,CAAC;AAE7D,gBAAgB;AAChB,MAAM,CAAC,MAAM,8BAA8B,GAAG,4BAA4B,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\n/** @internal */\nexport const CATEGORY_CLASS_NAME = \"BisCore.Category\";\n\n/** @internal */\nexport const DEFINITION_CONTAINER_CLASS_NAME = \"BisCore.DefinitionContainer\";\n\n/** @internal */\nexport const DEFINITION_ELEMENT_CLASS_NAME = \"BisCore.DefinitionElement\";\n\n/** @internal */\nexport const DEFINITION_MODEL_CLASS_NAME = \"BisCore.DefinitionModel\";\n\n/** @internal */\nexport const DRAWING_CATEGORY_CLASS_NAME = \"BisCore.DrawingCategory\";\n\n/** @internal */\nexport const ELEMENT_CLASS_NAME = \"BisCore.Element\";\n\n/** @internal */\nexport const GEOMETRIC_ELEMENT_2D_CLASS_NAME = \"BisCore.GeometricElement2d\";\n\n/** @internal */\nexport const GEOMETRIC_ELEMENT_3D_CLASS_NAME = \"BisCore.GeometricElement3d\";\n\n/** @internal */\nexport const GEOMETRIC_MODEL_2D_CLASS_NAME = \"BisCore.GeometricModel2d\";\n\n/** @internal */\nexport const GEOMETRIC_MODEL_3D_CLASS_NAME = \"BisCore.GeometricModel3d\";\n\n/** @internal */\nexport const INFORMATION_PARTITION_ELEMENT_CLASS_NAME = \"BisCore.InformationPartitionElement\";\n\n/** @internal */\nexport const MODEL_CLASS_NAME = \"BisCore.Model\";\n\n/** @internal */\nexport const SPATIAL_CATEGORY_CLASS_NAME = \"BisCore.SpatialCategory\";\n\n/** @internal */\nexport const SUBJECT_CLASS_NAME = \"BisCore.Subject\";\n\n/** @internal */\nexport const SUB_CATEGORY_CLASS_NAME = \"BisCore.SubCategory\";\n\n/** @internal */\nexport const SUB_MODELED_ELEMENT_CLASS_NAME = \"BisCore.ISubModeledElement\";\n"]}

package/lib/esm/tree-widget-react/components/trees/common/internal/ModelCategoryElementsCountCache.d.ts

@@ -0,0 +1,16 @@
+/// <reference types="node" resolution-mode="require"/>
+import type { Id64String } from "@itwin/core-bentley";
+import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+/** @internal */
+export declare class ModelCategoryElementsCountCache implements Disposable {
+    private _queryExecutor;
+    private _elementsClassName;
+    private _cache;
+    private _requestsStream;
+    private _subscription;
+    constructor(_queryExecutor: LimitingECSqlQueryExecutor, _elementsClassName: string);
+    private queryCategoryElementCounts;
+    [Symbol.dispose](): void;
+    getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number>;
+}
+//# sourceMappingURL=ModelCategoryElementsCountCache.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/internal/ModelCategoryElementsCountCache.js

@@ -0,0 +1,72 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { bufferTime, filter, firstValueFrom, mergeAll, mergeMap, ReplaySubject, Subject } from "rxjs";
+import { assert } from "@itwin/core-bentley";
+/** @internal */
+export class ModelCategoryElementsCountCache {
+    constructor(_queryExecutor, _elementsClassName) {
+        this._queryExecutor = _queryExecutor;
+        this._elementsClassName = _elementsClassName;
+        this._cache = new Map();
+        this._requestsStream = new Subject();
+        this._subscription = this._requestsStream
+            .pipe(bufferTime(20), filter((requests) => requests.length > 0), mergeMap(async (requests) => this.queryCategoryElementCounts(requests)), mergeAll())
+            .subscribe({
+            next: ({ modelId, categoryId, elementsCount }) => {
+                const subject = this._cache.get(`${modelId}-${categoryId}`);
+                assert(!!subject);
+                subject.next(elementsCount);
+            },
+        });
+    }
+    async queryCategoryElementCounts(input) {
+        const reader = this._queryExecutor.createQueryReader({
+            ctes: [
+                `
+            CategoryElements(id, modelId, categoryId) AS (
+              SELECT ECInstanceId, Model.Id, Category.Id
+              FROM ${this._elementsClassName}
+              WHERE
+                Parent.Id IS NULL
+                AND (
+                  ${input.map(({ modelId, categoryId }) => `Model.Id = ${modelId} AND Category.Id = ${categoryId}`).join(" OR ")}
+                )
+
+              UNION ALL
+
+              SELECT c.ECInstanceId, p.modelId, p.categoryId
+              FROM ${this._elementsClassName} c
+              JOIN CategoryElements p ON c.Parent.Id = p.id
+            )
+          `,
+            ],
+            ecsql: `
+          SELECT modelId, categoryId, COUNT(id) elementsCount
+          FROM CategoryElements
+          GROUP BY modelId, categoryId
+        `,
+        }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" });
+        const result = new Array();
+        for await (const row of reader) {
+            result.push({ modelId: row.modelId, categoryId: row.categoryId, elementsCount: row.elementsCount });
+        }
+        return result;
+    }
+    [Symbol.dispose]() {
+        this._subscription.unsubscribe();
+    }
+    async getCategoryElementsCount(modelId, categoryId) {
+        const cacheKey = `${modelId}-${categoryId}`;
+        let result = this._cache.get(cacheKey);
+        if (result !== undefined) {
+            return firstValueFrom(result);
+        }
+        result = new ReplaySubject(1);
+        this._cache.set(cacheKey, result);
+        this._requestsStream.next({ modelId, categoryId });
+        return firstValueFrom(result);
+    }
+}
+//# sourceMappingURL=ModelCategoryElementsCountCache.js.map

package/lib/esm/tree-widget-react/components/trees/common/internal/ModelCategoryElementsCountCache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ModelCategoryElementsCountCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/ModelCategoryElementsCountCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACtG,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAS7C,gBAAgB;AAChB,MAAM,OAAO,+BAA+B;IAK1C,YACU,cAA0C,EAC1C,kBAA0B;QAD1B,mBAAc,GAAd,cAAc,CAA4B;QAC1C,uBAAkB,GAAlB,kBAAkB,CAAQ;QAN5B,WAAM,GAAG,IAAI,GAAG,EAAqC,CAAC;QACtD,oBAAe,GAAG,IAAI,OAAO,EAAmD,CAAC;QAOvF,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,eAAe;aACtC,IAAI,CACH,UAAU,CAAC,EAAE,CAAC,EACd,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EACzC,QAAQ,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,CAAC,EACvE,QAAQ,EAAE,CACX;aACA,SAAS,CAAC;YACT,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;gBAC5D,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9B,CAAC;SACF,CAAC,CAAC;IACP,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACtC,KAA6D;QAE7D,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAClD;YACE,IAAI,EAAE;gBACJ;;;qBAGW,IAAI,CAAC,kBAAkB;;;;oBAIxB,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,cAAc,OAAO,sBAAsB,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;;;;;;qBAM3G,IAAI,CAAC,kBAAkB;;;WAGjC;aACF;YACD,KAAK,EAAE;;;;SAIN;SACF,EACD,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CACxD,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,KAAK,EAAkE,CAAC;QAC3F,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE;YAC9B,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,aAAa,EAAE,GAAG,CAAC,aAAa,EAAE,CAAC,CAAC;SACrG;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;IACnC,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,MAAM,QAAQ,GAAqB,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;QAC9D,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,MAAM,KAAK,SAAS,EAAE;YACxB,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;SAC/B;QAED,MAAM,GAAG,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAClC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;IAChC,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 { bufferTime, filter, firstValueFrom, mergeAll, mergeMap, ReplaySubject, Subject } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\n\nimport type { Id64String } from \"@itwin/core-bentley\";\nimport type { Subscription } from \"rxjs\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { CategoryId, ModelId } from \"./Types.js\";\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\n/** @internal */\nexport class ModelCategoryElementsCountCache implements Disposable {\n  private _cache = new Map<ModelCategoryKey, Subject<number>>();\n  private _requestsStream = new Subject<{ modelId: Id64String; categoryId: Id64String }>();\n  private _subscription: Subscription;\n\n  public constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    private _elementsClassName: string,\n  ) {\n    this._subscription = this._requestsStream\n      .pipe(\n        bufferTime(20),\n        filter((requests) => requests.length > 0),\n        mergeMap(async (requests) => this.queryCategoryElementCounts(requests)),\n        mergeAll(),\n      )\n      .subscribe({\n        next: ({ modelId, categoryId, elementsCount }) => {\n          const subject = this._cache.get(`${modelId}-${categoryId}`);\n          assert(!!subject);\n          subject.next(elementsCount);\n        },\n      });\n  }\n\n  private async queryCategoryElementCounts(\n    input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n  ): Promise<Array<{ modelId: number; categoryId: number; elementsCount: number }>> {\n    const reader = this._queryExecutor.createQueryReader(\n      {\n        ctes: [\n          `\n            CategoryElements(id, modelId, categoryId) AS (\n              SELECT ECInstanceId, Model.Id, Category.Id\n              FROM ${this._elementsClassName}\n              WHERE\n                Parent.Id IS NULL\n                AND (\n                  ${input.map(({ modelId, categoryId }) => `Model.Id = ${modelId} AND Category.Id = ${categoryId}`).join(\" OR \")}\n                )\n\n              UNION ALL\n\n              SELECT c.ECInstanceId, p.modelId, p.categoryId\n              FROM ${this._elementsClassName} c\n              JOIN CategoryElements p ON c.Parent.Id = p.id\n            )\n          `,\n        ],\n        ecsql: `\n          SELECT modelId, categoryId, COUNT(id) elementsCount\n          FROM CategoryElements\n          GROUP BY modelId, categoryId\n        `,\n      },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" },\n    );\n\n    const result = new Array<{ modelId: number; categoryId: number; elementsCount: number }>();\n    for await (const row of reader) {\n      result.push({ modelId: row.modelId, categoryId: row.categoryId, elementsCount: row.elementsCount });\n    }\n    return result;\n  }\n\n  public [Symbol.dispose]() {\n    this._subscription.unsubscribe();\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    const cacheKey: ModelCategoryKey = `${modelId}-${categoryId}`;\n    let result = this._cache.get(cacheKey);\n    if (result !== undefined) {\n      return firstValueFrom(result);\n    }\n\n    result = new ReplaySubject(1);\n    this._cache.set(cacheKey, result);\n    this._requestsStream.next({ modelId, categoryId });\n    return firstValueFrom(result);\n  }\n}\n"]}