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

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

@@ -1 +1 @@
-{"version":3,"file":"UseCategoriesTree.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/categories-tree/UseCategoriesTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AACxD,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAC9C,OAAO,WAAW,MAAM,sCAAsC,CAAC;AAC/D,OAAO,cAAc,MAAM,+CAA+C,CAAC;AAC3E,OAAO,QAAQ,MAAM,gCAAgC,CAAC;AACtD,OAAO,sBAAsB,MAAM,gDAAgD,CAAC;AACpF,OAAO,UAAU,MAAM,kCAAkC,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AAChI,OAAO,EAAE,OAAO,EAAE,MAAM,+BAA+B,CAAC;AACxD,OAAO,EAAE,mBAAmB,EAAE,MAAM,wDAAwD,CAAC;AAC7F,OAAO,EAAE,WAAW,EAAE,MAAM,gDAAgD,CAAC;AAC7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,wBAAwB,EAAE,6BAA6B,EAAE,MAAM,+BAA+B,CAAC;AACxG,OAAO,EAAE,sBAAsB,EAAE,MAAM,sCAAsC,CAAC;AAC9E,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,EAAE,+BAA+B,EAAE,MAAM,0DAA0D,CAAC;AAC3G,OAAO,EAAE,4BAA4B,EAAE,MAAM,uCAAuC,CAAC;AAkCrF;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,EAChC,MAAM,EACN,UAAU,EACV,oBAAoB,EACpB,gBAAgB,EAChB,eAAe,GACQ;IACvB,MAAM,sBAAsB,GAAG,OAAO,CACpC,GAAG,EAAE,CAAC,CAAC;QACL,GAAG,6BAA6B;QAChC,GAAG,eAAe;KACnB,CAAC;IACF,uDAAuD;IACvD,MAAM,CAAC,MAAM,CAAC,eAAe,IAAI,EAAE,CAAC,CACrC,CAAC;IACF,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IAC5D,MAAM,WAAW,GAAG,OAAO,EAAE,CAAC;IAE9B,MAAM,EAAE,QAAQ,EAAE,yBAAyB,EAAE,GAAG,WAAW,CAAoD;QAC7G,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,WAAW;QACX,kBAAkB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;QAC7D,WAAW;KACZ,CAAC,CAAC;IAEH,MAAM,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,GAAG,6BAA6B,CAAC;QACzF,UAAU;QACV,QAAQ;QACR,QAAQ,EAAE,yBAAyB;QACnC,WAAW;QACX,eAAe,EAAE,sBAAsB;KACxC,CAAC,CAAC;IAEH,MAAM,sBAAsB,GAAG,WAAW,CACxC,CAAC,KAAK,EAAE,EAAE;QACR,OAAO,IAAI,wBAAwB,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,yBAAyB,EAAE,EAAE,eAAe,EAAE,sBAAsB,EAAE,CAAC,CAAC;IAC9I,CAAC,EACD,CAAC,QAAQ,EAAE,yBAAyB,EAAE,sBAAsB,CAAC,CAC9D,CAAC;IAEF,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,gBAAgB,CAAC;QACpD,sBAAsB;QACtB,MAAM;QACN,yBAAyB;QACzB,sBAAsB;QACtB,QAAQ;QACR,oBAAoB;QACpB,WAAW;KACZ,CAAC,CAAC;IAEH,OAAO;QACL,mBAAmB,EAAE;YACnB,QAAQ,EAAE,oBAAoB;YAC9B,sBAAsB;YACtB,gBAAgB,EAAE,QAAQ;YAC1B,wBAAwB;YACxB,gBAAgB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,4BAA4B,CAAC,MAAM,EAAE,cAAc,EAAE,gBAAgB,CAAC,EAAE,CAAC,MAAM,EAAE,cAAc,EAAE,gBAAgB,CAAC,CAAC;YACnJ,aAAa,EAAE,MAAM;SACtB;QACD,aAAa,EAAE;YACb,cAAc,EAAE,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAC,kBAAkB,IAAC,IAAI,EAAE,IAAI,GAAI,EAAE,EAAE,CAAC;YAC7E,WAAW;SACZ;KACF,CAAC;AACJ,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAe,EAAE,KAAoC,EAAE,gBAAkC;IAC7H,IAAI,KAAK,EAAE,CAAC;QACV,IAAI,KAAK,KAAK,sBAAsB,EAAE,CAAC;YACrC,OAAO,KAAC,oBAAoB,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;QAC1D,CAAC;QACD,OAAO,KAAC,kBAAkB,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;IACxD,CAAC;IACD,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,KAAC,eAAe,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;IACrD,CAAC;IACD,IAAI,gBAAgB,EAAE,CAAC;QACrB,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IACD,OAAO,KAAC,gBAAgB,IAAC,IAAI,EAAE,WAAW,GAAI,CAAC;AACjD,CAAC;AAED,YAAY;AACZ,MAAM,UAAU,kBAAkB,CAAC,EAAE,IAAI,EAAuC;IAC9E,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;QACtD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,QAAQ,IAAI,CAAC,QAAQ,CAAC,YAAa,CAAC,OAAO,EAAE,CAAC;YAC5C,KAAK,aAAa;gBAChB,OAAO,WAAW,CAAC;YACrB,KAAK,qBAAqB;gBACxB,OAAO,cAAc,CAAC;YACxB,KAAK,2BAA2B;gBAC9B,OAAO,sBAAsB,CAAC;YAChC,KAAK,WAAW;gBACd,OAAO,UAAU,CAAC;YACpB,KAAK,eAAe;gBAClB,OAAO,QAAQ,CAAC;YAClB;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,KAAC,IAAI,IAAC,IAAI,EAAE,OAAO,EAAE,GAAI,CAAC;AACnC,CAAC;AAED,SAAS,WAAW,CAAC,IAA+B;IAClD,OAAO,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,WAAW,CAAC;AACjD,CAAC;AAED,SAAS,6BAA6B,CAAC,KAMtC;IACC,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAC9D,MAAM,EAAE,wBAAwB,EAAE,aAAa,EAAE,sBAAsB,EAAE,GAAG,mBAAmB,CAAsD;QACnJ,UAAU;QACV,QAAQ;QACR,kBAAkB,EAAE,WAAW,CAC7B,KAAK,EAAE,iBAAkE,EAAE,EAAE,CAC3E,kBAAkB,CAAC,EAAE,GAAG,iBAAiB,EAAE,WAAW,EAAE,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC,EACvF,CAAC,QAAQ,CAAC,CACX;QACD,mCAAmC,EAAE,WAAW,CAC9C,CAAC,aAAa,EAAE,EAAE,CAAC,mCAAmC,CAAC,EAAE,GAAG,aAAa,EAAE,eAAe,EAAE,KAAK,CAAC,eAAe,EAAE,CAAC,EACpH,CAAC,KAAK,CAAC,eAAe,CAAC,CACxB;QACD,WAAW;KACZ,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,QAAQ,EAAE,CAAC,2BAA2B,EAAE,CAAC;IAC3C,CAAC,EAAE,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC,CAAC;IAE9B,OAAO;QACL,wBAAwB;QACxB,sBAAsB;KACvB,CAAC;AACJ,CAAC;AAED,SAAS,mCAAmC,CAC1C,KAAmI;IAEnI,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC;IAC5D,OAAO,IAAI,+BAA+B,CAAC;QACzC,8BAA8B,EAAE,IAAI;QACpC,QAAQ,EAAE,QAAQ,EAAE;QACpB,QAAQ;QACR,eAAe;KAChB,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,kBAAkB,CAC/B,KAA6G;IAE7G,MAAM,EAAE,cAAc,EAAE,YAAY,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IACtE,OAAO,4BAA4B,CAAC;QAClC,YAAY;QACZ,cAAc;QACd,QAAQ,EAAE,QAAQ,EAAE;QACpB,iBAAiB,EAAE,WAAW,CAAC,aAAa;QAC5C,wBAAwB,EAAE,WAAW,CAAC,YAAY;QAClD,sBAAsB,EAAE,WAAW,CAAC,UAAU;KAC/C,CAAC,CAAC;AACL,CAAC;AAED,SAAS,WAAW,CAAC,KAAkD;IACrE,OAAO,IAAI,sBAAsB,CAAC,wBAAwB,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC;AAC7H,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useEffect, useMemo } from \"react\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { Icon } from \"@stratakit/foundations\";\nimport categorySvg from \"@stratakit/icons/bis-category-3d.svg\";\nimport subcategorySvg from \"@stratakit/icons/bis-category-subcategory.svg\";\nimport classSvg from \"@stratakit/icons/bis-class.svg\";\nimport definitionContainerSvg from \"@stratakit/icons/bis-definitions-container.svg\";\nimport elementSvg from \"@stratakit/icons/bis-element.svg\";\nimport { EmptyTreeContent, FilterUnknownError, NoFilterMatches, TooManyFilterMatches } from \"../common/components/EmptyTree.js\";\nimport { useGuid } from \"../common/internal/useGuid.js\";\nimport { useCachedVisibility } from \"../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport { useIdsCache } from \"../common/internal/useTreeHooks/UseIdsCache.js\";\nimport { getClassesByView } from \"../common/internal/Utils.js\";\nimport { CategoriesTreeDefinition, defaultHierarchyConfiguration } from \"./CategoriesTreeDefinition.js\";\nimport { CategoriesTreeIdsCache } from \"./internal/CategoriesTreeIdsCache.js\";\nimport { useFilteredPaths } from \"./internal/UseFilteredPaths.js\";\nimport { CategoriesTreeVisibilityHandler } from \"./internal/visibility/CategoriesTreeVisibilityHandler.js\";\nimport { createFilteredCategoriesTree } from \"./internal/visibility/FilteredTree.js\";\n\nimport type { ReactNode } from \"react\";\nimport type { GuidString, Id64Array } from \"@itwin/core-bentley\";\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { CategoryInfo } from \"../common/CategoriesVisibilityUtils.js\";\nimport type { VisibilityTreeProps } from \"../common/components/VisibilityTree.js\";\nimport type { VisibilityTreeRendererProps } from \"../common/components/VisibilityTreeRenderer.js\";\nimport type { CreateFilteredTreeProps, CreateTreeSpecificVisibilityHandlerProps } from \"../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport type { CreateCacheProps } from \"../common/internal/useTreeHooks/UseIdsCache.js\";\nimport type { FilteredTree } from \"../common/internal/visibility/BaseFilteredTree.js\";\nimport type { TreeWidgetViewport } from \"../common/TreeWidgetViewport.js\";\nimport type { CategoriesTreeHierarchyConfiguration } from \"./CategoriesTreeDefinition.js\";\nimport type { CategoriesTreeFilteringError } from \"./internal/UseFilteredPaths.js\";\nimport type { CategoriesTreeFilterTargets } from \"./internal/visibility/FilteredTree.js\";\n\n/** @beta */\nexport interface UseCategoriesTreeProps {\n  activeView: TreeWidgetViewport;\n  onCategoriesFiltered?: (props: { categories: CategoryInfo[] | undefined; models?: Id64Array }) => void;\n  filter?: string;\n  emptyTreeContent?: ReactNode;\n  hierarchyConfig?: Partial<CategoriesTreeHierarchyConfiguration>;\n}\n\n/** @beta */\ninterface UseCategoriesTreeResult {\n  categoriesTreeProps: Pick<\n    VisibilityTreeProps,\n    \"treeName\" | \"getHierarchyDefinition\" | \"getFilteredPaths\" | \"visibilityHandlerFactory\" | \"highlightText\" | \"emptyTreeContent\"\n  >;\n  rendererProps: Required<Pick<VisibilityTreeRendererProps, \"getDecorations\" | \"getSublabel\">>;\n}\n\n/**\n * Custom hook to create and manage state for the categories tree.\n * @beta\n */\nexport function useCategoriesTree({\n  filter,\n  activeView,\n  onCategoriesFiltered,\n  emptyTreeContent,\n  hierarchyConfig,\n}: UseCategoriesTreeProps): UseCategoriesTreeResult {\n  const hierarchyConfiguration = useMemo<CategoriesTreeHierarchyConfiguration>(\n    () => ({\n      ...defaultHierarchyConfiguration,\n      ...hierarchyConfig,\n    }),\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    Object.values(hierarchyConfig ?? {}),\n  );\n  const viewType = activeView.viewType === \"2d\" ? \"2d\" : \"3d\";\n  const componentId = useGuid();\n\n  const { getCache: getCategoriesTreeIdsCache } = useIdsCache<CategoriesTreeIdsCache, { viewType: \"2d\" | \"3d\" }>({\n    imodel: activeView.iModel,\n    createCache,\n    cacheSpecificProps: useMemo(() => ({ viewType }), [viewType]),\n    componentId,\n  });\n\n  const { visibilityHandlerFactory, onFilteredPathsChanged } = useCategoriesCachedVisibility({\n    activeView,\n    viewType,\n    getCache: getCategoriesTreeIdsCache,\n    componentId,\n    hierarchyConfig: hierarchyConfiguration,\n  });\n\n  const getHierarchyDefinition = useCallback<VisibilityTreeProps[\"getHierarchyDefinition\"]>(\n    (props) => {\n      return new CategoriesTreeDefinition({ ...props, viewType, idsCache: getCategoriesTreeIdsCache(), hierarchyConfig: hierarchyConfiguration });\n    },\n    [viewType, getCategoriesTreeIdsCache, hierarchyConfiguration],\n  );\n\n  const { getPaths, filteringError } = useFilteredPaths({\n    hierarchyConfiguration,\n    filter,\n    getCategoriesTreeIdsCache,\n    onFilteredPathsChanged,\n    viewType,\n    onCategoriesFiltered,\n    componentId,\n  });\n\n  return {\n    categoriesTreeProps: {\n      treeName: \"categories-tree-v2\",\n      getHierarchyDefinition,\n      getFilteredPaths: getPaths,\n      visibilityHandlerFactory,\n      emptyTreeContent: useMemo(() => getEmptyTreeContentComponent(filter, filteringError, emptyTreeContent), [filter, filteringError, emptyTreeContent]),\n      highlightText: filter,\n    },\n    rendererProps: {\n      getDecorations: useCallback((node) => <CategoriesTreeIcon node={node} />, []),\n      getSublabel,\n    },\n  };\n}\n\nfunction getEmptyTreeContentComponent(filter?: string, error?: CategoriesTreeFilteringError, emptyTreeContent?: React.ReactNode) {\n  if (error) {\n    if (error === \"tooManyFilterMatches\") {\n      return <TooManyFilterMatches base={\"categoriesTree\"} />;\n    }\n    return <FilterUnknownError base={\"categoriesTree\"} />;\n  }\n  if (filter) {\n    return <NoFilterMatches base={\"categoriesTree\"} />;\n  }\n  if (emptyTreeContent) {\n    return emptyTreeContent;\n  }\n  return <EmptyTreeContent icon={categorySvg} />;\n}\n\n/** @beta */\nexport function CategoriesTreeIcon({ node }: { node: PresentationHierarchyNode }) {\n  if (node.nodeData.extendedData?.imageId === undefined) {\n    return undefined;\n  }\n\n  const getIcon = () => {\n    switch (node.nodeData.extendedData!.imageId) {\n      case \"icon-layers\":\n        return categorySvg;\n      case \"icon-layers-isolate\":\n        return subcategorySvg;\n      case \"icon-definition-container\":\n        return definitionContainerSvg;\n      case \"icon-item\":\n        return elementSvg;\n      case \"icon-ec-class\":\n        return classSvg;\n      default:\n        return undefined;\n    }\n  };\n\n  return <Icon href={getIcon()} />;\n}\n\nfunction getSublabel(node: PresentationHierarchyNode) {\n  return node.nodeData.extendedData?.description;\n}\n\nfunction useCategoriesCachedVisibility(props: {\n  activeView: TreeWidgetViewport;\n  getCache: () => CategoriesTreeIdsCache;\n  viewType: \"2d\" | \"3d\";\n  componentId: GuidString;\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration;\n}) {\n  const { activeView, getCache, viewType, componentId } = props;\n  const { visibilityHandlerFactory, filteredPaths, onFilteredPathsChanged } = useCachedVisibility<CategoriesTreeIdsCache, CategoriesTreeFilterTargets>({\n    activeView,\n    getCache,\n    createFilteredTree: useCallback(\n      async (filteredTreeProps: CreateFilteredTreeProps<CategoriesTreeIdsCache>) =>\n        createFilteredTree({ ...filteredTreeProps, viewClasses: getClassesByView(viewType) }),\n      [viewType],\n    ),\n    createTreeSpecificVisibilityHandler: useCallback(\n      (specificProps) => createTreeSpecificVisibilityHandler({ ...specificProps, hierarchyConfig: props.hierarchyConfig }),\n      [props.hierarchyConfig],\n    ),\n    componentId,\n  });\n\n  useEffect(() => {\n    getCache().clearFilteredElementsModels();\n  }, [filteredPaths, getCache]);\n\n  return {\n    visibilityHandlerFactory,\n    onFilteredPathsChanged,\n  };\n}\n\nfunction createTreeSpecificVisibilityHandler(\n  props: CreateTreeSpecificVisibilityHandlerProps<CategoriesTreeIdsCache> & { hierarchyConfig: CategoriesTreeHierarchyConfiguration },\n) {\n  const { info, getCache, viewport, hierarchyConfig } = props;\n  return new CategoriesTreeVisibilityHandler({\n    alwaysAndNeverDrawnElementInfo: info,\n    idsCache: getCache(),\n    viewport,\n    hierarchyConfig,\n  });\n}\n\nasync function createFilteredTree(\n  props: CreateFilteredTreeProps<CategoriesTreeIdsCache> & { viewClasses: ReturnType<typeof getClassesByView> },\n): Promise<FilteredTree<CategoriesTreeFilterTargets>> {\n  const { filteringPaths, imodelAccess, getCache, viewClasses } = props;\n  return createFilteredCategoriesTree({\n    imodelAccess,\n    filteringPaths,\n    idsCache: getCache(),\n    categoryClassName: viewClasses.categoryClass,\n    categoryElementClassName: viewClasses.elementClass,\n    categoryModelClassName: viewClasses.modelClass,\n  });\n}\n\nfunction createCache(props: CreateCacheProps<{ viewType: \"2d\" | \"3d\" }>) {\n  return new CategoriesTreeIdsCache(createECSqlQueryExecutor(props.imodel), props.specificProps.viewType, props.componentId);\n}\n"]}
+{"version":3,"file":"UseCategoriesTree.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/categories-tree/UseCategoriesTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AACxD,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAC9C,OAAO,WAAW,MAAM,sCAAsC,CAAC;AAC/D,OAAO,cAAc,MAAM,+CAA+C,CAAC;AAC3E,OAAO,QAAQ,MAAM,gCAAgC,CAAC;AACtD,OAAO,sBAAsB,MAAM,gDAAgD,CAAC;AACpF,OAAO,UAAU,MAAM,kCAAkC,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AAChI,OAAO,EAAE,OAAO,EAAE,MAAM,+BAA+B,CAAC;AACxD,OAAO,EAAE,mBAAmB,EAAE,MAAM,wDAAwD,CAAC;AAC7F,OAAO,EAAE,WAAW,EAAE,MAAM,gDAAgD,CAAC;AAC7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,wBAAwB,EAAE,6BAA6B,EAAE,MAAM,+BAA+B,CAAC;AACxG,OAAO,EAAE,sBAAsB,EAAE,MAAM,sCAAsC,CAAC;AAC9E,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EAAE,+BAA+B,EAAE,MAAM,0DAA0D,CAAC;AAC3G,OAAO,EAAE,iCAAiC,EAAE,MAAM,4CAA4C,CAAC;AAmC/F;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,EAChC,UAAU,EACV,UAAU,EACV,oBAAoB,EACpB,gBAAgB,EAChB,eAAe,EACf,gBAAgB,GACO;IACvB,MAAM,sBAAsB,GAAG,OAAO,CACpC,GAAG,EAAE,CAAC,CAAC;QACL,GAAG,6BAA6B;QAChC,GAAG,eAAe;KACnB,CAAC;IACF,uDAAuD;IACvD,MAAM,CAAC,MAAM,CAAC,eAAe,IAAI,EAAE,CAAC,CACrC,CAAC;IACF,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IAC5D,MAAM,WAAW,GAAG,OAAO,EAAE,CAAC;IAE9B,MAAM,EAAE,QAAQ,EAAE,yBAAyB,EAAE,GAAG,WAAW,CAAoD;QAC7G,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,WAAW;QACX,kBAAkB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;QAC7D,WAAW;KACZ,CAAC,CAAC;IAEH,MAAM,EAAE,wBAAwB,EAAE,oBAAoB,EAAE,GAAG,6BAA6B,CAAC;QACvF,UAAU;QACV,QAAQ;QACR,QAAQ,EAAE,yBAAyB;QACnC,WAAW;QACX,eAAe,EAAE,sBAAsB;KACxC,CAAC,CAAC;IAEH,MAAM,sBAAsB,GAAG,WAAW,CACxC,CAAC,KAAK,EAAE,EAAE;QACR,OAAO,IAAI,wBAAwB,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,yBAAyB,EAAE,EAAE,eAAe,EAAE,sBAAsB,EAAE,CAAC,CAAC;IAC9I,CAAC,EACD,CAAC,QAAQ,EAAE,yBAAyB,EAAE,sBAAsB,CAAC,CAC9D,CAAC;IAEF,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,cAAc,CAAC;QAC/C,sBAAsB;QACtB,UAAU;QACV,yBAAyB;QACzB,oBAAoB;QACpB,QAAQ;QACR,oBAAoB;QACpB,WAAW;KACZ,CAAC,CAAC;IAEH,OAAO;QACL,SAAS,EAAE;YACT,QAAQ,EAAE,oBAAoB;YAC9B,sBAAsB;YACtB,cAAc,EAAE,QAAQ;YACxB,wBAAwB;YACxB,gBAAgB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,4BAA4B,CAAC,UAAU,EAAE,WAAW,EAAE,gBAAgB,CAAC,EAAE,CAAC,UAAU,EAAE,WAAW,EAAE,gBAAgB,CAAC,CAAC;YACrJ,aAAa,EAAE,UAAU;SAC1B;QACD,gBAAgB,EAAE,CAAC,IAAI,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,aAAa,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC;YACzC,WAAW,EAAE,KAAC,kBAAkB,IAAC,IAAI,EAAE,IAAI,GAAI;YAC/C,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,WAAW;YACpD,GAAG,gBAAgB,EAAE,CAAC,IAAI,EAAE,aAAa,CAAC;SAC3C,CAAC;KACH,CAAC;AACJ,CAAC;AAED,SAAS,4BAA4B,CAAC,UAAmB,EAAE,KAAiC,EAAE,gBAAkC;IAC9H,IAAI,KAAK,EAAE,CAAC;QACV,IAAI,KAAK,KAAK,sBAAsB,EAAE,CAAC;YACrC,OAAO,KAAC,oBAAoB,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;QAC1D,CAAC;QACD,OAAO,KAAC,kBAAkB,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;IACxD,CAAC;IACD,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,KAAC,eAAe,IAAC,IAAI,EAAE,gBAAgB,GAAI,CAAC;IACrD,CAAC;IACD,IAAI,gBAAgB,EAAE,CAAC;QACrB,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IACD,OAAO,KAAC,gBAAgB,IAAC,IAAI,EAAE,WAAW,GAAI,CAAC;AACjD,CAAC;AAED,YAAY;AACZ,MAAM,UAAU,kBAAkB,CAAC,EAAE,IAAI,EAAuC;IAC9E,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;QACtD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,QAAQ,IAAI,CAAC,QAAQ,CAAC,YAAa,CAAC,OAAO,EAAE,CAAC;YAC5C,KAAK,aAAa;gBAChB,OAAO,WAAW,CAAC;YACrB,KAAK,qBAAqB;gBACxB,OAAO,cAAc,CAAC;YACxB,KAAK,2BAA2B;gBAC9B,OAAO,sBAAsB,CAAC;YAChC,KAAK,WAAW;gBACd,OAAO,UAAU,CAAC;YACpB,KAAK,eAAe;gBAClB,OAAO,QAAQ,CAAC;YAClB;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,KAAC,IAAI,IAAC,IAAI,EAAE,OAAO,EAAE,GAAI,CAAC;AACnC,CAAC;AAED,SAAS,6BAA6B,CAAC,KAMtC;IACC,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAC9D,MAAM,EAAE,wBAAwB,EAAE,WAAW,EAAE,oBAAoB,EAAE,GAAG,mBAAmB,CAAsD;QAC/I,UAAU;QACV,QAAQ;QACR,uBAAuB,EAAE,WAAW,CAClC,KAAK,EAAE,iBAAuE,EAAE,EAAE,CAChF,uBAAuB,CAAC,EAAE,GAAG,iBAAiB,EAAE,WAAW,EAAE,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC,EAC5F,CAAC,QAAQ,CAAC,CACX;QACD,mCAAmC,EAAE,WAAW,CAC9C,CAAC,aAAa,EAAE,EAAE,CAAC,mCAAmC,CAAC,EAAE,GAAG,aAAa,EAAE,eAAe,EAAE,KAAK,CAAC,eAAe,EAAE,CAAC,EACpH,CAAC,KAAK,CAAC,eAAe,CAAC,CACxB;QACD,WAAW;KACZ,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,QAAQ,EAAE,CAAC,2BAA2B,EAAE,CAAC;IAC3C,CAAC,EAAE,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC;IAE5B,OAAO;QACL,wBAAwB;QACxB,oBAAoB;KACrB,CAAC;AACJ,CAAC;AAED,SAAS,mCAAmC,CAC1C,KAAmI;IAEnI,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC;IAC5D,OAAO,IAAI,+BAA+B,CAAC;QACzC,8BAA8B,EAAE,IAAI;QACpC,QAAQ,EAAE,QAAQ,EAAE;QACpB,QAAQ;QACR,eAAe;KAChB,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,KAAkH;IAElH,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IACnE,OAAO,iCAAiC,CAAC;QACvC,YAAY;QACZ,WAAW;QACX,QAAQ,EAAE,QAAQ,EAAE;QACpB,iBAAiB,EAAE,WAAW,CAAC,aAAa;QAC5C,wBAAwB,EAAE,WAAW,CAAC,YAAY;QAClD,sBAAsB,EAAE,WAAW,CAAC,UAAU;KAC/C,CAAC,CAAC;AACL,CAAC;AAED,SAAS,WAAW,CAAC,KAAkD;IACrE,OAAO,IAAI,sBAAsB,CAAC,wBAAwB,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC;AAC7H,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useEffect, useMemo } from \"react\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { Icon } from \"@stratakit/foundations\";\nimport categorySvg from \"@stratakit/icons/bis-category-3d.svg\";\nimport subcategorySvg from \"@stratakit/icons/bis-category-subcategory.svg\";\nimport classSvg from \"@stratakit/icons/bis-class.svg\";\nimport definitionContainerSvg from \"@stratakit/icons/bis-definitions-container.svg\";\nimport elementSvg from \"@stratakit/icons/bis-element.svg\";\nimport { EmptyTreeContent, NoSearchMatches, SearchUnknownError, TooManySearchMatches } from \"../common/components/EmptyTree.js\";\nimport { useGuid } from \"../common/internal/useGuid.js\";\nimport { useCachedVisibility } from \"../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport { useIdsCache } from \"../common/internal/useTreeHooks/UseIdsCache.js\";\nimport { getClassesByView } from \"../common/internal/Utils.js\";\nimport { CategoriesTreeDefinition, defaultHierarchyConfiguration } from \"./CategoriesTreeDefinition.js\";\nimport { CategoriesTreeIdsCache } from \"./internal/CategoriesTreeIdsCache.js\";\nimport { useSearchPaths } from \"./internal/UseSearchPaths.js\";\nimport { CategoriesTreeVisibilityHandler } from \"./internal/visibility/CategoriesTreeVisibilityHandler.js\";\nimport { createCategoriesSearchResultsTree } from \"./internal/visibility/SearchResultsTree.js\";\n\nimport type { ReactNode } from \"react\";\nimport type { GuidString, Id64Array } from \"@itwin/core-bentley\";\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { CategoryInfo } from \"../common/CategoriesVisibilityUtils.js\";\nimport type { VisibilityTreeProps } from \"../common/components/VisibilityTree.js\";\nimport type { ExtendedVisibilityTreeRendererProps } from \"../common/components/VisibilityTreeRenderer.js\";\nimport type { CreateSearchResultsTreeProps, CreateTreeSpecificVisibilityHandlerProps } from \"../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport type { CreateCacheProps } from \"../common/internal/useTreeHooks/UseIdsCache.js\";\nimport type { SearchResultsTree } from \"../common/internal/visibility/BaseSearchResultsTree.js\";\nimport type { TreeWidgetViewport } from \"../common/TreeWidgetViewport.js\";\nimport type { CategoriesTreeHierarchyConfiguration } from \"./CategoriesTreeDefinition.js\";\nimport type { CategoriesTreeSearchError } from \"./internal/UseSearchPaths.js\";\nimport type { CategoriesTreeSearchTargets } from \"./internal/visibility/SearchResultsTree.js\";\n\n/** @beta */\nexport interface UseCategoriesTreeProps {\n  activeView: TreeWidgetViewport;\n  onCategoriesFiltered?: (props: { categories: CategoryInfo[] | undefined; models?: Id64Array }) => void;\n  searchText?: string;\n  emptyTreeContent?: ReactNode;\n  hierarchyConfig?: Partial<CategoriesTreeHierarchyConfiguration>;\n  getTreeItemProps?: ExtendedVisibilityTreeRendererProps[\"getTreeItemProps\"];\n}\n\n/** @beta */\ninterface UseCategoriesTreeResult {\n  treeProps: Pick<\n    VisibilityTreeProps,\n    \"treeName\" | \"getHierarchyDefinition\" | \"getSearchPaths\" | \"visibilityHandlerFactory\" | \"highlightText\" | \"emptyTreeContent\"\n  >;\n  getTreeItemProps: Required<ExtendedVisibilityTreeRendererProps>[\"getTreeItemProps\"];\n}\n\n/**\n * Custom hook to create and manage state for the categories tree.\n * @beta\n */\nexport function useCategoriesTree({\n  searchText,\n  activeView,\n  onCategoriesFiltered,\n  emptyTreeContent,\n  hierarchyConfig,\n  getTreeItemProps,\n}: UseCategoriesTreeProps): UseCategoriesTreeResult {\n  const hierarchyConfiguration = useMemo<CategoriesTreeHierarchyConfiguration>(\n    () => ({\n      ...defaultHierarchyConfiguration,\n      ...hierarchyConfig,\n    }),\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    Object.values(hierarchyConfig ?? {}),\n  );\n  const viewType = activeView.viewType === \"2d\" ? \"2d\" : \"3d\";\n  const componentId = useGuid();\n\n  const { getCache: getCategoriesTreeIdsCache } = useIdsCache<CategoriesTreeIdsCache, { viewType: \"2d\" | \"3d\" }>({\n    imodel: activeView.iModel,\n    createCache,\n    cacheSpecificProps: useMemo(() => ({ viewType }), [viewType]),\n    componentId,\n  });\n\n  const { visibilityHandlerFactory, onSearchPathsChanged } = useCategoriesCachedVisibility({\n    activeView,\n    viewType,\n    getCache: getCategoriesTreeIdsCache,\n    componentId,\n    hierarchyConfig: hierarchyConfiguration,\n  });\n\n  const getHierarchyDefinition = useCallback<VisibilityTreeProps[\"getHierarchyDefinition\"]>(\n    (props) => {\n      return new CategoriesTreeDefinition({ ...props, viewType, idsCache: getCategoriesTreeIdsCache(), hierarchyConfig: hierarchyConfiguration });\n    },\n    [viewType, getCategoriesTreeIdsCache, hierarchyConfiguration],\n  );\n\n  const { getPaths, searchError } = useSearchPaths({\n    hierarchyConfiguration,\n    searchText,\n    getCategoriesTreeIdsCache,\n    onSearchPathsChanged,\n    viewType,\n    onCategoriesFiltered,\n    componentId,\n  });\n\n  return {\n    treeProps: {\n      treeName: \"categories-tree-v2\",\n      getHierarchyDefinition,\n      getSearchPaths: getPaths,\n      visibilityHandlerFactory,\n      emptyTreeContent: useMemo(() => getEmptyTreeContentComponent(searchText, searchError, emptyTreeContent), [searchText, searchError, emptyTreeContent]),\n      highlightText: searchText,\n    },\n    getTreeItemProps: (node, rendererProps) => ({\n      ...rendererProps.getTreeItemProps?.(node),\n      decorations: <CategoriesTreeIcon node={node} />,\n      description: node.nodeData.extendedData?.description,\n      ...getTreeItemProps?.(node, rendererProps),\n    }),\n  };\n}\n\nfunction getEmptyTreeContentComponent(searchText?: string, error?: CategoriesTreeSearchError, emptyTreeContent?: React.ReactNode) {\n  if (error) {\n    if (error === \"tooManySearchMatches\") {\n      return <TooManySearchMatches base={\"categoriesTree\"} />;\n    }\n    return <SearchUnknownError base={\"categoriesTree\"} />;\n  }\n  if (searchText) {\n    return <NoSearchMatches base={\"categoriesTree\"} />;\n  }\n  if (emptyTreeContent) {\n    return emptyTreeContent;\n  }\n  return <EmptyTreeContent icon={categorySvg} />;\n}\n\n/** @beta */\nexport function CategoriesTreeIcon({ node }: { node: PresentationHierarchyNode }) {\n  if (node.nodeData.extendedData?.imageId === undefined) {\n    return undefined;\n  }\n\n  const getIcon = () => {\n    switch (node.nodeData.extendedData!.imageId) {\n      case \"icon-layers\":\n        return categorySvg;\n      case \"icon-layers-isolate\":\n        return subcategorySvg;\n      case \"icon-definition-container\":\n        return definitionContainerSvg;\n      case \"icon-item\":\n        return elementSvg;\n      case \"icon-ec-class\":\n        return classSvg;\n      default:\n        return undefined;\n    }\n  };\n\n  return <Icon href={getIcon()} />;\n}\n\nfunction useCategoriesCachedVisibility(props: {\n  activeView: TreeWidgetViewport;\n  getCache: () => CategoriesTreeIdsCache;\n  viewType: \"2d\" | \"3d\";\n  componentId: GuidString;\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration;\n}) {\n  const { activeView, getCache, viewType, componentId } = props;\n  const { visibilityHandlerFactory, searchPaths, onSearchPathsChanged } = useCachedVisibility<CategoriesTreeIdsCache, CategoriesTreeSearchTargets>({\n    activeView,\n    getCache,\n    createSearchResultsTree: useCallback(\n      async (filteredTreeProps: CreateSearchResultsTreeProps<CategoriesTreeIdsCache>) =>\n        createSearchResultsTree({ ...filteredTreeProps, viewClasses: getClassesByView(viewType) }),\n      [viewType],\n    ),\n    createTreeSpecificVisibilityHandler: useCallback(\n      (specificProps) => createTreeSpecificVisibilityHandler({ ...specificProps, hierarchyConfig: props.hierarchyConfig }),\n      [props.hierarchyConfig],\n    ),\n    componentId,\n  });\n\n  useEffect(() => {\n    getCache().clearFilteredElementsModels();\n  }, [searchPaths, getCache]);\n\n  return {\n    visibilityHandlerFactory,\n    onSearchPathsChanged,\n  };\n}\n\nfunction createTreeSpecificVisibilityHandler(\n  props: CreateTreeSpecificVisibilityHandlerProps<CategoriesTreeIdsCache> & { hierarchyConfig: CategoriesTreeHierarchyConfiguration },\n) {\n  const { info, getCache, viewport, hierarchyConfig } = props;\n  return new CategoriesTreeVisibilityHandler({\n    alwaysAndNeverDrawnElementInfo: info,\n    idsCache: getCache(),\n    viewport,\n    hierarchyConfig,\n  });\n}\n\nasync function createSearchResultsTree(\n  props: CreateSearchResultsTreeProps<CategoriesTreeIdsCache> & { viewClasses: ReturnType<typeof getClassesByView> },\n): Promise<SearchResultsTree<CategoriesTreeSearchTargets>> {\n  const { searchPaths, imodelAccess, getCache, viewClasses } = props;\n  return createCategoriesSearchResultsTree({\n    imodelAccess,\n    searchPaths,\n    idsCache: getCache(),\n    categoryClassName: viewClasses.categoryClass,\n    categoryElementClassName: viewClasses.elementClass,\n    categoryModelClassName: viewClasses.modelClass,\n  });\n}\n\nfunction createCache(props: CreateCacheProps<{ viewType: \"2d\" | \"3d\" }>) {\n  return new CategoriesTreeIdsCache(createECSqlQueryExecutor(props.imodel), props.specificProps.viewType, props.componentId);\n}\n"]}

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

@@ -21,7 +21,7 @@ export declare class CategoriesTreeIdsCache implements Disposable {
     private queryCategories;
     private queryIsDefinitionContainersSupported;
     private queryDefinitionContainers;
-    private queryVisibleSubCategories;
+    private querySubCategories;
     private getModelsCategoriesInfo;
     private getElementModelsCategories;
     private queryModeledElements;
@@ -67,7 +67,7 @@ export declare class CategoriesTreeIdsCache implements Disposable {
         categories: CategoryInfo[];
         definitionContainers: Array<DefinitionContainerId>;
     }>;
-    getSubCategories(categoryIds: Id64Arg): Observable<Map<CategoryId, Array<SubCategoryId>>>;
+    getSubCategories(categoryId: Id64String): Observable<Array<SubCategoryId>>;
     getIsDefinitionContainerSupported(): Observable<boolean>;
 }
 //# sourceMappingURL=CategoriesTreeIdsCache.d.ts.map

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

@@ -6,6 +6,7 @@ import { defer, EMPTY, forkJoin, from, map, mergeMap, of, reduce, shareReplay, t
 import { Guid, Id64 } from "@itwin/core-bentley";
 import { CLASS_NAME_DefinitionContainer, CLASS_NAME_Model, CLASS_NAME_SubCategory } from "../../common/internal/ClassNameDefinitions.js";
 import { ModelCategoryElementsCountCache } from "../../common/internal/ModelCategoryElementsCountCache.js";
+import { catchBeSQLiteInterrupts } from "../../common/internal/UseErrorState.js";
 import { getClassesByView, joinId64Arg } from "../../common/internal/Utils.js";
 /** @internal */
 export class CategoriesTreeIdsCache {
@@ -48,7 +49,7 @@ export class CategoriesTreeIdsCache {
                 limit: "unbounded",
                 restartToken: `${this.#componentName}/${this.#componentId}/filtered-element-models/${Guid.createValue()}`,
             });
-        }).pipe(map((row) => {
+        }).pipe(catchBeSQLiteInterrupts, map((row) => {
             return { modelId: row.modelId, id: row.id };
         }));
     }
@@ -75,7 +76,7 @@ export class CategoriesTreeIdsCache {
         GROUP BY modelId, categoryId
       `;
             return this.#queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/element-models-and-categories` });
-        }).pipe(map((row) => {
+        }).pipe(catchBeSQLiteInterrupts, map((row) => {
             return { modelId: row.modelId, categoryId: row.categoryId };
         }));
     }
@@ -107,7 +108,7 @@ export class CategoriesTreeIdsCache {
             GROUP BY this.ECInstanceId
           `;
             return this.#queryExecutor.createQueryReader({ ecsql: categoriesQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/categories` });
-        }).pipe(map((row) => {
+        }).pipe(catchBeSQLiteInterrupts, map((row) => {
             return {
                 id: row.id,
                 modelId: row.modelId,
@@ -130,7 +131,7 @@ export class CategoriesTreeIdsCache {
           AND c.Name = 'DefinitionContainer'
       `;
             return this.#queryExecutor.createQueryReader({ ecsql: query }, { restartToken: `${this.#componentName}/${this.#componentId}/is-definition-container-supported` });
-        }).pipe(toArray(), map((rows) => rows.length > 0));
+        }).pipe(catchBeSQLiteInterrupts, toArray(), map((rows) => rows.length > 0));
     }
     queryDefinitionContainers() {
         return defer(() => {
@@ -182,24 +183,23 @@ export class CategoriesTreeIdsCache {
         SELECT dc.ECInstanceId id, dc.ModelId modelId, MAX(dc.HasElements) hasElements FROM ${DEFINITION_CONTAINERS_CTE} dc GROUP BY dc.ECInstanceId
       `;
             return this.#queryExecutor.createQueryReader({ ctes, ecsql: definitionsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/definition-containers` });
-        }).pipe(map((row) => {
+        }).pipe(catchBeSQLiteInterrupts, map((row) => {
             return { id: row.id, modelId: row.modelId, hasElements: !!row.hasElements };
         }));
     }
-    queryVisibleSubCategories(categoryIds) {
+    querySubCategories() {
         return defer(() => {
             const definitionsQuery = `
-      SELECT
+        SELECT
           sc.ECInstanceId id,
           sc.Parent.Id categoryId
         FROM
           ${CLASS_NAME_SubCategory} sc
         WHERE
           NOT sc.IsPrivate
-          AND sc.Parent.Id IN (${categoryIds.join(",")})
       `;
             return this.#queryExecutor.createQueryReader({ ecsql: definitionsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/visible-sub-categories` });
-        }).pipe(map((row) => {
+        }).pipe(catchBeSQLiteInterrupts, map((row) => {
             return { id: row.id, parentId: row.categoryId };
         }));
     }
@@ -259,7 +259,7 @@ export class CategoriesTreeIdsCache {
             AND m.ECInstanceId IN (SELECT Model.Id FROM ${this.#categoryElementClass})
         `;
             return this.#queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/modeled-elements` });
-        }).pipe(map((row) => {
+        }).pipe(catchBeSQLiteInterrupts, map((row) => {
             return { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };
         }));
     }
@@ -292,20 +292,18 @@ export class CategoriesTreeIdsCache {
         }));
     }
     getSubCategoriesInfo() {
-        this.#subCategoriesInfo ??= this.getModelsCategoriesInfo()
-            .pipe(mergeMap((modelsCategoriesInfo) => from(modelsCategoriesInfo.values())), reduce((acc, modelCategoriesInfo) => {
-            acc.push(...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.subCategoryChildCount > 1).map((categoryInfo) => categoryInfo.id));
-            return acc;
-        }, new Array()), mergeMap((categoriesWithMoreThanOneSubCategory) => {
-            const allSubCategories = new Map();
-            if (categoriesWithMoreThanOneSubCategory.length === 0) {
-                return of(allSubCategories);
+        this.#subCategoriesInfo ??= this.querySubCategories()
+            .pipe(reduce((acc, queriedSubCategory) => {
+            acc.subCategoryCategories.set(queriedSubCategory.id, queriedSubCategory.parentId);
+            const entry = acc.categorySubCategories.get(queriedSubCategory.parentId);
+            if (entry) {
+                entry.push(queriedSubCategory.id);
             }
-            return this.queryVisibleSubCategories(categoriesWithMoreThanOneSubCategory).pipe(reduce((acc, queriedSubCategory) => {
-                acc.set(queriedSubCategory.id, { categoryId: queriedSubCategory.parentId });
-                return acc;
-            }, allSubCategories));
-        }))
+            else {
+                acc.categorySubCategories.set(queriedSubCategory.parentId, [queriedSubCategory.id]);
+            }
+            return acc;
+        }, { subCategoryCategories: new Map(), categorySubCategories: new Map() }))
             .pipe(shareReplay());
         return this.#subCategoriesInfo;
     }
@@ -405,12 +403,16 @@ export class CategoriesTreeIdsCache {
     }
     getInstanceKeyPaths(props) {
         if ("subCategoryId" in props) {
-            return this.getSubCategoriesInfo().pipe(mergeMap((subCategoriesInfo) => {
-                const subCategoryInfo = subCategoriesInfo.get(props.subCategoryId);
-                if (subCategoryInfo === undefined) {
+            return this.getSubCategoriesInfo().pipe(mergeMap(({ subCategoryCategories, categorySubCategories }) => {
+                const categoryOfSubCategory = subCategoryCategories.get(props.subCategoryId);
+                if (categoryOfSubCategory === undefined) {
+                    return of([]);
+                }
+                const subCategories = categorySubCategories.get(categoryOfSubCategory);
+                if (!subCategories || subCategories.length <= 1) {
                     return of([]);
                 }
-                return this.getInstanceKeyPaths({ categoryId: subCategoryInfo.categoryId }).pipe(map((pathToCategory) => [...pathToCategory, { id: props.subCategoryId, className: CLASS_NAME_SubCategory }]));
+                return this.getInstanceKeyPaths({ categoryId: categoryOfSubCategory }).pipe(map((pathToCategory) => [...pathToCategory, { id: props.subCategoryId, className: CLASS_NAME_SubCategory }]));
             }));
         }
         if ("categoryId" in props) {
@@ -472,25 +474,8 @@ export class CategoriesTreeIdsCache {
             }, new Array())),
         });
     }
-    getSubCategories(categoryIds) {
-        return this.getSubCategoriesInfo().pipe(mergeMap((subCategoriesInfo) => subCategoriesInfo.entries()), reduce((acc, [subCategoryId, subCategoryInfo]) => {
-            let entry = acc.get(subCategoryInfo.categoryId);
-            if (!entry) {
-                entry = [];
-                acc.set(subCategoryInfo.categoryId, entry);
-            }
-            entry.push(subCategoryId);
-            return acc;
-        }, new Map()), map((subCategoriesMap) => {
-            const result = new Map();
-            for (const categoryId of Id64.iterable(categoryIds)) {
-                const entry = subCategoriesMap.get(categoryId);
-                if (entry) {
-                    result.set(categoryId, entry);
-                }
-            }
-            return result;
-        }));
+    getSubCategories(categoryId) {
+        return this.getSubCategoriesInfo().pipe(map(({ categorySubCategories }) => categorySubCategories.get(categoryId) ?? []));
     }
     getIsDefinitionContainerSupported() {
         this.#isDefinitionContainerSupported ??= this.queryIsDefinitionContainersSupported().pipe(shareReplay());

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

@@ -1 +1 @@
-{"version":3,"file":"CategoriesTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrG,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,8BAA8B,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,+CAA+C,CAAC;AACzI,OAAO,EAAE,+BAA+B,EAAE,MAAM,0DAA0D,CAAC;AAC3G,OAAO,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AAgC/E,gBAAgB;AAChB,MAAM,OAAO,sBAAsB;IACjC,yBAAyB,CAA8E;IACvG,qBAAqB,CAAuD;IAC5E,wBAAwB,CAAsF;IAC9G,kBAAkB,CAA8D;IACvE,sBAAsB,CAAkC;IACjE,iCAAiC,CAAwE;IACzG,cAAc,CAAS;IACvB,qBAAqB,CAAS;IAC9B,mBAAmB,CAAS;IAC5B,+BAA+B,CAAkC;IACjE,uBAAuB,CAAkD;IACzE,cAAc,CAA6B;IAC3C,YAAY,CAAa;IACzB,cAAc,CAAS;IAEvB,YAAY,aAAyC,EAAE,QAAqB,EAAE,WAAwB;QACpG,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC/E,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,qBAAqB,GAAG,YAAY,CAAC;QAC1C,IAAI,CAAC,mBAAmB,GAAG,UAAU,CAAC;QACtC,IAAI,CAAC,YAAY,GAAG,WAAW,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QACtD,IAAI,CAAC,cAAc,GAAG,wBAAwB,CAAC;QAC/C,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IAC5H,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,2BAA2B,CAAC,kBAA2B;QAI7D,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;eAEL,IAAI,CAAC,qBAAqB;iCACR,WAAW,CAAC,kBAAkB,EAAE,GAAG,CAAC;OAC9D,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB;gBACE,SAAS,EAAE,oBAAoB;gBAC/B,KAAK,EAAE,WAAW;gBAClB,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,4BAA4B,IAAI,CAAC,WAAW,EAAE,EAAE;aAC1G,CACF,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC;QAC9C,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,yBAAyB,CAAC,kBAA2B;QAC1D,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,EAAE,CAAC,IAAI,GAAG,EAAsB,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC,uBAAuB,KAAK,IAAI,CAAC,2BAA2B,CAAC,kBAAkB,CAAC,CAAC,IAAI,CACxF,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,EAAE;YAC9B,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACrB,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAsB,CAAC,EACjC,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;IAEM,2BAA2B;QAChC,IAAI,CAAC,uBAAuB,GAAG,SAAS,CAAC;IAC3C,CAAC;IAEO,2BAA2B;QAIjC,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;eAEL,IAAI,CAAC,mBAAmB;eACxB,IAAI,CAAC,qBAAqB;;;OAGlC,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,gCAAgC,EAAE,CACnJ,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QAC9D,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,eAAe;QAOrB,OAAO,IAAI,CAAC,iCAAiC,EAAE,CAAC,IAAI,CAClD,QAAQ,CAAC,CAAC,8BAA8B,EAAE,EAAE,CAC1C,KAAK,CAAC,GAAG,EAAE;YACT,MAAM,eAAe,GAAG;;;;;gBAMlB,8BAA8B;gBAC5B,CAAC,CAAC;sEACkD,8BAA8B;;;oBAGhF;gBACF,CAAC,CAAC,OACN;;iCAEmB,IAAI,CAAC,qBAAqB;;;;gBAI3C,IAAI,CAAC,cAAc;qBACd,sBAAsB;qBACtB,gBAAgB;;;;;WAK1B,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,eAAe,EAAE,EAC1B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,aAAa,EAAE,CAChI,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO;gBACL,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,+BAA+B,EAAE,GAAG,CAAC,+BAA+B;gBACpE,qBAAqB,EAAE,GAAG,CAAC,qBAAqB;gBAChD,WAAW,EAAE,CAAC,CAAC,GAAG,CAAC,WAAW;aAC/B,CAAC;QACJ,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;IAEO,oCAAoC;QAC1C,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;;;;;;;;OASb,CAAC;YAEF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,oCAAoC,EAAE,CAClG,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,OAAO,EAAE,EACT,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAC/B,CAAC;IACJ,CAAC;IAEO,yBAAyB;QAC/B,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,iKAAiK;YACjK,2EAA2E;YAC3E,2GAA2G;YAC3G,kFAAkF;YAClF,MAAM,yBAAyB,GAAG,sBAAsB,CAAC;YACzD,MAAM,qBAAqB,GAAG,kBAAkB,CAAC;YACjD,MAAM,IAAI,GAAG;gBACX,GAAG,qBAAqB;;;;;qBAKX,IAAI,CAAC,qBAAqB;;;;;cAKjC,IAAI,CAAC,cAAc;;;UAGvB;gBACF;YACI,yBAAyB;;;;;mBAKlB,8BAA8B;mBAC9B,qBAAqB;;;;;;;;;;gBAUxB,yBAAyB;qBACpB,8BAA8B;;;SAG1C;aACF,CAAC;YACF,MAAM,gBAAgB,GAAG;8FAC+D,yBAAyB;OAChH,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,EACjC,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,wBAAwB,EAAE,CAC3I,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;QAC9E,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,yBAAyB,CAAC,WAAsB;QACtD,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,gBAAgB,GAAG;;;;;YAKnB,sBAAsB;;;iCAGD,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC;OAC/C,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAC3B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,yBAAyB,EAAE,CAC5I,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QAClD,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,uBAAuB;QAC7B,IAAI,CAAC,qBAAqB,KAAK,IAAI,CAAC,eAAe,EAAE;aAClD,IAAI,CACH,MAAM,CAAC,CAAC,GAAG,EAAE,eAAe,EAAE,EAAE;YAC9B,IAAI,eAAe,GAAG,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YACvD,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;gBAClC,eAAe,GAAG,EAAE,+BAA+B,EAAE,eAAe,CAAC,+BAA+B,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;gBAC5H,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;YACpD,CAAC;YACD,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC;gBACnC,EAAE,EAAE,eAAe,CAAC,EAAE;gBACtB,qBAAqB,EAAE,eAAe,CAAC,qBAAqB;gBAC5D,WAAW,EAAE,eAAe,CAAC,WAAW;aACzC,CAAC,CAAC;YACH,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAA2B,CAAC,CACvC;aACA,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEO,0BAA0B;QAChC,IAAI,CAAC,wBAAwB,KAAK,QAAQ,CAAC;YACzC,eAAe,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CACtD,MAAM,CAAC,CAAC,GAAG,EAAE,eAAe,EAAE,EAAE;gBAC9B,IAAI,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBAClD,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;oBAC7B,UAAU,GAAG,EAAE,WAAW,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;oBACxC,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;gBAC/C,CAAC;gBACD,UAAU,CAAC,WAAW,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gBACvD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,GAAG,EAAqC,CAAC,CACjD;YACD,gCAAgC,EAAE,IAAI,CAAC,mCAAmC,EAAE;SAC7E,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,EAAE,eAAe,EAAE,gCAAgC,EAAE,EAAE,EAAE;YAC5D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkE,CAAC;YACzF,MAAM,SAAS,GAAG,IAAI,GAAG,EAAc,CAAC;YACxC,gCAAgC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CACvD,WAAW,CAAC,OAAO,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CACjH,CAAC;YACF,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,eAAe,EAAE,CAAC;gBACpD,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC1C,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,UAAU,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC;YAC3E,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EACF,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,wBAAwB,CAAC;IACvC,CAAC;IAEO,oBAAoB;QAM1B,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;;;;iBAKH,IAAI,CAAC,mBAAmB;iBACxB,IAAI,CAAC,qBAAqB;;;0DAGe,IAAI,CAAC,qBAAqB;SAC3E,CAAC;YACJ,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,mBAAmB,EAAE,CACtI,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,cAAc,EAAE,GAAG,CAAC,cAAc,EAAE,CAAC;QAC1I,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,mCAAmC;QACzC,IAAI,CAAC,iCAAiC,KAAK,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CACzE,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,EAAE,EAAE;YACxD,IAAI,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAClC,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,UAAU,GAAG,IAAI,GAAG,EAAE,CAAC;gBACvB,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YAC/B,CAAC;YACD,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACjD,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;YAC1D,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YACtC,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAA4C,CAAC,EACvD,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,iCAAiC,CAAC;IAChD,CAAC;IAEM,4BAA4B,CAAC,OAAmB,EAAE,WAAoB;QAC3E,OAAO,IAAI,CAAC,mCAAmC,EAAE,CAAC,IAAI,CACpD,QAAQ,CAAC,CAAC,gCAAgC,EAAE,EAAE;YAC5C,MAAM,MAAM,GAAG,IAAI,KAAK,EAAa,CAAC;YACtC,MAAM,WAAW,GAAG,gCAAgC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAClE,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;YACpB,CAAC;YACD,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAC1C,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,EAChD,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EACpE,OAAO,EAAE,CACV,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,oBAAoB;QAC1B,IAAI,CAAC,kBAAkB,KAAK,IAAI,CAAC,uBAAuB,EAAE;aACvD,IAAI,CACH,QAAQ,CAAC,CAAC,oBAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,CAAC,EACvE,MAAM,CAAC,CAAC,GAAG,EAAE,mBAAmB,EAAE,EAAE;YAClC,GAAG,CAAC,IAAI,CACN,GAAG,mBAAmB,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,qBAAqB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC,CAC/I,CAAC;YACF,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,KAAK,EAAc,CAAC,EAC3B,QAAQ,CAAC,CAAC,oCAAoC,EAAE,EAAE;YAChD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAA+B,CAAC;YAChE,IAAI,oCAAoC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtD,OAAO,EAAE,CAAC,gBAAgB,CAAC,CAAC;YAC9B,CAAC;YACD,OAAO,IAAI,CAAC,yBAAyB,CAAC,oCAAoC,CAAC,CAAC,IAAI,CAC9E,MAAM,CAAC,CAAC,GAAG,EAAE,kBAAkB,EAAE,EAAE;gBACjC,GAAG,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,kBAAkB,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC5E,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,gBAAgB,CAAC,CACrB,CAAC;QACJ,CAAC,CAAC,CACH;aACA,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAEO,2BAA2B;QACjC,IAAI,CAAC,yBAAyB,KAAK,QAAQ,CAAC;YAC1C,8BAA8B,EAAE,IAAI,CAAC,iCAAiC,EAAE;YACxE,oBAAoB,EAAE,IAAI,CAAC,uBAAuB,EAAE;SACrD,CAAC;aACC,IAAI,CACH,QAAQ,CAAC,CAAC,EAAE,8BAA8B,EAAE,oBAAoB,EAAE,EAAE,EAAE;YACpE,MAAM,wBAAwB,GAAG,IAAI,GAAG,EAAkD,CAAC;YAC3F,IAAI,CAAC,8BAA8B,IAAI,oBAAoB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACvE,OAAO,EAAE,CAAC,wBAAwB,CAAC,CAAC;YACtC,CAAC;YACD,OAAO,IAAI,CAAC,yBAAyB,EAAE,CAAC,IAAI,CAC1C,MAAM,CAAC,CAAC,GAAG,EAAE,0BAA0B,EAAE,EAAE;gBACzC,MAAM,mBAAmB,GAAG,oBAAoB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC;gBACpF,GAAG,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,EAAE;oBACrC,eAAe,EAAE,mBAAmB,EAAE,eAAe,IAAI,EAAE;oBAC3D,OAAO,EAAE,0BAA0B,CAAC,OAAO;oBAC3C,yBAAyB,EAAE,EAAE;oBAC7B,+BAA+B,EAAE,KAAK;oBACtC,WAAW,EAAE,0BAA0B,CAAC,WAAW;iBACpD,CAAC,CAAC;gBACH,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,wBAAwB,CAAC,EAC5B,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;gBACb,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,MAAM,EAAE,CAAC;oBACtE,MAAM,yBAAyB,GAAG,MAAM,CAAC,GAAG,CAAC,uBAAuB,CAAC,OAAO,CAAC,CAAC;oBAC9E,IAAI,yBAAyB,KAAK,SAAS,EAAE,CAAC;wBAC5C,yBAAyB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,qBAAqB,EAAE,WAAW,EAAE,uBAAuB,CAAC,WAAW,EAAE,CAAC,CAAC;wBAC1I,uBAAuB,CAAC,+BAA+B,GAAG,IAAI,CAAC;wBAC/D,yBAAyB,CAAC,WAAW,GAAG,yBAAyB,CAAC,WAAW,IAAI,uBAAuB,CAAC,WAAW,CAAC;oBACvH,CAAC;gBACH,CAAC;gBAED,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CACH;aACA,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC,yBAAyB,CAAC;IACxC,CAAC;IAEM,+CAA+C,CAAC,EACrD,4BAA4B,EAC5B,YAAY,GAIb;QACC,OAAO,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CAC5C,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE,CACpC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,4BAA4B,CAAC,CAAC,CAAC,IAAI,CACpD,MAAM,CACJ,CAAC,GAAG,EAAE,2BAA2B,EAAE,EAAE;YACnC,MAAM,6BAA6B,GAAG,wBAAwB,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;YAChG,IAAI,6BAA6B,KAAK,SAAS,EAAE,CAAC;gBAChD,GAAG,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,6BAA6B,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;gBAChJ,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,6BAA6B,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC,CAAC;YAC3G,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CACzF,CACF,CACF,CACF,CAAC;IACJ,CAAC;IAEM,0BAA0B,CAAC,WAAoB,EAAE,gBAA0B;QAChF,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAC3C,QAAQ,CAAC,CAAC,uBAAuB,EAAE,EAAE,CACnC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CACnC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;YACjB,MAAM,cAAc,GAAG,IAAI,KAAK,EAAW,CAAC;YAC5C,uBAAuB,CAAC,OAAO,CAAC,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE;gBACnF,IAAI,CAAC,gBAAgB,IAAI,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;oBACpE,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;QAC5F,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;IAEM,mBAAmB,CAAC,OAAmB;QAC5C,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAC3C,GAAG,CAAC,CAAC,uBAAuB,EAAE,EAAE;YAC9B,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC;QACxE,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,gBAAgB;QACrB,OAAO,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CACxC,QAAQ,CAAC,CAAC,oBAAoB,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,EACjE,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE;YAClC,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACjD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAc,CAAC,CAC1B,CAAC;IACJ,CAAC;IAEM,WAAW,CAAC,SAAqB;QACtC,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,uBAAuB,EAAE,EAAE,CAAC,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC1H,CAAC;IAEM,yBAAyB,CAAC,EAC/B,sBAAsB,EACtB,sBAAsB,GAIvB;QACC,OAAO,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CAC5C,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE,CACpC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAC9C,QAAQ,CAAC,CAAC,qBAAqB,EAAE,EAAE;YACjC,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;YACpF,IAAI,uBAAuB,KAAK,SAAS,EAAE,CAAC;gBAC1C,OAAO,EAAE,CAAC,EAAE,gBAAgB,EAAE,SAAS,EAAE,kBAAkB,EAAE,SAAS,EAAE,CAAC,CAAC;YAC5E,CAAC;YACD,MAAM,2BAA2B,GAAG,uBAAuB,CAAC,yBAAyB,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;YAC1G,OAAO,CACL,2BAA2B,CAAC,MAAM,GAAG,CAAC;gBACpC,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC;oBAC7B,sBAAsB,EAAE,2BAA2B;oBACnD,sBAAsB;iBACvB,CAAC;gBACJ,CAAC,CAAC,EAAE,CAAC,IAAI,GAAG,EAAc,CAAC,CAC9B,CAAC,IAAI,CACJ,GAAG,CAAC,CAAC,kBAAkB,EAAE,EAAE;gBACzB,OAAO;oBACL,gBAAgB,EAAE,mBAAmB,CAAC,uBAAuB,CAAC,eAAe,EAAE,sBAAsB,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACrI,kBAAkB;iBACnB,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,EACF,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,EAAE,EAAE;YACvD,gBAAgB,EAAE,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YAC/D,kBAAkB,EAAE,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YACjE,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAc,CAAC,CAC1B,CACF,CACF,CAAC;IACJ,CAAC;IAEM,mBAAmB,CACxB,KAAyG;QAEzG,IAAI,eAAe,IAAI,KAAK,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CACrC,QAAQ,CAAC,CAAC,iBAAiB,EAAE,EAAE;gBAC7B,MAAM,eAAe,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;gBACnE,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;oBAClC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;gBAChB,CAAC;gBACD,OAAO,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC,IAAI,CAC9E,GAAG,CAAC,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,GAAG,cAAc,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,sBAAsB,EAAE,CAAC,CAAC,CAC7G,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CACxC,QAAQ,CAAC,CAAC,oBAAoB,EAAE,EAAE;gBAChC,KAAK,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC,IAAI,oBAAoB,EAAE,CAAC;oBAClE,IAAI,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;wBACvG,MAAM,WAAW,GAAG,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC;wBAC7E,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE,CAAC;4BACzD,OAAO,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;wBAC3B,CAAC;wBAED,OAAO,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,OAAO,EAAE,CAAC,CAAC,IAAI,CACtE,GAAG,CAAC,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC,GAAG,yBAAyB,EAAE,WAAW,CAAC,CAAC,CAChF,CAAC;oBACJ,CAAC;gBACH,CAAC;gBACD,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YAChB,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CAC5C,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE;YACpC,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;YAC1F,IAAI,uBAAuB,KAAK,SAAS,EAAE,CAAC;gBAC1C,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YAChB,CAAC;YACD,MAAM,WAAW,GAAG,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,8BAA8B,EAAE,CAAC;YACnG,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE,CAAC;gBAC7D,OAAO,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;YAC3B,CAAC;YACD,OAAO,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,uBAAuB,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAC9F,GAAG,CAAC,CAAC,+BAA+B,EAAE,EAAE,CAAC,CAAC,GAAG,+BAA+B,EAAE,WAAW,CAAC,CAAC,CAC5F,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QACzE,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,uCAAuC,CAAC,KAE9C;QACC,OAAO,QAAQ,CAAC;YACd,UAAU,EAAE,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,oBAAoB,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,EACjE,MAAM,CAAC,CAAC,GAAG,EAAE,mBAAmB,EAAE,EAAE;gBAClC,mBAAmB,CAAC,mBAAmB,CAAC,eAAe,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC;gBACnI,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,KAAK,EAAc,CAAC,CAC5B;YACD,oBAAoB,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CAC3D,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE,CAAC,wBAAwB,CAAC,OAAO,EAAE,CAAC,EAC1E,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,EAAE,EAAE;gBAC/D,IAAI,uBAAuB,CAAC,WAAW,IAAI,CAAC,CAAC,KAAK,EAAE,YAAY,EAAE,CAAC;oBACjE,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;gBAClC,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,KAAK,EAAc,CAAC,CAC5B;SACF,CAAC,CAAC;IACL,CAAC;IAEM,wCAAwC,CAAC,KAE/C;QACC,OAAO,QAAQ,CAAC;YACd,UAAU,EAAE,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,oBAAoB,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,EACjE,MAAM,CAAC,CAAC,GAAG,EAAE,mBAAmB,EAAE,EAAE;gBAClC,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE,CAAC;oBACzD,mBAAmB,CAAC,mBAAmB,CAAC,eAAe,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;gBAClI,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,KAAK,EAAgB,CAAC,CAC9B;YACD,oBAAoB,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CAC3D,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE,CAAC,wBAAwB,CAAC,OAAO,EAAE,CAAC,EAC1E,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,EAAE,EAAE;gBAC/D,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,IAAI,CAAC,uBAAuB,CAAC,WAAW,IAAI,CAAC,CAAC,KAAK,EAAE,YAAY,CAAC,EAAE,CAAC;oBAC/H,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;gBAClC,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,KAAK,EAAc,CAAC,CAC5B;SACF,CAAC,CAAC;IACL,CAAC;IAEM,gBAAgB,CAAC,WAAoB;QAC1C,OAAO,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CACrC,QAAQ,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC,EAC5D,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,aAAa,EAAE,eAAe,CAAC,EAAE,EAAE;YAC/C,IAAI,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;YAChD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,KAAK,GAAG,EAAE,CAAC;gBACX,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAC7C,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC1B,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAiC,CAAC,EAC5C,GAAG,CAAC,CAAC,gBAAgB,EAAE,EAAE;YACvB,MAAM,MAAM,GAAG,IAAI,GAAG,EAAiC,CAAC;YACxD,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;gBACpD,MAAM,KAAK,GAAG,gBAAgB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBAC/C,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,iCAAiC;QACtC,IAAI,CAAC,+BAA+B,KAAK,IAAI,CAAC,oCAAoC,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACzG,OAAO,IAAI,CAAC,+BAA+B,CAAC;IAC9C,CAAC;CACF;AAED,SAAS,mBAAmB,CAAsC,IAAS,EAAE,YAAiC;IAC5G,OAAO,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;AAC/E,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 { defer, EMPTY, forkJoin, from, map, mergeMap, of, reduce, shareReplay, toArray } from \"rxjs\";\nimport { Guid, Id64 } from \"@itwin/core-bentley\";\nimport { CLASS_NAME_DefinitionContainer, CLASS_NAME_Model, CLASS_NAME_SubCategory } from \"../../common/internal/ClassNameDefinitions.js\";\nimport { ModelCategoryElementsCountCache } from \"../../common/internal/ModelCategoryElementsCountCache.js\";\nimport { getClassesByView, joinId64Arg } from \"../../common/internal/Utils.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString, Id64Arg, Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\nimport type { CategoryId, DefinitionContainerId, ElementId, ModelId, SubCategoryId } from \"../../common/internal/Types.js\";\n\ninterface DefinitionContainerInfo {\n  modelId: Id64String;\n  parentDefinitionContainerExists: boolean;\n  childCategories: CategoryInfo[];\n  childDefinitionContainers: Array<{ id: Id64String; hasElements: boolean }>;\n  hasElements: boolean;\n}\n\ninterface CategoriesInfo {\n  childCategories: CategoryInfo[];\n  parentDefinitionContainerExists: boolean;\n}\n\n/** @internal */\nexport interface CategoryInfo {\n  id: CategoryId;\n  subCategoryChildCount: number;\n  hasElements: boolean;\n}\n\ninterface SubCategoryInfo {\n  categoryId: Id64String;\n}\n\n/** @internal */\nexport class CategoriesTreeIdsCache implements Disposable {\n  #definitionContainersInfo: Observable<Map<DefinitionContainerId, DefinitionContainerInfo>> | undefined;\n  #modelsCategoriesInfo: Observable<Map<ModelId, CategoriesInfo>> | undefined;\n  #elementModelsCategories: Observable<Map<ModelId, { categoryIds: Id64Set; isSubModel: boolean }>> | undefined;\n  #subCategoriesInfo: Observable<Map<SubCategoryId, SubCategoryInfo>> | undefined;\n  readonly #categoryElementCounts: ModelCategoryElementsCountCache;\n  #modelWithCategoryModeledElements: Observable<Map<ModelId, Map<CategoryId, Set<ElementId>>>> | undefined;\n  #categoryClass: string;\n  #categoryElementClass: string;\n  #categoryModelClass: string;\n  #isDefinitionContainerSupported: Observable<boolean> | undefined;\n  #filteredElementsModels: Observable<Map<ElementId, ModelId>> | undefined;\n  #queryExecutor: LimitingECSqlQueryExecutor;\n  #componentId: GuidString;\n  #componentName: string;\n\n  constructor(queryExecutor: LimitingECSqlQueryExecutor, viewType: \"3d\" | \"2d\", componentId?: GuidString) {\n    this.#queryExecutor = queryExecutor;\n    const { categoryClass, elementClass, modelClass } = getClassesByView(viewType);\n    this.#categoryClass = categoryClass;\n    this.#categoryElementClass = elementClass;\n    this.#categoryModelClass = modelClass;\n    this.#componentId = componentId ?? Guid.createValue();\n    this.#componentName = \"CategoriesTreeIdsCache\";\n    this.#categoryElementCounts = new ModelCategoryElementsCountCache(this.#queryExecutor, [elementClass], this.#componentId);\n  }\n\n  public [Symbol.dispose]() {\n    this.#categoryElementCounts[Symbol.dispose]();\n  }\n\n  private queryFilteredElementsModels(filteredElementIds: Id64Arg): Observable<{\n    modelId: Id64String;\n    id: ElementId;\n  }> {\n    return defer(() => {\n      const query = `\n        SELECT Model.Id modelId, ECInstanceId id\n        FROM ${this.#categoryElementClass}\n        WHERE ECInstanceId IN (${joinId64Arg(filteredElementIds, \",\")})\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: query },\n        {\n          rowFormat: \"ECSqlPropertyNames\",\n          limit: \"unbounded\",\n          restartToken: `${this.#componentName}/${this.#componentId}/filtered-element-models/${Guid.createValue()}`,\n        },\n      );\n    }).pipe(\n      map((row) => {\n        return { modelId: row.modelId, id: row.id };\n      }),\n    );\n  }\n\n  public getFilteredElementsModels(filteredElementIds: Id64Arg) {\n    if (Id64.sizeOf(filteredElementIds) === 0) {\n      return of(new Map<ElementId, ModelId>());\n    }\n\n    this.#filteredElementsModels ??= this.queryFilteredElementsModels(filteredElementIds).pipe(\n      reduce((acc, { modelId, id }) => {\n        acc.set(id, modelId);\n        return acc;\n      }, new Map<ElementId, ModelId>()),\n      shareReplay(),\n    );\n    return this.#filteredElementsModels;\n  }\n\n  public clearFilteredElementsModels() {\n    this.#filteredElementsModels = undefined;\n  }\n\n  private queryElementModelCategories(): Observable<{\n    modelId: Id64String;\n    categoryId: Id64String;\n  }> {\n    return defer(() => {\n      const query = `\n        SELECT this.Model.Id modelId, this.Category.Id categoryId\n        FROM ${this.#categoryModelClass} m\n        JOIN ${this.#categoryElementClass} this ON m.ECInstanceId = this.Model.Id\n        WHERE this.Parent.Id IS NULL AND m.IsPrivate = false\n        GROUP BY modelId, categoryId\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: query },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/element-models-and-categories` },\n      );\n    }).pipe(\n      map((row) => {\n        return { modelId: row.modelId, categoryId: row.categoryId };\n      }),\n    );\n  }\n\n  private queryCategories(): Observable<{\n    id: CategoryId;\n    modelId: Id64String;\n    parentDefinitionContainerExists: boolean;\n    subCategoryChildCount: number;\n    hasElements: boolean;\n  }> {\n    return this.getIsDefinitionContainerSupported().pipe(\n      mergeMap((isDefinitionContainerSupported) =>\n        defer(() => {\n          const categoriesQuery = `\n            SELECT\n              this.ECInstanceId id,\n              COUNT(sc.ECInstanceId) subCategoryChildCount,\n              this.Model.Id modelId,\n              ${\n                isDefinitionContainerSupported\n                  ? `\n                  IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${CLASS_NAME_DefinitionContainer} dc),\n                    true,\n                    false\n                  )`\n                  : \"false\"\n              } parentDefinitionContainerExists,\n              IFNULL(\n                (SELECT 1 FROM ${this.#categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId LIMIT 1),\n                0\n              ) hasElements\n            FROM\n              ${this.#categoryClass} this\n              JOIN ${CLASS_NAME_SubCategory} sc ON sc.Parent.Id = this.ECInstanceId\n              JOIN ${CLASS_NAME_Model} m ON m.ECInstanceId = this.Model.Id\n            WHERE\n              NOT this.IsPrivate\n              AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))\n            GROUP BY this.ECInstanceId\n          `;\n          return this.#queryExecutor.createQueryReader(\n            { ecsql: categoriesQuery },\n            { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/categories` },\n          );\n        }).pipe(\n          map((row) => {\n            return {\n              id: row.id,\n              modelId: row.modelId,\n              parentDefinitionContainerExists: row.parentDefinitionContainerExists,\n              subCategoryChildCount: row.subCategoryChildCount,\n              hasElements: !!row.hasElements,\n            };\n          }),\n        ),\n      ),\n    );\n  }\n\n  private queryIsDefinitionContainersSupported(): Observable<boolean> {\n    return defer(() => {\n      const query = `\n        SELECT\n          1\n        FROM\n          ECDbMeta.ECSchemaDef s\n          JOIN ECDbMeta.ECClassDef c ON c.Schema.Id = s.ECInstanceId\n        WHERE\n          s.Name = 'BisCore'\n          AND c.Name = 'DefinitionContainer'\n      `;\n\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: query },\n        { restartToken: `${this.#componentName}/${this.#componentId}/is-definition-container-supported` },\n      );\n    }).pipe(\n      toArray(),\n      map((rows) => rows.length > 0),\n    );\n  }\n\n  private queryDefinitionContainers(): Observable<{ id: DefinitionContainerId; modelId: Id64String; hasElements: boolean }> {\n    return defer(() => {\n      // DefinitionModel ECInstanceId will always be the same as modeled DefinitionContainer ECInstanceId, if this wasn't the case, we would need to do something like:\n      //  JOIN BisCore.DefinitionModel dm ON dm.ECInstanceId = ${modelIdAccessor}\n      //  JOIN BisCore.DefinitionModelBreaksDownDefinitionContainer dr ON dr.SourceECInstanceId = dm.ECInstanceId\n      //  JOIN BisCore.DefinitionContainer dc ON dc.ECInstanceId = dr.TargetECInstanceId\n      const DEFINITION_CONTAINERS_CTE = \"DefinitionContainers\";\n      const CATEGORIES_MODELS_CTE = \"CategoriesModels\";\n      const ctes = [\n        `${CATEGORIES_MODELS_CTE}(ModelId, HasElements) AS (\n          SELECT\n            c.Model.Id,\n            IFNULL((\n              SELECT 1\n              FROM ${this.#categoryElementClass} e\n              WHERE e.Category.Id = c.ECInstanceId\n              LIMIT 1\n            ), 0)\n          FROM\n            ${this.#categoryClass} c\n          WHERE\n            NOT c.IsPrivate\n        )`,\n        `\n          ${DEFINITION_CONTAINERS_CTE}(ECInstanceId, ModelId, HasElements) AS (\n            SELECT\n              dc.ECInstanceId,\n              dc.Model.Id,\n              c.HasElements\n            FROM ${CLASS_NAME_DefinitionContainer} dc\n            JOIN ${CATEGORIES_MODELS_CTE} c ON dc.ECInstanceId = c.ModelId\n            WHERE NOT dc.IsPrivate\n\n            UNION ALL\n\n            SELECT\n              pdc.ECInstanceId,\n              pdc.Model.Id,\n              cdc.HasElements\n            FROM\n              ${DEFINITION_CONTAINERS_CTE} cdc\n              JOIN ${CLASS_NAME_DefinitionContainer} pdc ON pdc.ECInstanceId = cdc.ModelId\n            WHERE NOT pdc.IsPrivate\n          )\n        `,\n      ];\n      const definitionsQuery = `\n        SELECT dc.ECInstanceId id, dc.ModelId modelId, MAX(dc.HasElements) hasElements FROM ${DEFINITION_CONTAINERS_CTE} dc GROUP BY dc.ECInstanceId\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ctes, ecsql: definitionsQuery },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/definition-containers` },\n      );\n    }).pipe(\n      map((row) => {\n        return { id: row.id, modelId: row.modelId, hasElements: !!row.hasElements };\n      }),\n    );\n  }\n\n  private queryVisibleSubCategories(categoryIds: Id64Array): Observable<{ id: SubCategoryId; parentId: CategoryId }> {\n    return defer(() => {\n      const definitionsQuery = `\n      SELECT\n          sc.ECInstanceId id,\n          sc.Parent.Id categoryId\n        FROM\n          ${CLASS_NAME_SubCategory} sc\n        WHERE\n          NOT sc.IsPrivate\n          AND sc.Parent.Id IN (${categoryIds.join(\",\")})\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: definitionsQuery },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/visible-sub-categories` },\n      );\n    }).pipe(\n      map((row) => {\n        return { id: row.id, parentId: row.categoryId };\n      }),\n    );\n  }\n\n  private getModelsCategoriesInfo() {\n    this.#modelsCategoriesInfo ??= this.queryCategories()\n      .pipe(\n        reduce((acc, queriedCategory) => {\n          let modelCategories = acc.get(queriedCategory.modelId);\n          if (modelCategories === undefined) {\n            modelCategories = { parentDefinitionContainerExists: queriedCategory.parentDefinitionContainerExists, childCategories: [] };\n            acc.set(queriedCategory.modelId, modelCategories);\n          }\n          modelCategories.childCategories.push({\n            id: queriedCategory.id,\n            subCategoryChildCount: queriedCategory.subCategoryChildCount,\n            hasElements: queriedCategory.hasElements,\n          });\n          return acc;\n        }, new Map<ModelId, CategoriesInfo>()),\n      )\n      .pipe(shareReplay());\n    return this.#modelsCategoriesInfo;\n  }\n\n  private getElementModelsCategories() {\n    this.#elementModelsCategories ??= forkJoin({\n      modelCategories: this.queryElementModelCategories().pipe(\n        reduce((acc, queriedCategory) => {\n          let modelEntry = acc.get(queriedCategory.modelId);\n          if (modelEntry === undefined) {\n            modelEntry = { categoryIds: new Set() };\n            acc.set(queriedCategory.modelId, modelEntry);\n          }\n          modelEntry.categoryIds.add(queriedCategory.categoryId);\n          return acc;\n        }, new Map<ModelId, { categoryIds: Id64Set }>()),\n      ),\n      modelWithCategoryModeledElements: this.getModelWithCategoryModeledElements(),\n    }).pipe(\n      map(({ modelCategories, modelWithCategoryModeledElements }) => {\n        const result = new Map<ModelId, { categoryIds: Set<CategoryId>; isSubModel: boolean }>();\n        const subModels = new Set<Id64String>();\n        modelWithCategoryModeledElements.forEach((categoryMap) =>\n          categoryMap.forEach((categorySubModels) => categorySubModels.forEach((subModelId) => subModels.add(subModelId))),\n        );\n        for (const [modelId, modelEntry] of modelCategories) {\n          const isSubModel = subModels.has(modelId);\n          result.set(modelId, { categoryIds: modelEntry.categoryIds, isSubModel });\n        }\n        return result;\n      }),\n      shareReplay(),\n    );\n    return this.#elementModelsCategories;\n  }\n\n  private queryModeledElements(): Observable<{\n    modelId: Id64String;\n    modeledElementId: Id64String;\n    categoryId: Id64String;\n    rootCategoryId: Id64String;\n  }> {\n    return defer(() => {\n      const query = `\n          SELECT\n            pe.ECInstanceId modeledElementId,\n            pe.Category.Id categoryId,\n            pe.Model.Id modelId\n          FROM ${this.#categoryModelClass} m\n          JOIN ${this.#categoryElementClass} pe ON pe.ECInstanceId = m.ModeledElement.Id\n          WHERE\n            m.IsPrivate = false\n            AND m.ECInstanceId IN (SELECT Model.Id FROM ${this.#categoryElementClass})\n        `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: query },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/modeled-elements` },\n      );\n    }).pipe(\n      map((row) => {\n        return { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };\n      }),\n    );\n  }\n\n  private getModelWithCategoryModeledElements() {\n    this.#modelWithCategoryModeledElements ??= this.queryModeledElements().pipe(\n      reduce((acc, { modelId, categoryId, modeledElementId }) => {\n        let modelEntry = acc.get(modelId);\n        if (!modelEntry) {\n          modelEntry = new Map();\n          acc.set(modelId, modelEntry);\n        }\n        const categoryEntry = modelEntry.get(categoryId);\n        if (!categoryEntry) {\n          modelEntry.set(categoryId, new Set([modeledElementId]));\n        } else {\n          categoryEntry.add(modeledElementId);\n        }\n        return acc;\n      }, new Map<ModelId, Map<CategoryId, Set<ElementId>>>()),\n      shareReplay(),\n    );\n    return this.#modelWithCategoryModeledElements;\n  }\n\n  public getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Arg): Observable<Id64Array> {\n    return this.getModelWithCategoryModeledElements().pipe(\n      mergeMap((modelWithCategoryModeledElements) => {\n        const result = new Array<ElementId>();\n        const categoryMap = modelWithCategoryModeledElements.get(modelId);\n        if (!categoryMap) {\n          return of(result);\n        }\n        return from(Id64.iterable(categoryIds)).pipe(\n          map((categoryId) => categoryMap.get(categoryId)),\n          mergeMap((elementsSet) => (elementsSet ? from(elementsSet) : EMPTY)),\n          toArray(),\n        );\n      }),\n    );\n  }\n\n  private getSubCategoriesInfo() {\n    this.#subCategoriesInfo ??= this.getModelsCategoriesInfo()\n      .pipe(\n        mergeMap((modelsCategoriesInfo) => from(modelsCategoriesInfo.values())),\n        reduce((acc, modelCategoriesInfo) => {\n          acc.push(\n            ...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.subCategoryChildCount > 1).map((categoryInfo) => categoryInfo.id),\n          );\n          return acc;\n        }, new Array<CategoryId>()),\n        mergeMap((categoriesWithMoreThanOneSubCategory) => {\n          const allSubCategories = new Map<Id64String, SubCategoryInfo>();\n          if (categoriesWithMoreThanOneSubCategory.length === 0) {\n            return of(allSubCategories);\n          }\n          return this.queryVisibleSubCategories(categoriesWithMoreThanOneSubCategory).pipe(\n            reduce((acc, queriedSubCategory) => {\n              acc.set(queriedSubCategory.id, { categoryId: queriedSubCategory.parentId });\n              return acc;\n            }, allSubCategories),\n          );\n        }),\n      )\n      .pipe(shareReplay());\n    return this.#subCategoriesInfo;\n  }\n\n  private getDefinitionContainersInfo() {\n    this.#definitionContainersInfo ??= forkJoin({\n      isDefinitionContainerSupported: this.getIsDefinitionContainerSupported(),\n      modelsCategoriesInfo: this.getModelsCategoriesInfo(),\n    })\n      .pipe(\n        mergeMap(({ isDefinitionContainerSupported, modelsCategoriesInfo }) => {\n          const definitionContainersInfo = new Map<DefinitionContainerId, DefinitionContainerInfo>();\n          if (!isDefinitionContainerSupported || modelsCategoriesInfo.size === 0) {\n            return of(definitionContainersInfo);\n          }\n          return this.queryDefinitionContainers().pipe(\n            reduce((acc, queriedDefinitionContainer) => {\n              const modelCategoriesInfo = modelsCategoriesInfo.get(queriedDefinitionContainer.id);\n              acc.set(queriedDefinitionContainer.id, {\n                childCategories: modelCategoriesInfo?.childCategories ?? [],\n                modelId: queriedDefinitionContainer.modelId,\n                childDefinitionContainers: [],\n                parentDefinitionContainerExists: false,\n                hasElements: queriedDefinitionContainer.hasElements,\n              });\n              return acc;\n            }, definitionContainersInfo),\n            map((result) => {\n              for (const [definitionContainerId, definitionContainerInfo] of result) {\n                const parentDefinitionContainer = result.get(definitionContainerInfo.modelId);\n                if (parentDefinitionContainer !== undefined) {\n                  parentDefinitionContainer.childDefinitionContainers.push({ id: definitionContainerId, hasElements: definitionContainerInfo.hasElements });\n                  definitionContainerInfo.parentDefinitionContainerExists = true;\n                  parentDefinitionContainer.hasElements = parentDefinitionContainer.hasElements || definitionContainerInfo.hasElements;\n                }\n              }\n\n              return result;\n            }),\n          );\n        }),\n      )\n      .pipe(shareReplay());\n    return this.#definitionContainersInfo;\n  }\n\n  public getDirectChildDefinitionContainersAndCategories({\n    parentDefinitionContainerIds,\n    includeEmpty,\n  }: {\n    parentDefinitionContainerIds: Id64Arg;\n    includeEmpty?: boolean;\n  }): Observable<{ categories: CategoryInfo[]; definitionContainers: Array<DefinitionContainerId> }> {\n    return this.getDefinitionContainersInfo().pipe(\n      mergeMap((definitionContainersInfo) =>\n        from(Id64.iterable(parentDefinitionContainerIds)).pipe(\n          reduce(\n            (acc, parentDefinitionContainerId) => {\n              const parentDefinitionContainerInfo = definitionContainersInfo.get(parentDefinitionContainerId);\n              if (parentDefinitionContainerInfo !== undefined) {\n                acc.definitionContainers.push(...applyElementsFilter(parentDefinitionContainerInfo.childDefinitionContainers, includeEmpty).map((dc) => dc.id));\n                acc.categories.push(...applyElementsFilter(parentDefinitionContainerInfo.childCategories, includeEmpty));\n              }\n              return acc;\n            },\n            { definitionContainers: new Array<Id64String>(), categories: new Array<CategoryInfo>() },\n          ),\n        ),\n      ),\n    );\n  }\n\n  public getCategoriesElementModels(categoryIds: Id64Arg, includeSubModels?: boolean): Observable<{ id: CategoryId; models: Array<ModelId> | undefined }> {\n    return this.getElementModelsCategories().pipe(\n      mergeMap((elementModelsCategories) =>\n        from(Id64.iterable(categoryIds)).pipe(\n          map((categoryId) => {\n            const categoryModels = new Array<ModelId>();\n            elementModelsCategories.forEach(({ categoryIds: categories, isSubModel }, modelId) => {\n              if ((includeSubModels || !isSubModel) && categories.has(categoryId)) {\n                categoryModels.push(modelId);\n              }\n            });\n            return { id: categoryId, models: categoryModels.length > 0 ? categoryModels : undefined };\n          }),\n        ),\n      ),\n    );\n  }\n\n  public getModelCategoryIds(modelId: Id64String): Observable<Id64Array> {\n    return this.getElementModelsCategories().pipe(\n      map((elementModelsCategories) => {\n        return [...(elementModelsCategories.get(modelId)?.categoryIds ?? [])];\n      }),\n    );\n  }\n\n  public getAllCategories(): Observable<Id64Set> {\n    return this.getModelsCategoriesInfo().pipe(\n      mergeMap((modelsCategoriesInfo) => modelsCategoriesInfo.values()),\n      reduce((acc, { childCategories }) => {\n        childCategories.forEach(({ id }) => acc.add(id));\n        return acc;\n      }, new Set<Id64String>()),\n    );\n  }\n\n  public hasSubModel(elementId: Id64String): Observable<boolean> {\n    return this.getElementModelsCategories().pipe(map((elementModelsCategories) => elementModelsCategories.has(elementId)));\n  }\n\n  public getAllContainedCategories({\n    definitionContainerIds,\n    includeEmptyCategories,\n  }: {\n    definitionContainerIds: Id64Arg;\n    includeEmptyCategories?: boolean;\n  }): Observable<Id64Set> {\n    return this.getDefinitionContainersInfo().pipe(\n      mergeMap((definitionContainersInfo) =>\n        from(Id64.iterable(definitionContainerIds)).pipe(\n          mergeMap((definitionContainerId) => {\n            const definitionContainerInfo = definitionContainersInfo.get(definitionContainerId);\n            if (definitionContainerInfo === undefined) {\n              return of({ directCategories: undefined, indirectCategories: undefined });\n            }\n            const childDefinitionContainerIds = definitionContainerInfo.childDefinitionContainers.map(({ id }) => id);\n            return (\n              childDefinitionContainerIds.length > 0\n                ? this.getAllContainedCategories({\n                    definitionContainerIds: childDefinitionContainerIds,\n                    includeEmptyCategories,\n                  })\n                : of(new Set<CategoryId>())\n            ).pipe(\n              map((indirectCategories) => {\n                return {\n                  directCategories: applyElementsFilter(definitionContainerInfo.childCategories, includeEmptyCategories).map((category) => category.id),\n                  indirectCategories,\n                };\n              }),\n            );\n          }),\n          reduce((acc, { directCategories, indirectCategories }) => {\n            directCategories?.forEach((categoryId) => acc.add(categoryId));\n            indirectCategories?.forEach((categoryId) => acc.add(categoryId));\n            return acc;\n          }, new Set<CategoryId>()),\n        ),\n      ),\n    );\n  }\n\n  public getInstanceKeyPaths(\n    props: { categoryId: Id64String } | { definitionContainerId: Id64String } | { subCategoryId: Id64String },\n  ): Observable<InstanceKey[]> {\n    if (\"subCategoryId\" in props) {\n      return this.getSubCategoriesInfo().pipe(\n        mergeMap((subCategoriesInfo) => {\n          const subCategoryInfo = subCategoriesInfo.get(props.subCategoryId);\n          if (subCategoryInfo === undefined) {\n            return of([]);\n          }\n          return this.getInstanceKeyPaths({ categoryId: subCategoryInfo.categoryId }).pipe(\n            map((pathToCategory) => [...pathToCategory, { id: props.subCategoryId, className: CLASS_NAME_SubCategory }]),\n          );\n        }),\n      );\n    }\n\n    if (\"categoryId\" in props) {\n      return this.getModelsCategoriesInfo().pipe(\n        mergeMap((modelsCategoriesInfo) => {\n          for (const [modelId, modelCategoriesInfo] of modelsCategoriesInfo) {\n            if (modelCategoriesInfo.childCategories.find((childCategory) => childCategory.id === props.categoryId)) {\n              const instanceKey = { id: props.categoryId, className: this.#categoryClass };\n              if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n                return of([instanceKey]);\n              }\n\n              return this.getInstanceKeyPaths({ definitionContainerId: modelId }).pipe(\n                map((pathToDefinitionContainer) => [...pathToDefinitionContainer, instanceKey]),\n              );\n            }\n          }\n          return of([]);\n        }),\n      );\n    }\n    return this.getDefinitionContainersInfo().pipe(\n      mergeMap((definitionContainersInfo) => {\n        const definitionContainerInfo = definitionContainersInfo.get(props.definitionContainerId);\n        if (definitionContainerInfo === undefined) {\n          return of([]);\n        }\n        const instanceKey = { id: props.definitionContainerId, className: CLASS_NAME_DefinitionContainer };\n        if (!definitionContainerInfo.parentDefinitionContainerExists) {\n          return of([instanceKey]);\n        }\n        return this.getInstanceKeyPaths({ definitionContainerId: definitionContainerInfo.modelId }).pipe(\n          map((pathToParentDefinitionContainer) => [...pathToParentDefinitionContainer, instanceKey]),\n        );\n      }),\n    );\n  }\n\n  public getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Observable<number> {\n    return this.#categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  public getAllDefinitionContainersAndCategories(props?: {\n    includeEmpty?: boolean;\n  }): Observable<{ categories: Array<CategoryId>; definitionContainers: Array<DefinitionContainerId> }> {\n    return forkJoin({\n      categories: this.getModelsCategoriesInfo().pipe(\n        mergeMap((modelsCategoriesInfo) => modelsCategoriesInfo.values()),\n        reduce((acc, modelCategoriesInfo) => {\n          applyElementsFilter(modelCategoriesInfo.childCategories, props?.includeEmpty).forEach((categoryInfo) => acc.push(categoryInfo.id));\n          return acc;\n        }, new Array<Id64String>()),\n      ),\n      definitionContainers: this.getDefinitionContainersInfo().pipe(\n        mergeMap((definitionContainersInfo) => definitionContainersInfo.entries()),\n        reduce((acc, [definitionContainerId, definitionContainerInfo]) => {\n          if (definitionContainerInfo.hasElements || !!props?.includeEmpty) {\n            acc.push(definitionContainerId);\n          }\n          return acc;\n        }, new Array<Id64String>()),\n      ),\n    });\n  }\n\n  public getRootDefinitionContainersAndCategories(props?: {\n    includeEmpty?: boolean;\n  }): Observable<{ categories: CategoryInfo[]; definitionContainers: Array<DefinitionContainerId> }> {\n    return forkJoin({\n      categories: this.getModelsCategoriesInfo().pipe(\n        mergeMap((modelsCategoriesInfo) => modelsCategoriesInfo.values()),\n        reduce((acc, modelCategoriesInfo) => {\n          if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n            applyElementsFilter(modelCategoriesInfo.childCategories, props?.includeEmpty).forEach((categoryInfo) => acc.push(categoryInfo));\n          }\n          return acc;\n        }, new Array<CategoryInfo>()),\n      ),\n      definitionContainers: this.getDefinitionContainersInfo().pipe(\n        mergeMap((definitionContainersInfo) => definitionContainersInfo.entries()),\n        reduce((acc, [definitionContainerId, definitionContainerInfo]) => {\n          if (!definitionContainerInfo.parentDefinitionContainerExists && (definitionContainerInfo.hasElements || !!props?.includeEmpty)) {\n            acc.push(definitionContainerId);\n          }\n          return acc;\n        }, new Array<Id64String>()),\n      ),\n    });\n  }\n\n  public getSubCategories(categoryIds: Id64Arg): Observable<Map<CategoryId, Array<SubCategoryId>>> {\n    return this.getSubCategoriesInfo().pipe(\n      mergeMap((subCategoriesInfo) => subCategoriesInfo.entries()),\n      reduce((acc, [subCategoryId, subCategoryInfo]) => {\n        let entry = acc.get(subCategoryInfo.categoryId);\n        if (!entry) {\n          entry = [];\n          acc.set(subCategoryInfo.categoryId, entry);\n        }\n        entry.push(subCategoryId);\n        return acc;\n      }, new Map<CategoryId, Array<Id64String>>()),\n      map((subCategoriesMap) => {\n        const result = new Map<CategoryId, Array<Id64String>>();\n        for (const categoryId of Id64.iterable(categoryIds)) {\n          const entry = subCategoriesMap.get(categoryId);\n          if (entry) {\n            result.set(categoryId, entry);\n          }\n        }\n        return result;\n      }),\n    );\n  }\n\n  public getIsDefinitionContainerSupported(): Observable<boolean> {\n    this.#isDefinitionContainerSupported ??= this.queryIsDefinitionContainersSupported().pipe(shareReplay());\n    return this.#isDefinitionContainerSupported;\n  }\n}\n\nfunction applyElementsFilter<T extends { hasElements?: boolean }>(list: T[], includeEmpty: boolean | undefined): T[] {\n  return includeEmpty ? list : list.filter(({ hasElements }) => !!hasElements);\n}\n"]}
+{"version":3,"file":"CategoriesTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrG,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,8BAA8B,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,+CAA+C,CAAC;AACzI,OAAO,EAAE,+BAA+B,EAAE,MAAM,0DAA0D,CAAC;AAC3G,OAAO,EAAE,uBAAuB,EAAE,MAAM,wCAAwC,CAAC;AACjF,OAAO,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AA4B/E,gBAAgB;AAChB,MAAM,OAAO,sBAAsB;IACjC,yBAAyB,CAA8E;IACvG,qBAAqB,CAAuD;IAC5E,wBAAwB,CAAsF;IAC9G,kBAAkB,CAEJ;IACL,sBAAsB,CAAkC;IACjE,iCAAiC,CAAwE;IACzG,cAAc,CAAS;IACvB,qBAAqB,CAAS;IAC9B,mBAAmB,CAAS;IAC5B,+BAA+B,CAAkC;IACjE,uBAAuB,CAAkD;IACzE,cAAc,CAA6B;IAC3C,YAAY,CAAa;IACzB,cAAc,CAAS;IAEvB,YAAY,aAAyC,EAAE,QAAqB,EAAE,WAAwB;QACpG,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC/E,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,qBAAqB,GAAG,YAAY,CAAC;QAC1C,IAAI,CAAC,mBAAmB,GAAG,UAAU,CAAC;QACtC,IAAI,CAAC,YAAY,GAAG,WAAW,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QACtD,IAAI,CAAC,cAAc,GAAG,wBAAwB,CAAC;QAC/C,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IAC5H,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,2BAA2B,CAAC,kBAA2B;QAI7D,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;eAEL,IAAI,CAAC,qBAAqB;iCACR,WAAW,CAAC,kBAAkB,EAAE,GAAG,CAAC;OAC9D,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB;gBACE,SAAS,EAAE,oBAAoB;gBAC/B,KAAK,EAAE,WAAW;gBAClB,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,4BAA4B,IAAI,CAAC,WAAW,EAAE,EAAE;aAC1G,CACF,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC;QAC9C,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,yBAAyB,CAAC,kBAA2B;QAC1D,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,EAAE,CAAC,IAAI,GAAG,EAAsB,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC,uBAAuB,KAAK,IAAI,CAAC,2BAA2B,CAAC,kBAAkB,CAAC,CAAC,IAAI,CACxF,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,EAAE;YAC9B,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACrB,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAsB,CAAC,EACjC,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;IAEM,2BAA2B;QAChC,IAAI,CAAC,uBAAuB,GAAG,SAAS,CAAC;IAC3C,CAAC;IAEO,2BAA2B;QAIjC,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;eAEL,IAAI,CAAC,mBAAmB;eACxB,IAAI,CAAC,qBAAqB;;;OAGlC,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,gCAAgC,EAAE,CACnJ,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QAC9D,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,eAAe;QAOrB,OAAO,IAAI,CAAC,iCAAiC,EAAE,CAAC,IAAI,CAClD,QAAQ,CAAC,CAAC,8BAA8B,EAAE,EAAE,CAC1C,KAAK,CAAC,GAAG,EAAE;YACT,MAAM,eAAe,GAAG;;;;;gBAMlB,8BAA8B;gBAC5B,CAAC,CAAC;sEACkD,8BAA8B;;;oBAGhF;gBACF,CAAC,CAAC,OACN;;iCAEmB,IAAI,CAAC,qBAAqB;;;;gBAI3C,IAAI,CAAC,cAAc;qBACd,sBAAsB;qBACtB,gBAAgB;;;;;WAK1B,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,eAAe,EAAE,EAC1B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,aAAa,EAAE,CAChI,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO;gBACL,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,+BAA+B,EAAE,GAAG,CAAC,+BAA+B;gBACpE,qBAAqB,EAAE,GAAG,CAAC,qBAAqB;gBAChD,WAAW,EAAE,CAAC,CAAC,GAAG,CAAC,WAAW;aAC/B,CAAC;QACJ,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;IAEO,oCAAoC;QAC1C,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;;;;;;;;OASb,CAAC;YAEF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,oCAAoC,EAAE,CAClG,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,OAAO,EAAE,EACT,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAC/B,CAAC;IACJ,CAAC;IAEO,yBAAyB;QAC/B,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,iKAAiK;YACjK,2EAA2E;YAC3E,2GAA2G;YAC3G,kFAAkF;YAClF,MAAM,yBAAyB,GAAG,sBAAsB,CAAC;YACzD,MAAM,qBAAqB,GAAG,kBAAkB,CAAC;YACjD,MAAM,IAAI,GAAG;gBACX,GAAG,qBAAqB;;;;;qBAKX,IAAI,CAAC,qBAAqB;;;;;cAKjC,IAAI,CAAC,cAAc;;;UAGvB;gBACF;YACI,yBAAyB;;;;;mBAKlB,8BAA8B;mBAC9B,qBAAqB;;;;;;;;;;gBAUxB,yBAAyB;qBACpB,8BAA8B;;;SAG1C;aACF,CAAC;YACF,MAAM,gBAAgB,GAAG;8FAC+D,yBAAyB;OAChH,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,EACjC,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,wBAAwB,EAAE,CAC3I,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;QAC9E,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,kBAAkB;QACxB,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,gBAAgB,GAAG;;;;;YAKnB,sBAAsB;;;OAG3B,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAC3B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,yBAAyB,EAAE,CAC5I,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QAClD,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,uBAAuB;QAC7B,IAAI,CAAC,qBAAqB,KAAK,IAAI,CAAC,eAAe,EAAE;aAClD,IAAI,CACH,MAAM,CAAC,CAAC,GAAG,EAAE,eAAe,EAAE,EAAE;YAC9B,IAAI,eAAe,GAAG,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YACvD,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;gBAClC,eAAe,GAAG,EAAE,+BAA+B,EAAE,eAAe,CAAC,+BAA+B,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;gBAC5H,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;YACpD,CAAC;YACD,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC;gBACnC,EAAE,EAAE,eAAe,CAAC,EAAE;gBACtB,qBAAqB,EAAE,eAAe,CAAC,qBAAqB;gBAC5D,WAAW,EAAE,eAAe,CAAC,WAAW;aACzC,CAAC,CAAC;YACH,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAA2B,CAAC,CACvC;aACA,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEO,0BAA0B;QAChC,IAAI,CAAC,wBAAwB,KAAK,QAAQ,CAAC;YACzC,eAAe,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CACtD,MAAM,CAAC,CAAC,GAAG,EAAE,eAAe,EAAE,EAAE;gBAC9B,IAAI,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBAClD,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;oBAC7B,UAAU,GAAG,EAAE,WAAW,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;oBACxC,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;gBAC/C,CAAC;gBACD,UAAU,CAAC,WAAW,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gBACvD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,GAAG,EAAqC,CAAC,CACjD;YACD,gCAAgC,EAAE,IAAI,CAAC,mCAAmC,EAAE;SAC7E,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,EAAE,eAAe,EAAE,gCAAgC,EAAE,EAAE,EAAE;YAC5D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkE,CAAC;YACzF,MAAM,SAAS,GAAG,IAAI,GAAG,EAAc,CAAC;YACxC,gCAAgC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CACvD,WAAW,CAAC,OAAO,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CACjH,CAAC;YACF,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,eAAe,EAAE,CAAC;gBACpD,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC1C,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,UAAU,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC;YAC3E,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EACF,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,wBAAwB,CAAC;IACvC,CAAC;IAEO,oBAAoB;QAM1B,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;;;;iBAKH,IAAI,CAAC,mBAAmB;iBACxB,IAAI,CAAC,qBAAqB;;;0DAGe,IAAI,CAAC,qBAAqB;SAC3E,CAAC;YACJ,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,mBAAmB,EAAE,CACtI,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,cAAc,EAAE,GAAG,CAAC,cAAc,EAAE,CAAC;QAC1I,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,mCAAmC;QACzC,IAAI,CAAC,iCAAiC,KAAK,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CACzE,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,EAAE,EAAE;YACxD,IAAI,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAClC,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,UAAU,GAAG,IAAI,GAAG,EAAE,CAAC;gBACvB,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YAC/B,CAAC;YACD,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACjD,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;YAC1D,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YACtC,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAA4C,CAAC,EACvD,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,iCAAiC,CAAC;IAChD,CAAC;IAEM,4BAA4B,CAAC,OAAmB,EAAE,WAAoB;QAC3E,OAAO,IAAI,CAAC,mCAAmC,EAAE,CAAC,IAAI,CACpD,QAAQ,CAAC,CAAC,gCAAgC,EAAE,EAAE;YAC5C,MAAM,MAAM,GAAG,IAAI,KAAK,EAAa,CAAC;YACtC,MAAM,WAAW,GAAG,gCAAgC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAClE,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;YACpB,CAAC;YACD,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAC1C,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,EAChD,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EACpE,OAAO,EAAE,CACV,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,oBAAoB;QAC1B,IAAI,CAAC,kBAAkB,KAAK,IAAI,CAAC,kBAAkB,EAAE;aAClD,IAAI,CACH,MAAM,CACJ,CAAC,GAAG,EAAE,kBAAkB,EAAE,EAAE;YAC1B,GAAG,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,EAAE,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClF,MAAM,KAAK,GAAG,GAAG,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YACzE,IAAI,KAAK,EAAE,CAAC;gBACV,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC;YACtF,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,qBAAqB,EAAE,IAAI,GAAG,EAA6B,EAAE,qBAAqB,EAAE,IAAI,GAAG,EAAoC,EAAE,CACpI,CACF;aACA,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAEO,2BAA2B;QACjC,IAAI,CAAC,yBAAyB,KAAK,QAAQ,CAAC;YAC1C,8BAA8B,EAAE,IAAI,CAAC,iCAAiC,EAAE;YACxE,oBAAoB,EAAE,IAAI,CAAC,uBAAuB,EAAE;SACrD,CAAC;aACC,IAAI,CACH,QAAQ,CAAC,CAAC,EAAE,8BAA8B,EAAE,oBAAoB,EAAE,EAAE,EAAE;YACpE,MAAM,wBAAwB,GAAG,IAAI,GAAG,EAAkD,CAAC;YAC3F,IAAI,CAAC,8BAA8B,IAAI,oBAAoB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACvE,OAAO,EAAE,CAAC,wBAAwB,CAAC,CAAC;YACtC,CAAC;YACD,OAAO,IAAI,CAAC,yBAAyB,EAAE,CAAC,IAAI,CAC1C,MAAM,CAAC,CAAC,GAAG,EAAE,0BAA0B,EAAE,EAAE;gBACzC,MAAM,mBAAmB,GAAG,oBAAoB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC;gBACpF,GAAG,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,EAAE;oBACrC,eAAe,EAAE,mBAAmB,EAAE,eAAe,IAAI,EAAE;oBAC3D,OAAO,EAAE,0BAA0B,CAAC,OAAO;oBAC3C,yBAAyB,EAAE,EAAE;oBAC7B,+BAA+B,EAAE,KAAK;oBACtC,WAAW,EAAE,0BAA0B,CAAC,WAAW;iBACpD,CAAC,CAAC;gBACH,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,wBAAwB,CAAC,EAC5B,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;gBACb,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,MAAM,EAAE,CAAC;oBACtE,MAAM,yBAAyB,GAAG,MAAM,CAAC,GAAG,CAAC,uBAAuB,CAAC,OAAO,CAAC,CAAC;oBAC9E,IAAI,yBAAyB,KAAK,SAAS,EAAE,CAAC;wBAC5C,yBAAyB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,qBAAqB,EAAE,WAAW,EAAE,uBAAuB,CAAC,WAAW,EAAE,CAAC,CAAC;wBAC1I,uBAAuB,CAAC,+BAA+B,GAAG,IAAI,CAAC;wBAC/D,yBAAyB,CAAC,WAAW,GAAG,yBAAyB,CAAC,WAAW,IAAI,uBAAuB,CAAC,WAAW,CAAC;oBACvH,CAAC;gBACH,CAAC;gBAED,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CACH;aACA,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC,yBAAyB,CAAC;IACxC,CAAC;IAEM,+CAA+C,CAAC,EACrD,4BAA4B,EAC5B,YAAY,GAIb;QACC,OAAO,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CAC5C,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE,CACpC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,4BAA4B,CAAC,CAAC,CAAC,IAAI,CACpD,MAAM,CACJ,CAAC,GAAG,EAAE,2BAA2B,EAAE,EAAE;YACnC,MAAM,6BAA6B,GAAG,wBAAwB,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;YAChG,IAAI,6BAA6B,KAAK,SAAS,EAAE,CAAC;gBAChD,GAAG,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,6BAA6B,CAAC,yBAAyB,EAAE,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;gBAChJ,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,6BAA6B,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC,CAAC;YAC3G,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CACzF,CACF,CACF,CACF,CAAC;IACJ,CAAC;IAEM,0BAA0B,CAAC,WAAoB,EAAE,gBAA0B;QAChF,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAC3C,QAAQ,CAAC,CAAC,uBAAuB,EAAE,EAAE,CACnC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CACnC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;YACjB,MAAM,cAAc,GAAG,IAAI,KAAK,EAAW,CAAC;YAC5C,uBAAuB,CAAC,OAAO,CAAC,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE;gBACnF,IAAI,CAAC,gBAAgB,IAAI,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;oBACpE,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;QAC5F,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;IAEM,mBAAmB,CAAC,OAAmB;QAC5C,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAC3C,GAAG,CAAC,CAAC,uBAAuB,EAAE,EAAE;YAC9B,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC;QACxE,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,gBAAgB;QACrB,OAAO,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CACxC,QAAQ,CAAC,CAAC,oBAAoB,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,EACjE,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE;YAClC,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACjD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAc,CAAC,CAC1B,CAAC;IACJ,CAAC;IAEM,WAAW,CAAC,SAAqB;QACtC,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,uBAAuB,EAAE,EAAE,CAAC,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC1H,CAAC;IAEM,yBAAyB,CAAC,EAC/B,sBAAsB,EACtB,sBAAsB,GAIvB;QACC,OAAO,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CAC5C,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE,CACpC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAC9C,QAAQ,CAAC,CAAC,qBAAqB,EAAE,EAAE;YACjC,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;YACpF,IAAI,uBAAuB,KAAK,SAAS,EAAE,CAAC;gBAC1C,OAAO,EAAE,CAAC,EAAE,gBAAgB,EAAE,SAAS,EAAE,kBAAkB,EAAE,SAAS,EAAE,CAAC,CAAC;YAC5E,CAAC;YACD,MAAM,2BAA2B,GAAG,uBAAuB,CAAC,yBAAyB,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;YAC1G,OAAO,CACL,2BAA2B,CAAC,MAAM,GAAG,CAAC;gBACpC,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC;oBAC7B,sBAAsB,EAAE,2BAA2B;oBACnD,sBAAsB;iBACvB,CAAC;gBACJ,CAAC,CAAC,EAAE,CAAC,IAAI,GAAG,EAAc,CAAC,CAC9B,CAAC,IAAI,CACJ,GAAG,CAAC,CAAC,kBAAkB,EAAE,EAAE;gBACzB,OAAO;oBACL,gBAAgB,EAAE,mBAAmB,CAAC,uBAAuB,CAAC,eAAe,EAAE,sBAAsB,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACrI,kBAAkB;iBACnB,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,EACF,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,EAAE,EAAE;YACvD,gBAAgB,EAAE,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YAC/D,kBAAkB,EAAE,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YACjE,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAc,CAAC,CAC1B,CACF,CACF,CAAC;IACJ,CAAC;IAEM,mBAAmB,CACxB,KAAyG;QAEzG,IAAI,eAAe,IAAI,KAAK,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CACrC,QAAQ,CAAC,CAAC,EAAE,qBAAqB,EAAE,qBAAqB,EAAE,EAAE,EAAE;gBAC5D,MAAM,qBAAqB,GAAG,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;gBAC7E,IAAI,qBAAqB,KAAK,SAAS,EAAE,CAAC;oBACxC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;gBAChB,CAAC;gBACD,MAAM,aAAa,GAAG,qBAAqB,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;gBACvE,IAAI,CAAC,aAAa,IAAI,aAAa,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;oBAChD,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;gBAChB,CAAC;gBACD,OAAO,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,qBAAqB,EAAE,CAAC,CAAC,IAAI,CACzE,GAAG,CAAC,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,GAAG,cAAc,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,sBAAsB,EAAE,CAAC,CAAC,CAC7G,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CACxC,QAAQ,CAAC,CAAC,oBAAoB,EAAE,EAAE;gBAChC,KAAK,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC,IAAI,oBAAoB,EAAE,CAAC;oBAClE,IAAI,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;wBACvG,MAAM,WAAW,GAAG,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC;wBAC7E,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE,CAAC;4BACzD,OAAO,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;wBAC3B,CAAC;wBAED,OAAO,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,OAAO,EAAE,CAAC,CAAC,IAAI,CACtE,GAAG,CAAC,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC,GAAG,yBAAyB,EAAE,WAAW,CAAC,CAAC,CAChF,CAAC;oBACJ,CAAC;gBACH,CAAC;gBACD,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YAChB,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CAC5C,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE;YACpC,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;YAC1F,IAAI,uBAAuB,KAAK,SAAS,EAAE,CAAC;gBAC1C,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YAChB,CAAC;YACD,MAAM,WAAW,GAAG,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,8BAA8B,EAAE,CAAC;YACnG,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE,CAAC;gBAC7D,OAAO,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;YAC3B,CAAC;YACD,OAAO,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,uBAAuB,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAC9F,GAAG,CAAC,CAAC,+BAA+B,EAAE,EAAE,CAAC,CAAC,GAAG,+BAA+B,EAAE,WAAW,CAAC,CAAC,CAC5F,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QACzE,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,uCAAuC,CAAC,KAE9C;QACC,OAAO,QAAQ,CAAC;YACd,UAAU,EAAE,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,oBAAoB,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,EACjE,MAAM,CAAC,CAAC,GAAG,EAAE,mBAAmB,EAAE,EAAE;gBAClC,mBAAmB,CAAC,mBAAmB,CAAC,eAAe,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC;gBACnI,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,KAAK,EAAc,CAAC,CAC5B;YACD,oBAAoB,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CAC3D,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE,CAAC,wBAAwB,CAAC,OAAO,EAAE,CAAC,EAC1E,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,EAAE,EAAE;gBAC/D,IAAI,uBAAuB,CAAC,WAAW,IAAI,CAAC,CAAC,KAAK,EAAE,YAAY,EAAE,CAAC;oBACjE,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;gBAClC,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,KAAK,EAAc,CAAC,CAC5B;SACF,CAAC,CAAC;IACL,CAAC;IAEM,wCAAwC,CAAC,KAE/C;QACC,OAAO,QAAQ,CAAC;YACd,UAAU,EAAE,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,oBAAoB,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,EACjE,MAAM,CAAC,CAAC,GAAG,EAAE,mBAAmB,EAAE,EAAE;gBAClC,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE,CAAC;oBACzD,mBAAmB,CAAC,mBAAmB,CAAC,eAAe,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;gBAClI,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,KAAK,EAAgB,CAAC,CAC9B;YACD,oBAAoB,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CAC3D,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE,CAAC,wBAAwB,CAAC,OAAO,EAAE,CAAC,EAC1E,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,EAAE,EAAE;gBAC/D,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,IAAI,CAAC,uBAAuB,CAAC,WAAW,IAAI,CAAC,CAAC,KAAK,EAAE,YAAY,CAAC,EAAE,CAAC;oBAC/H,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;gBAClC,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,KAAK,EAAc,CAAC,CAC5B;SACF,CAAC,CAAC;IACL,CAAC;IAEM,gBAAgB,CAAC,UAAsB;QAC5C,OAAO,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,qBAAqB,EAAE,EAAE,EAAE,CAAC,qBAAqB,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC3H,CAAC;IAEM,iCAAiC;QACtC,IAAI,CAAC,+BAA+B,KAAK,IAAI,CAAC,oCAAoC,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACzG,OAAO,IAAI,CAAC,+BAA+B,CAAC;IAC9C,CAAC;CACF;AAED,SAAS,mBAAmB,CAAsC,IAAS,EAAE,YAAiC;IAC5G,OAAO,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;AAC/E,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 { defer, EMPTY, forkJoin, from, map, mergeMap, of, reduce, shareReplay, toArray } from \"rxjs\";\nimport { Guid, Id64 } from \"@itwin/core-bentley\";\nimport { CLASS_NAME_DefinitionContainer, CLASS_NAME_Model, CLASS_NAME_SubCategory } from \"../../common/internal/ClassNameDefinitions.js\";\nimport { ModelCategoryElementsCountCache } from \"../../common/internal/ModelCategoryElementsCountCache.js\";\nimport { catchBeSQLiteInterrupts } from \"../../common/internal/UseErrorState.js\";\nimport { getClassesByView, joinId64Arg } from \"../../common/internal/Utils.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString, Id64Arg, Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\nimport type { CategoryId, DefinitionContainerId, ElementId, ModelId, SubCategoryId } from \"../../common/internal/Types.js\";\n\ninterface DefinitionContainerInfo {\n  modelId: Id64String;\n  parentDefinitionContainerExists: boolean;\n  childCategories: CategoryInfo[];\n  childDefinitionContainers: Array<{ id: Id64String; hasElements: boolean }>;\n  hasElements: boolean;\n}\n\ninterface CategoriesInfo {\n  childCategories: CategoryInfo[];\n  parentDefinitionContainerExists: boolean;\n}\n\n/** @internal */\nexport interface CategoryInfo {\n  id: CategoryId;\n  subCategoryChildCount: number;\n  hasElements: boolean;\n}\n\n/** @internal */\nexport class CategoriesTreeIdsCache implements Disposable {\n  #definitionContainersInfo: Observable<Map<DefinitionContainerId, DefinitionContainerInfo>> | undefined;\n  #modelsCategoriesInfo: Observable<Map<ModelId, CategoriesInfo>> | undefined;\n  #elementModelsCategories: Observable<Map<ModelId, { categoryIds: Id64Set; isSubModel: boolean }>> | undefined;\n  #subCategoriesInfo:\n    | Observable<{ subCategoryCategories: Map<SubCategoryId, CategoryId>; categorySubCategories: Map<CategoryId, Array<SubCategoryId>> }>\n    | undefined;\n  readonly #categoryElementCounts: ModelCategoryElementsCountCache;\n  #modelWithCategoryModeledElements: Observable<Map<ModelId, Map<CategoryId, Set<ElementId>>>> | undefined;\n  #categoryClass: string;\n  #categoryElementClass: string;\n  #categoryModelClass: string;\n  #isDefinitionContainerSupported: Observable<boolean> | undefined;\n  #filteredElementsModels: Observable<Map<ElementId, ModelId>> | undefined;\n  #queryExecutor: LimitingECSqlQueryExecutor;\n  #componentId: GuidString;\n  #componentName: string;\n\n  constructor(queryExecutor: LimitingECSqlQueryExecutor, viewType: \"3d\" | \"2d\", componentId?: GuidString) {\n    this.#queryExecutor = queryExecutor;\n    const { categoryClass, elementClass, modelClass } = getClassesByView(viewType);\n    this.#categoryClass = categoryClass;\n    this.#categoryElementClass = elementClass;\n    this.#categoryModelClass = modelClass;\n    this.#componentId = componentId ?? Guid.createValue();\n    this.#componentName = \"CategoriesTreeIdsCache\";\n    this.#categoryElementCounts = new ModelCategoryElementsCountCache(this.#queryExecutor, [elementClass], this.#componentId);\n  }\n\n  public [Symbol.dispose]() {\n    this.#categoryElementCounts[Symbol.dispose]();\n  }\n\n  private queryFilteredElementsModels(filteredElementIds: Id64Arg): Observable<{\n    modelId: Id64String;\n    id: ElementId;\n  }> {\n    return defer(() => {\n      const query = `\n        SELECT Model.Id modelId, ECInstanceId id\n        FROM ${this.#categoryElementClass}\n        WHERE ECInstanceId IN (${joinId64Arg(filteredElementIds, \",\")})\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: query },\n        {\n          rowFormat: \"ECSqlPropertyNames\",\n          limit: \"unbounded\",\n          restartToken: `${this.#componentName}/${this.#componentId}/filtered-element-models/${Guid.createValue()}`,\n        },\n      );\n    }).pipe(\n      catchBeSQLiteInterrupts,\n      map((row) => {\n        return { modelId: row.modelId, id: row.id };\n      }),\n    );\n  }\n\n  public getFilteredElementsModels(filteredElementIds: Id64Arg) {\n    if (Id64.sizeOf(filteredElementIds) === 0) {\n      return of(new Map<ElementId, ModelId>());\n    }\n\n    this.#filteredElementsModels ??= this.queryFilteredElementsModels(filteredElementIds).pipe(\n      reduce((acc, { modelId, id }) => {\n        acc.set(id, modelId);\n        return acc;\n      }, new Map<ElementId, ModelId>()),\n      shareReplay(),\n    );\n    return this.#filteredElementsModels;\n  }\n\n  public clearFilteredElementsModels() {\n    this.#filteredElementsModels = undefined;\n  }\n\n  private queryElementModelCategories(): Observable<{\n    modelId: Id64String;\n    categoryId: Id64String;\n  }> {\n    return defer(() => {\n      const query = `\n        SELECT this.Model.Id modelId, this.Category.Id categoryId\n        FROM ${this.#categoryModelClass} m\n        JOIN ${this.#categoryElementClass} this ON m.ECInstanceId = this.Model.Id\n        WHERE this.Parent.Id IS NULL AND m.IsPrivate = false\n        GROUP BY modelId, categoryId\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: query },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/element-models-and-categories` },\n      );\n    }).pipe(\n      catchBeSQLiteInterrupts,\n      map((row) => {\n        return { modelId: row.modelId, categoryId: row.categoryId };\n      }),\n    );\n  }\n\n  private queryCategories(): Observable<{\n    id: CategoryId;\n    modelId: Id64String;\n    parentDefinitionContainerExists: boolean;\n    subCategoryChildCount: number;\n    hasElements: boolean;\n  }> {\n    return this.getIsDefinitionContainerSupported().pipe(\n      mergeMap((isDefinitionContainerSupported) =>\n        defer(() => {\n          const categoriesQuery = `\n            SELECT\n              this.ECInstanceId id,\n              COUNT(sc.ECInstanceId) subCategoryChildCount,\n              this.Model.Id modelId,\n              ${\n                isDefinitionContainerSupported\n                  ? `\n                  IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${CLASS_NAME_DefinitionContainer} dc),\n                    true,\n                    false\n                  )`\n                  : \"false\"\n              } parentDefinitionContainerExists,\n              IFNULL(\n                (SELECT 1 FROM ${this.#categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId LIMIT 1),\n                0\n              ) hasElements\n            FROM\n              ${this.#categoryClass} this\n              JOIN ${CLASS_NAME_SubCategory} sc ON sc.Parent.Id = this.ECInstanceId\n              JOIN ${CLASS_NAME_Model} m ON m.ECInstanceId = this.Model.Id\n            WHERE\n              NOT this.IsPrivate\n              AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))\n            GROUP BY this.ECInstanceId\n          `;\n          return this.#queryExecutor.createQueryReader(\n            { ecsql: categoriesQuery },\n            { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/categories` },\n          );\n        }).pipe(\n          catchBeSQLiteInterrupts,\n          map((row) => {\n            return {\n              id: row.id,\n              modelId: row.modelId,\n              parentDefinitionContainerExists: row.parentDefinitionContainerExists,\n              subCategoryChildCount: row.subCategoryChildCount,\n              hasElements: !!row.hasElements,\n            };\n          }),\n        ),\n      ),\n    );\n  }\n\n  private queryIsDefinitionContainersSupported(): Observable<boolean> {\n    return defer(() => {\n      const query = `\n        SELECT\n          1\n        FROM\n          ECDbMeta.ECSchemaDef s\n          JOIN ECDbMeta.ECClassDef c ON c.Schema.Id = s.ECInstanceId\n        WHERE\n          s.Name = 'BisCore'\n          AND c.Name = 'DefinitionContainer'\n      `;\n\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: query },\n        { restartToken: `${this.#componentName}/${this.#componentId}/is-definition-container-supported` },\n      );\n    }).pipe(\n      catchBeSQLiteInterrupts,\n      toArray(),\n      map((rows) => rows.length > 0),\n    );\n  }\n\n  private queryDefinitionContainers(): Observable<{ id: DefinitionContainerId; modelId: Id64String; hasElements: boolean }> {\n    return defer(() => {\n      // DefinitionModel ECInstanceId will always be the same as modeled DefinitionContainer ECInstanceId, if this wasn't the case, we would need to do something like:\n      //  JOIN BisCore.DefinitionModel dm ON dm.ECInstanceId = ${modelIdAccessor}\n      //  JOIN BisCore.DefinitionModelBreaksDownDefinitionContainer dr ON dr.SourceECInstanceId = dm.ECInstanceId\n      //  JOIN BisCore.DefinitionContainer dc ON dc.ECInstanceId = dr.TargetECInstanceId\n      const DEFINITION_CONTAINERS_CTE = \"DefinitionContainers\";\n      const CATEGORIES_MODELS_CTE = \"CategoriesModels\";\n      const ctes = [\n        `${CATEGORIES_MODELS_CTE}(ModelId, HasElements) AS (\n          SELECT\n            c.Model.Id,\n            IFNULL((\n              SELECT 1\n              FROM ${this.#categoryElementClass} e\n              WHERE e.Category.Id = c.ECInstanceId\n              LIMIT 1\n            ), 0)\n          FROM\n            ${this.#categoryClass} c\n          WHERE\n            NOT c.IsPrivate\n        )`,\n        `\n          ${DEFINITION_CONTAINERS_CTE}(ECInstanceId, ModelId, HasElements) AS (\n            SELECT\n              dc.ECInstanceId,\n              dc.Model.Id,\n              c.HasElements\n            FROM ${CLASS_NAME_DefinitionContainer} dc\n            JOIN ${CATEGORIES_MODELS_CTE} c ON dc.ECInstanceId = c.ModelId\n            WHERE NOT dc.IsPrivate\n\n            UNION ALL\n\n            SELECT\n              pdc.ECInstanceId,\n              pdc.Model.Id,\n              cdc.HasElements\n            FROM\n              ${DEFINITION_CONTAINERS_CTE} cdc\n              JOIN ${CLASS_NAME_DefinitionContainer} pdc ON pdc.ECInstanceId = cdc.ModelId\n            WHERE NOT pdc.IsPrivate\n          )\n        `,\n      ];\n      const definitionsQuery = `\n        SELECT dc.ECInstanceId id, dc.ModelId modelId, MAX(dc.HasElements) hasElements FROM ${DEFINITION_CONTAINERS_CTE} dc GROUP BY dc.ECInstanceId\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ctes, ecsql: definitionsQuery },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/definition-containers` },\n      );\n    }).pipe(\n      catchBeSQLiteInterrupts,\n      map((row) => {\n        return { id: row.id, modelId: row.modelId, hasElements: !!row.hasElements };\n      }),\n    );\n  }\n\n  private querySubCategories(): Observable<{ id: SubCategoryId; parentId: CategoryId }> {\n    return defer(() => {\n      const definitionsQuery = `\n        SELECT\n          sc.ECInstanceId id,\n          sc.Parent.Id categoryId\n        FROM\n          ${CLASS_NAME_SubCategory} sc\n        WHERE\n          NOT sc.IsPrivate\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: definitionsQuery },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/visible-sub-categories` },\n      );\n    }).pipe(\n      catchBeSQLiteInterrupts,\n      map((row) => {\n        return { id: row.id, parentId: row.categoryId };\n      }),\n    );\n  }\n\n  private getModelsCategoriesInfo() {\n    this.#modelsCategoriesInfo ??= this.queryCategories()\n      .pipe(\n        reduce((acc, queriedCategory) => {\n          let modelCategories = acc.get(queriedCategory.modelId);\n          if (modelCategories === undefined) {\n            modelCategories = { parentDefinitionContainerExists: queriedCategory.parentDefinitionContainerExists, childCategories: [] };\n            acc.set(queriedCategory.modelId, modelCategories);\n          }\n          modelCategories.childCategories.push({\n            id: queriedCategory.id,\n            subCategoryChildCount: queriedCategory.subCategoryChildCount,\n            hasElements: queriedCategory.hasElements,\n          });\n          return acc;\n        }, new Map<ModelId, CategoriesInfo>()),\n      )\n      .pipe(shareReplay());\n    return this.#modelsCategoriesInfo;\n  }\n\n  private getElementModelsCategories() {\n    this.#elementModelsCategories ??= forkJoin({\n      modelCategories: this.queryElementModelCategories().pipe(\n        reduce((acc, queriedCategory) => {\n          let modelEntry = acc.get(queriedCategory.modelId);\n          if (modelEntry === undefined) {\n            modelEntry = { categoryIds: new Set() };\n            acc.set(queriedCategory.modelId, modelEntry);\n          }\n          modelEntry.categoryIds.add(queriedCategory.categoryId);\n          return acc;\n        }, new Map<ModelId, { categoryIds: Id64Set }>()),\n      ),\n      modelWithCategoryModeledElements: this.getModelWithCategoryModeledElements(),\n    }).pipe(\n      map(({ modelCategories, modelWithCategoryModeledElements }) => {\n        const result = new Map<ModelId, { categoryIds: Set<CategoryId>; isSubModel: boolean }>();\n        const subModels = new Set<Id64String>();\n        modelWithCategoryModeledElements.forEach((categoryMap) =>\n          categoryMap.forEach((categorySubModels) => categorySubModels.forEach((subModelId) => subModels.add(subModelId))),\n        );\n        for (const [modelId, modelEntry] of modelCategories) {\n          const isSubModel = subModels.has(modelId);\n          result.set(modelId, { categoryIds: modelEntry.categoryIds, isSubModel });\n        }\n        return result;\n      }),\n      shareReplay(),\n    );\n    return this.#elementModelsCategories;\n  }\n\n  private queryModeledElements(): Observable<{\n    modelId: Id64String;\n    modeledElementId: Id64String;\n    categoryId: Id64String;\n    rootCategoryId: Id64String;\n  }> {\n    return defer(() => {\n      const query = `\n          SELECT\n            pe.ECInstanceId modeledElementId,\n            pe.Category.Id categoryId,\n            pe.Model.Id modelId\n          FROM ${this.#categoryModelClass} m\n          JOIN ${this.#categoryElementClass} pe ON pe.ECInstanceId = m.ModeledElement.Id\n          WHERE\n            m.IsPrivate = false\n            AND m.ECInstanceId IN (SELECT Model.Id FROM ${this.#categoryElementClass})\n        `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: query },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/modeled-elements` },\n      );\n    }).pipe(\n      catchBeSQLiteInterrupts,\n      map((row) => {\n        return { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };\n      }),\n    );\n  }\n\n  private getModelWithCategoryModeledElements() {\n    this.#modelWithCategoryModeledElements ??= this.queryModeledElements().pipe(\n      reduce((acc, { modelId, categoryId, modeledElementId }) => {\n        let modelEntry = acc.get(modelId);\n        if (!modelEntry) {\n          modelEntry = new Map();\n          acc.set(modelId, modelEntry);\n        }\n        const categoryEntry = modelEntry.get(categoryId);\n        if (!categoryEntry) {\n          modelEntry.set(categoryId, new Set([modeledElementId]));\n        } else {\n          categoryEntry.add(modeledElementId);\n        }\n        return acc;\n      }, new Map<ModelId, Map<CategoryId, Set<ElementId>>>()),\n      shareReplay(),\n    );\n    return this.#modelWithCategoryModeledElements;\n  }\n\n  public getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Arg): Observable<Id64Array> {\n    return this.getModelWithCategoryModeledElements().pipe(\n      mergeMap((modelWithCategoryModeledElements) => {\n        const result = new Array<ElementId>();\n        const categoryMap = modelWithCategoryModeledElements.get(modelId);\n        if (!categoryMap) {\n          return of(result);\n        }\n        return from(Id64.iterable(categoryIds)).pipe(\n          map((categoryId) => categoryMap.get(categoryId)),\n          mergeMap((elementsSet) => (elementsSet ? from(elementsSet) : EMPTY)),\n          toArray(),\n        );\n      }),\n    );\n  }\n\n  private getSubCategoriesInfo() {\n    this.#subCategoriesInfo ??= this.querySubCategories()\n      .pipe(\n        reduce(\n          (acc, queriedSubCategory) => {\n            acc.subCategoryCategories.set(queriedSubCategory.id, queriedSubCategory.parentId);\n            const entry = acc.categorySubCategories.get(queriedSubCategory.parentId);\n            if (entry) {\n              entry.push(queriedSubCategory.id);\n            } else {\n              acc.categorySubCategories.set(queriedSubCategory.parentId, [queriedSubCategory.id]);\n            }\n            return acc;\n          },\n          { subCategoryCategories: new Map<SubCategoryId, CategoryId>(), categorySubCategories: new Map<CategoryId, Array<SubCategoryId>>() },\n        ),\n      )\n      .pipe(shareReplay());\n    return this.#subCategoriesInfo;\n  }\n\n  private getDefinitionContainersInfo() {\n    this.#definitionContainersInfo ??= forkJoin({\n      isDefinitionContainerSupported: this.getIsDefinitionContainerSupported(),\n      modelsCategoriesInfo: this.getModelsCategoriesInfo(),\n    })\n      .pipe(\n        mergeMap(({ isDefinitionContainerSupported, modelsCategoriesInfo }) => {\n          const definitionContainersInfo = new Map<DefinitionContainerId, DefinitionContainerInfo>();\n          if (!isDefinitionContainerSupported || modelsCategoriesInfo.size === 0) {\n            return of(definitionContainersInfo);\n          }\n          return this.queryDefinitionContainers().pipe(\n            reduce((acc, queriedDefinitionContainer) => {\n              const modelCategoriesInfo = modelsCategoriesInfo.get(queriedDefinitionContainer.id);\n              acc.set(queriedDefinitionContainer.id, {\n                childCategories: modelCategoriesInfo?.childCategories ?? [],\n                modelId: queriedDefinitionContainer.modelId,\n                childDefinitionContainers: [],\n                parentDefinitionContainerExists: false,\n                hasElements: queriedDefinitionContainer.hasElements,\n              });\n              return acc;\n            }, definitionContainersInfo),\n            map((result) => {\n              for (const [definitionContainerId, definitionContainerInfo] of result) {\n                const parentDefinitionContainer = result.get(definitionContainerInfo.modelId);\n                if (parentDefinitionContainer !== undefined) {\n                  parentDefinitionContainer.childDefinitionContainers.push({ id: definitionContainerId, hasElements: definitionContainerInfo.hasElements });\n                  definitionContainerInfo.parentDefinitionContainerExists = true;\n                  parentDefinitionContainer.hasElements = parentDefinitionContainer.hasElements || definitionContainerInfo.hasElements;\n                }\n              }\n\n              return result;\n            }),\n          );\n        }),\n      )\n      .pipe(shareReplay());\n    return this.#definitionContainersInfo;\n  }\n\n  public getDirectChildDefinitionContainersAndCategories({\n    parentDefinitionContainerIds,\n    includeEmpty,\n  }: {\n    parentDefinitionContainerIds: Id64Arg;\n    includeEmpty?: boolean;\n  }): Observable<{ categories: CategoryInfo[]; definitionContainers: Array<DefinitionContainerId> }> {\n    return this.getDefinitionContainersInfo().pipe(\n      mergeMap((definitionContainersInfo) =>\n        from(Id64.iterable(parentDefinitionContainerIds)).pipe(\n          reduce(\n            (acc, parentDefinitionContainerId) => {\n              const parentDefinitionContainerInfo = definitionContainersInfo.get(parentDefinitionContainerId);\n              if (parentDefinitionContainerInfo !== undefined) {\n                acc.definitionContainers.push(...applyElementsFilter(parentDefinitionContainerInfo.childDefinitionContainers, includeEmpty).map((dc) => dc.id));\n                acc.categories.push(...applyElementsFilter(parentDefinitionContainerInfo.childCategories, includeEmpty));\n              }\n              return acc;\n            },\n            { definitionContainers: new Array<Id64String>(), categories: new Array<CategoryInfo>() },\n          ),\n        ),\n      ),\n    );\n  }\n\n  public getCategoriesElementModels(categoryIds: Id64Arg, includeSubModels?: boolean): Observable<{ id: CategoryId; models: Array<ModelId> | undefined }> {\n    return this.getElementModelsCategories().pipe(\n      mergeMap((elementModelsCategories) =>\n        from(Id64.iterable(categoryIds)).pipe(\n          map((categoryId) => {\n            const categoryModels = new Array<ModelId>();\n            elementModelsCategories.forEach(({ categoryIds: categories, isSubModel }, modelId) => {\n              if ((includeSubModels || !isSubModel) && categories.has(categoryId)) {\n                categoryModels.push(modelId);\n              }\n            });\n            return { id: categoryId, models: categoryModels.length > 0 ? categoryModels : undefined };\n          }),\n        ),\n      ),\n    );\n  }\n\n  public getModelCategoryIds(modelId: Id64String): Observable<Id64Array> {\n    return this.getElementModelsCategories().pipe(\n      map((elementModelsCategories) => {\n        return [...(elementModelsCategories.get(modelId)?.categoryIds ?? [])];\n      }),\n    );\n  }\n\n  public getAllCategories(): Observable<Id64Set> {\n    return this.getModelsCategoriesInfo().pipe(\n      mergeMap((modelsCategoriesInfo) => modelsCategoriesInfo.values()),\n      reduce((acc, { childCategories }) => {\n        childCategories.forEach(({ id }) => acc.add(id));\n        return acc;\n      }, new Set<Id64String>()),\n    );\n  }\n\n  public hasSubModel(elementId: Id64String): Observable<boolean> {\n    return this.getElementModelsCategories().pipe(map((elementModelsCategories) => elementModelsCategories.has(elementId)));\n  }\n\n  public getAllContainedCategories({\n    definitionContainerIds,\n    includeEmptyCategories,\n  }: {\n    definitionContainerIds: Id64Arg;\n    includeEmptyCategories?: boolean;\n  }): Observable<Id64Set> {\n    return this.getDefinitionContainersInfo().pipe(\n      mergeMap((definitionContainersInfo) =>\n        from(Id64.iterable(definitionContainerIds)).pipe(\n          mergeMap((definitionContainerId) => {\n            const definitionContainerInfo = definitionContainersInfo.get(definitionContainerId);\n            if (definitionContainerInfo === undefined) {\n              return of({ directCategories: undefined, indirectCategories: undefined });\n            }\n            const childDefinitionContainerIds = definitionContainerInfo.childDefinitionContainers.map(({ id }) => id);\n            return (\n              childDefinitionContainerIds.length > 0\n                ? this.getAllContainedCategories({\n                    definitionContainerIds: childDefinitionContainerIds,\n                    includeEmptyCategories,\n                  })\n                : of(new Set<CategoryId>())\n            ).pipe(\n              map((indirectCategories) => {\n                return {\n                  directCategories: applyElementsFilter(definitionContainerInfo.childCategories, includeEmptyCategories).map((category) => category.id),\n                  indirectCategories,\n                };\n              }),\n            );\n          }),\n          reduce((acc, { directCategories, indirectCategories }) => {\n            directCategories?.forEach((categoryId) => acc.add(categoryId));\n            indirectCategories?.forEach((categoryId) => acc.add(categoryId));\n            return acc;\n          }, new Set<CategoryId>()),\n        ),\n      ),\n    );\n  }\n\n  public getInstanceKeyPaths(\n    props: { categoryId: Id64String } | { definitionContainerId: Id64String } | { subCategoryId: Id64String },\n  ): Observable<InstanceKey[]> {\n    if (\"subCategoryId\" in props) {\n      return this.getSubCategoriesInfo().pipe(\n        mergeMap(({ subCategoryCategories, categorySubCategories }) => {\n          const categoryOfSubCategory = subCategoryCategories.get(props.subCategoryId);\n          if (categoryOfSubCategory === undefined) {\n            return of([]);\n          }\n          const subCategories = categorySubCategories.get(categoryOfSubCategory);\n          if (!subCategories || subCategories.length <= 1) {\n            return of([]);\n          }\n          return this.getInstanceKeyPaths({ categoryId: categoryOfSubCategory }).pipe(\n            map((pathToCategory) => [...pathToCategory, { id: props.subCategoryId, className: CLASS_NAME_SubCategory }]),\n          );\n        }),\n      );\n    }\n\n    if (\"categoryId\" in props) {\n      return this.getModelsCategoriesInfo().pipe(\n        mergeMap((modelsCategoriesInfo) => {\n          for (const [modelId, modelCategoriesInfo] of modelsCategoriesInfo) {\n            if (modelCategoriesInfo.childCategories.find((childCategory) => childCategory.id === props.categoryId)) {\n              const instanceKey = { id: props.categoryId, className: this.#categoryClass };\n              if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n                return of([instanceKey]);\n              }\n\n              return this.getInstanceKeyPaths({ definitionContainerId: modelId }).pipe(\n                map((pathToDefinitionContainer) => [...pathToDefinitionContainer, instanceKey]),\n              );\n            }\n          }\n          return of([]);\n        }),\n      );\n    }\n    return this.getDefinitionContainersInfo().pipe(\n      mergeMap((definitionContainersInfo) => {\n        const definitionContainerInfo = definitionContainersInfo.get(props.definitionContainerId);\n        if (definitionContainerInfo === undefined) {\n          return of([]);\n        }\n        const instanceKey = { id: props.definitionContainerId, className: CLASS_NAME_DefinitionContainer };\n        if (!definitionContainerInfo.parentDefinitionContainerExists) {\n          return of([instanceKey]);\n        }\n        return this.getInstanceKeyPaths({ definitionContainerId: definitionContainerInfo.modelId }).pipe(\n          map((pathToParentDefinitionContainer) => [...pathToParentDefinitionContainer, instanceKey]),\n        );\n      }),\n    );\n  }\n\n  public getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Observable<number> {\n    return this.#categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  public getAllDefinitionContainersAndCategories(props?: {\n    includeEmpty?: boolean;\n  }): Observable<{ categories: Array<CategoryId>; definitionContainers: Array<DefinitionContainerId> }> {\n    return forkJoin({\n      categories: this.getModelsCategoriesInfo().pipe(\n        mergeMap((modelsCategoriesInfo) => modelsCategoriesInfo.values()),\n        reduce((acc, modelCategoriesInfo) => {\n          applyElementsFilter(modelCategoriesInfo.childCategories, props?.includeEmpty).forEach((categoryInfo) => acc.push(categoryInfo.id));\n          return acc;\n        }, new Array<Id64String>()),\n      ),\n      definitionContainers: this.getDefinitionContainersInfo().pipe(\n        mergeMap((definitionContainersInfo) => definitionContainersInfo.entries()),\n        reduce((acc, [definitionContainerId, definitionContainerInfo]) => {\n          if (definitionContainerInfo.hasElements || !!props?.includeEmpty) {\n            acc.push(definitionContainerId);\n          }\n          return acc;\n        }, new Array<Id64String>()),\n      ),\n    });\n  }\n\n  public getRootDefinitionContainersAndCategories(props?: {\n    includeEmpty?: boolean;\n  }): Observable<{ categories: CategoryInfo[]; definitionContainers: Array<DefinitionContainerId> }> {\n    return forkJoin({\n      categories: this.getModelsCategoriesInfo().pipe(\n        mergeMap((modelsCategoriesInfo) => modelsCategoriesInfo.values()),\n        reduce((acc, modelCategoriesInfo) => {\n          if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n            applyElementsFilter(modelCategoriesInfo.childCategories, props?.includeEmpty).forEach((categoryInfo) => acc.push(categoryInfo));\n          }\n          return acc;\n        }, new Array<CategoryInfo>()),\n      ),\n      definitionContainers: this.getDefinitionContainersInfo().pipe(\n        mergeMap((definitionContainersInfo) => definitionContainersInfo.entries()),\n        reduce((acc, [definitionContainerId, definitionContainerInfo]) => {\n          if (!definitionContainerInfo.parentDefinitionContainerExists && (definitionContainerInfo.hasElements || !!props?.includeEmpty)) {\n            acc.push(definitionContainerId);\n          }\n          return acc;\n        }, new Array<Id64String>()),\n      ),\n    });\n  }\n\n  public getSubCategories(categoryId: Id64String): Observable<Array<SubCategoryId>> {\n    return this.getSubCategoriesInfo().pipe(map(({ categorySubCategories }) => categorySubCategories.get(categoryId) ?? []));\n  }\n\n  public getIsDefinitionContainerSupported(): Observable<boolean> {\n    this.#isDefinitionContainerSupported ??= this.queryIsDefinitionContainersSupported().pipe(shareReplay());\n    return this.#isDefinitionContainerSupported;\n  }\n}\n\nfunction applyElementsFilter<T extends { hasElements?: boolean }>(list: T[], includeEmpty: boolean | undefined): T[] {\n  return includeEmpty ? list : list.filter(({ hasElements }) => !!hasElements);\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/{UseFilteredPaths.d.ts → UseSearchPaths.d.ts}

@@ -5,23 +5,23 @@ import type { ModelId } from "../../common/internal/Types.js";
 import type { CategoriesTreeHierarchyConfiguration } from "../CategoriesTreeDefinition.js";
 import type { CategoriesTreeIdsCache } from "./CategoriesTreeIdsCache.js";
 /** @internal */
-export type CategoriesTreeFilteringError = "tooManyFilterMatches" | "unknownFilterError";
-type HierarchyFilteringPaths = Awaited<ReturnType<Required<VisibilityTreeProps>["getFilteredPaths"]>>;
+export type CategoriesTreeSearchError = "tooManySearchMatches" | "unknownSearchError";
+type HierarchySearchPaths = Awaited<ReturnType<Required<VisibilityTreeProps>["getSearchPaths"]>>;
 /** @internal */
-export declare function useFilteredPaths({ filter, viewType, hierarchyConfiguration, getCategoriesTreeIdsCache, onCategoriesFiltered, onFilteredPathsChanged, componentId, }: {
+export declare function useSearchPaths({ searchText, viewType, hierarchyConfiguration, getCategoriesTreeIdsCache, onCategoriesFiltered, onSearchPathsChanged, componentId, }: {
     viewType: "2d" | "3d";
-    filter?: string;
+    searchText?: string;
     hierarchyConfiguration: CategoriesTreeHierarchyConfiguration;
     getCategoriesTreeIdsCache: () => CategoriesTreeIdsCache;
     onCategoriesFiltered?: (categories: {
         categories: CategoryInfo[] | undefined;
         models?: Array<ModelId>;
     }) => void;
-    onFilteredPathsChanged: (paths: HierarchyFilteringPaths | undefined) => void;
+    onSearchPathsChanged: (paths: HierarchySearchPaths | undefined) => void;
     componentId: GuidString;
 }): {
-    getPaths: VisibilityTreeProps["getFilteredPaths"] | undefined;
-    filteringError: CategoriesTreeFilteringError | undefined;
+    getPaths: VisibilityTreeProps["getSearchPaths"] | undefined;
+    searchError: CategoriesTreeSearchError | undefined;
 };
 export {};
-//# sourceMappingURL=UseFilteredPaths.d.ts.map
+//# sourceMappingURL=UseSearchPaths.d.ts.map