@itwin/tree-widget-react 1.0.0-dev.3 → 1.0.0-dev.5

package/lib/cjs/components/trees/models-tree/ModelsTreeComponent.js.map

@@ -1 +1 @@
-{"version":3,"file":"ModelsTreeComponent.js","sourceRoot":"","sources":["../../../../../src/components/trees/models-tree/ModelsTreeComponent.tsx"],"names":[],"mappings":";;;;AAAA;;;+FAG+F;AAE/F,sCAAoC;AACpC,iCAA4E;AAC5E,oDAAkF;AAClF,oEAAqG;AACrG,wDAA0D;AAC1D,oDAAiD;AACjD,6DAA0D;AAC1D,iEAAiE;AACjE,qDAAkD;AAClD,6CAA0C;AAC1C,uEAA6H;AAqD7H;;;;GAIG;AACI,MAAM,mBAAmB,GAAG,CAAC,KAA8B,EAAE,EAAE;IACpE,MAAM,MAAM,GAAG,IAAA,uCAAyB,GAAE,CAAC;IAC3C,MAAM,QAAQ,GAAG,IAAA,+BAAiB,GAAE,CAAC;IAErC,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ;QACtB,OAAO,IAAI,CAAC;IAEd,OAAO,CACL,uBAAC,uBAAuB,OAAK,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,GAAI,CAC3E,CAAC;AACJ,CAAC,CAAC;AAVW,QAAA,mBAAmB,uBAU9B;AAEF;;;GAGG;AACH,2BAAmB,CAAC,aAAa,GAAG,aAAa,CAAC;AAElD;;;GAGG;AACH,2BAAmB,CAAC,aAAa,GAAG,aAAa,CAAC;AAElD;;;GAGG;AACH,2BAAmB,CAAC,YAAY,GAAG,YAAY,CAAC;AAEhD;;;GAGG;AACH,2BAAmB,CAAC,YAAY,GAAG,YAAY,CAAC;AAEhD;;;GAGG;AACH,2BAAmB,CAAC,YAAY,GAAG,YAAY,CAAC;AAEhD;;;GAGG;AACH,2BAAmB,CAAC,EAAE,GAAG,aAAa,CAAC;AAEvC;;;GAGG;AACH,2BAAmB,CAAC,QAAQ,GAAG,GAAG,EAAE,CAAC,uBAAU,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;AAEpE,SAAS,uBAAuB,CAAC,KAAuF;IACtH,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,IAAA,gBAAQ,EAAc,EAAE,CAAC,CAAC;IAExE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IAEnC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,eAAe,EAAE,GAAG,IAAA,0CAAqB,GAAE,CAAC;IAEjF,MAAM,WAAW,GAAG,IAAA,mBAAW,EAAC,KAAK,IAA0B,EAAE;QAC/D,MAAM,WAAW,GAAqB;YACpC,IAAI,EAAE,0BAA0B;YAChC,WAAW,EAAE,KAAK;SACnB,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC/D,OAAO,UAAU;aACd,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,gBAAgB,EAAyB,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;aACpF,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAgB,CAAC;IAC3C,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAEb,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,WAAW,EAAE;aACV,IAAI,CAAC,CAAC,UAAuB,EAAE,EAAE;YAChC,kBAAkB,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE;YACZ,kBAAkB,CAAC,EAAE,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;IACP,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;IAElB,OAAO,CACL,iCAAK,SAAS,EAAC,8BAA8B,aAC3C,uBAAC,uBAAU,IACT,aAAa,EAAE,aAAa,CAAC,cAAc,EAC3C,aAAa,EAAE,aAAa,CAAC,aAAa,EAC1C,iBAAiB,EAAE,aAAa,CAAC,uBAAuB,EACxD,WAAW,EAAE,aAAa,CAAC,kBAAkB,EAC7C,aAAa,EAAE,aAAa,CAAC,gBAAgB,YAE5C,KAAK,CAAC,aAAa;oBAClB,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CACvB,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CACb,uBAAC,gBAAQ,cACN,GAAG,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,IAD9B,KAAK,CAET,CACd;oBACD,CAAC,CAAC;wBACA,uBAAC,aAAa,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,IAAM,cAAc,CAAG;wBACjF,uBAAC,aAAa,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,IAAM,cAAc,CAAG;wBACjF,uBAAC,YAAY,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,IAAM,gBAAgB,CAAG;wBAClF,uBAAC,YAAY,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,IAAM,aAAa,CAAG;wBAC/E,uBAAC,YAAY,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,IAAM,aAAa,CAAG;qBAChF,GAEQ,EACb,gCAAK,SAAS,EAAC,0BAA0B,YACvC,uBAAC,qBAAS,cACP,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CACtB,uBAAC,uBAAU,OACL,KAAK,EACT,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,QAAQ,EACpB,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,aAAa,CAAC,gBAAgB,EAAE,EACtF,eAAe,EAAE,eAAe,GAChC,CACH,GACS,GACR,IACF,CACP,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAAkC;IACvD,OAAO,CACL,uBAAC,0BAAU,IACT,IAAI,EAAC,OAAO,EACZ,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,SAAS,CAAC,EACtC,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,IAAA,uCAAa,EAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,YAExF,uBAAC,uCAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAAkC;IACvD,OAAO,CACL,uBAAC,0BAAU,IACT,IAAI,EAAC,OAAO,EACZ,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,SAAS,CAAC,EACtC,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,IAAA,uCAAa,EAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,YAExF,uBAAC,uCAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAkC;IACtD,OAAO,CACL,uBAAC,0BAAU,IACT,IAAI,EAAC,OAAO,EACZ,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,QAAQ,CAAC,EACrC,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,IAAA,yCAAe,EAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,YAE1F,uBAAC,uCAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAkC;IACtD,MAAM,QAAQ,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QAC5B,OAAO,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACzF,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnB,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,IAAA,gBAAQ,EAAC,KAAK,CAAC,CAAC;IAEhE,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,IAAA,6CAAmB,EAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QACnE,OAAO,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,EAAY,EAAE,EAAE,CAAC,mBAAmB,CAAC,IAAA,6CAAmB,EAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACpI,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAE/B,OAAO,CACL,uBAAC,sBAAM,IACL,IAAI,EAAC,OAAO,EACZ,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,eAAe,CAAC,EAC5C,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,IAAA,sCAAY,EAAC,QAAQ,EAAE,gBAAgB,EAAE,KAAK,CAAC,QAAQ,CAAC,EAC5E,QAAQ,EAAE,QAAQ,CAAC,MAAM,KAAK,CAAC,EAC/B,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,uBAAC,uCAAiB,KAAG,CAAC,CAAC,CAAC,uBAAC,uCAAiB,KAAG,YAExE,uBAAU,CAAC,SAAS,CAAC,SAAS,CAAC,GACzB,CACV,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAkC;IACtD,MAAM,QAAQ,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QAC5B,OAAO,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC1F,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnB,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,IAAA,gBAAQ,EAAC,KAAK,CAAC,CAAC;IAEhE,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,IAAA,6CAAmB,EAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QACnE,OAAO,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,EAAY,EAAE,EAAE,CAAC,mBAAmB,CAAC,IAAA,6CAAmB,EAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACpI,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAE/B,OAAO,CACL,uBAAC,sBAAM,IACL,IAAI,EAAC,OAAO,EACZ,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,eAAe,CAAC,EAC5C,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,IAAA,sCAAY,EAAC,QAAQ,EAAE,gBAAgB,EAAE,KAAK,CAAC,QAAQ,CAAC,EAC5E,QAAQ,EAAE,QAAQ,CAAC,MAAM,KAAK,CAAC,EAC/B,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,uBAAC,uCAAiB,KAAG,CAAC,CAAC,CAAC,uBAAC,uCAAiB,KAAG,YAExE,uBAAU,CAAC,SAAS,CAAC,SAAS,CAAC,GACzB,CACV,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n* See LICENSE.md in the project root for license terms and full copyright notice.\n*--------------------------------------------------------------------------------------------*/\n\nimport \"../VisibilityTreeBase.scss\";\nimport { Fragment, useCallback, useEffect, useMemo, useState } from \"react\";\nimport { useActiveIModelConnection, useActiveViewport } from \"@itwin/appui-react\";\nimport { SvgVisibilityHalf, SvgVisibilityHide, SvgVisibilityShow } from \"@itwin/itwinui-icons-react\";\nimport { Button, IconButton } from \"@itwin/itwinui-react\";\nimport { TreeWidget } from \"../../../TreeWidget\";\nimport { TreeHeader } from \"../../tree-header/TreeHeader\";\nimport { useTreeFilteringState } from \"../../TreeFilteringState\";\nimport { AutoSizer } from \"../../utils/AutoSizer\";\nimport { ModelsTree } from \"./ModelsTree\";\nimport { areAllModelsVisible, hideAllModels, invertAllModels, showAllModels, toggleModels } from \"./ModelsVisibilityHandler\";\n\nimport type { GeometricModel3dProps, ModelQueryParams } from \"@itwin/core-common\";\nimport type { IModelConnection, ScreenViewport, Viewport } from \"@itwin/core-frontend\";\nimport type { TreeHeaderButtonProps } from \"../../tree-header/TreeHeader\";\nimport type { ModelsTreeProps } from \"./ModelsTree\";\n\n/**\n * Information about a single Model.\n * @public\n */\nexport interface ModelInfo {\n  id: string;\n  isPlanProjection?: boolean;\n}\n\n/**\n * Props that get passed to [[ModelsTreeComponent]] header button renderer.\n * @see ModelTreeComponentProps.headerButtons\n * @public\n */\nexport interface ModelsTreeHeaderButtonProps extends TreeHeaderButtonProps {\n  /** A list of models available in the iModel. */\n  models: ModelInfo[];\n}\n\n/**\n * Props for [[ModelsTreeComponent]].\n * @public\n */\nexport interface ModelTreeComponentProps extends Omit<ModelsTreeProps,\n| \"iModel\"\n| \"activeView\"\n| \"width\"\n| \"height\"\n| \"filterInfo\"\n| \"onFilterApplied\"\n> {\n  /**\n   * Renderers of header buttons. Defaults to:\n   * ```ts\n   * [\n   *   ModelsTreeComponent.ShowAllButton,\n   *   ModelsTreeComponent.HideAllButton,\n   *   ModelsTreeComponent.InvertButton,\n   *   ModelsTreeComponent.View2DButton,\n   *   ModelsTreeComponent.View3DButton,\n   * ]\n   * ```\n   */\n  headerButtons?: Array<(props: ModelsTreeHeaderButtonProps) => React.ReactNode>;\n}\n\n/**\n * A component that renders [[ModelsTree]] and a header with filtering capabilities\n * and header buttons.\n * @public\n */\nexport const ModelsTreeComponent = (props: ModelTreeComponentProps) => {\n  const iModel = useActiveIModelConnection();\n  const viewport = useActiveViewport();\n\n  if (!iModel || !viewport)\n    return null;\n\n  return (\n    <ModelsTreeComponentImpl {...props} iModel={iModel} viewport={viewport} />\n  );\n};\n\n/**\n * Renders a \"Show all\" button that enables display of all models.\n * @public\n */\nModelsTreeComponent.ShowAllButton = ShowAllButton;\n\n/**\n * Renders a \"Hide all\" button that disables display of all models.\n * @public\n */\nModelsTreeComponent.HideAllButton = HideAllButton;\n\n/**\n * Renders an \"Invert all\" button that inverts display of all models.\n * @public\n */\nModelsTreeComponent.InvertButton = InvertButton;\n\n/**\n * Renders a \"View 2D\" button that enables display of all plan projection models and disables all others.\n * @public\n */\nModelsTreeComponent.View2DButton = View2DButton;\n\n/**\n * Renders a \"View 3D\" button that enables display of all non-plan projection models and disables all plan projection ones.\n * @public\n */\nModelsTreeComponent.View3DButton = View3DButton;\n\n/**\n * Id of the component. May be used when a creating a [[TreeDefinition]] for [[TreeWidgetComponent]].\n * @public\n */\nModelsTreeComponent.id = \"models-tree\";\n\n/**\n * Label of the component. May be used when a creating a [[TreeDefinition]] for [[TreeWidgetComponent]].\n * @public\n */\nModelsTreeComponent.getLabel = () => TreeWidget.translate(\"models\");\n\nfunction ModelsTreeComponentImpl(props: ModelTreeComponentProps & { iModel: IModelConnection, viewport: ScreenViewport }) {\n  const [availableModels, setAvailableModels] = useState<ModelInfo[]>([]);\n\n  const { viewport, iModel } = props;\n\n  const { searchOptions, filterString, onFilterApplied } = useTreeFilteringState();\n\n  const queryModels = useCallback(async (): Promise<ModelInfo[]> => {\n    const queryParams: ModelQueryParams = {\n      from: \"BisCore.GeometricModel3d\",\n      wantPrivate: false,\n    };\n    const modelProps = await iModel.models.queryProps(queryParams);\n    return modelProps\n      .map(({ id, isPlanProjection }: GeometricModel3dProps) => ({ id, isPlanProjection }))\n      .filter(({ id }) => id) as ModelInfo[];\n  }, [iModel]);\n\n  useEffect(() => {\n    queryModels()\n      .then((modelInfos: ModelInfo[]) => {\n        setAvailableModels(modelInfos);\n      })\n      .catch((_e) => {\n        setAvailableModels([]);\n      });\n  }, [queryModels]);\n\n  return (\n    <div className=\"tree-widget-tree-with-header\">\n      <TreeHeader\n        onFilterClear={searchOptions.onFilterCancel}\n        onFilterStart={searchOptions.onFilterStart}\n        onSelectedChanged={searchOptions.onResultSelectedChanged}\n        resultCount={searchOptions.matchedResultCount}\n        selectedIndex={searchOptions.activeMatchIndex}\n      >\n        {props.headerButtons\n          ? props.headerButtons.map(\n            (btn, index) =>\n              <Fragment key={index}>\n                {btn({ viewport, models: availableModels })}\n              </Fragment>\n          )\n          : [\n            <ShowAllButton viewport={viewport} models={availableModels} key=\"show-all-btn\" />,\n            <HideAllButton viewport={viewport} models={availableModels} key=\"hide-all-btn\" />,\n            <InvertButton viewport={viewport} models={availableModels} key=\"invert-all-btn\" />,\n            <View2DButton viewport={viewport} models={availableModels} key=\"view-2d-btn\" />,\n            <View3DButton viewport={viewport} models={availableModels} key=\"view-3d-btn\" />,\n          ]\n        }\n      </TreeHeader>\n      <div className=\"tree-widget-tree-content\">\n        <AutoSizer>\n          {({ width, height }) => (\n            <ModelsTree\n              {...props}\n              iModel={iModel}\n              activeView={viewport}\n              width={width}\n              height={height}\n              filterInfo={{ filter: filterString, activeMatchIndex: searchOptions.activeMatchIndex }}\n              onFilterApplied={onFilterApplied}\n            />\n          )}\n        </AutoSizer>\n      </div>\n    </div>\n  );\n}\n\nfunction ShowAllButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size=\"small\"\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"showAll\")}\n      onClick={() => void showAllModels(props.models.map((model) => model.id), props.viewport)}\n    >\n      <SvgVisibilityShow />\n    </IconButton>\n  );\n}\n\nfunction HideAllButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size=\"small\"\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"hideAll\")}\n      onClick={() => void hideAllModels(props.models.map((model) => model.id), props.viewport)}\n    >\n      <SvgVisibilityHide />\n    </IconButton>\n  );\n}\n\nfunction InvertButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size=\"small\"\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"invert\")}\n      onClick={() => void invertAllModels(props.models.map((model) => model.id), props.viewport)}\n    >\n      <SvgVisibilityHalf />\n    </IconButton>\n  );\n}\n\nfunction View2DButton(props: ModelsTreeHeaderButtonProps) {\n  const models2d = useMemo(() => {\n    return props.models.filter((model) => model.isPlanProjection).map((model) => model.id);\n  }, [props.models]);\n\n  const [is2dToggleActive, setIs2dToggleActive] = useState(false);\n\n  useEffect(() => {\n    setIs2dToggleActive(areAllModelsVisible(models2d, props.viewport));\n    return props.viewport.onViewedModelsChanged.addListener((vp: Viewport) => setIs2dToggleActive(areAllModelsVisible(models2d, vp)));\n  }, [models2d, props.viewport]);\n\n  return (\n    <Button\n      size=\"small\"\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"toggle2DViews\")}\n      onClick={() => void toggleModels(models2d, is2dToggleActive, props.viewport)}\n      disabled={models2d.length === 0}\n      endIcon={is2dToggleActive ? <SvgVisibilityShow /> : <SvgVisibilityHide />}\n    >\n      {TreeWidget.translate(\"label2D\")}\n    </Button>\n  );\n}\n\nfunction View3DButton(props: ModelsTreeHeaderButtonProps) {\n  const models3d = useMemo(() => {\n    return props.models.filter((model) => !model.isPlanProjection).map((model) => model.id);\n  }, [props.models]);\n\n  const [is3dToggleActive, setIs3dToggleActive] = useState(false);\n\n  useEffect(() => {\n    setIs3dToggleActive(areAllModelsVisible(models3d, props.viewport));\n    return props.viewport.onViewedModelsChanged.addListener((vp: Viewport) => setIs3dToggleActive(areAllModelsVisible(models3d, vp)));\n  }, [models3d, props.viewport]);\n\n  return (\n    <Button\n      size=\"small\"\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"toggle3DViews\")}\n      onClick={() => void toggleModels(models3d, is3dToggleActive, props.viewport)}\n      disabled={models3d.length === 0}\n      endIcon={is3dToggleActive ? <SvgVisibilityShow /> : <SvgVisibilityHide />}\n    >\n      {TreeWidget.translate(\"label3D\")}\n    </Button>\n  );\n}\n"]}
+{"version":3,"file":"ModelsTreeComponent.js","sourceRoot":"","sources":["../../../../../src/components/trees/models-tree/ModelsTreeComponent.tsx"],"names":[],"mappings":";;;;AAAA;;;+FAG+F;AAE/F,sCAAoC;AACpC,iCAA4E;AAC5E,oDAAkF;AAClF,oEAAqG;AACrG,wDAA0D;AAC1D,oDAAiD;AACjD,6DAA0D;AAC1D,iEAAiE;AACjE,qDAAkD;AAClD,6CAA0C;AAC1C,uEAA6H;AAqD7H;;;;GAIG;AACI,MAAM,mBAAmB,GAAG,CAAC,KAA8B,EAAE,EAAE;IACpE,MAAM,MAAM,GAAG,IAAA,uCAAyB,GAAE,CAAC;IAC3C,MAAM,QAAQ,GAAG,IAAA,+BAAiB,GAAE,CAAC;IAErC,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ;QACtB,OAAO,IAAI,CAAC;IAEd,OAAO,CACL,uBAAC,uBAAuB,OAAK,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,GAAI,CAC3E,CAAC;AACJ,CAAC,CAAC;AAVW,QAAA,mBAAmB,uBAU9B;AAEF;;;GAGG;AACH,2BAAmB,CAAC,aAAa,GAAG,aAAa,CAAC;AAElD;;;GAGG;AACH,2BAAmB,CAAC,aAAa,GAAG,aAAa,CAAC;AAElD;;;GAGG;AACH,2BAAmB,CAAC,YAAY,GAAG,YAAY,CAAC;AAEhD;;;GAGG;AACH,2BAAmB,CAAC,YAAY,GAAG,YAAY,CAAC;AAEhD;;;GAGG;AACH,2BAAmB,CAAC,YAAY,GAAG,YAAY,CAAC;AAEhD;;;GAGG;AACH,2BAAmB,CAAC,EAAE,GAAG,aAAa,CAAC;AAEvC;;;GAGG;AACH,2BAAmB,CAAC,QAAQ,GAAG,GAAG,EAAE,CAAC,uBAAU,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;AAEpE,SAAS,uBAAuB,CAAC,KAAuF;IACtH,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,IAAA,gBAAQ,EAAc,EAAE,CAAC,CAAC;IAExE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IAEnC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,eAAe,EAAE,GAAG,IAAA,0CAAqB,GAAE,CAAC;IAEjF,MAAM,WAAW,GAAG,IAAA,mBAAW,EAAC,KAAK,IAA0B,EAAE;QAC/D,MAAM,WAAW,GAAqB;YACpC,IAAI,EAAE,0BAA0B;YAChC,WAAW,EAAE,KAAK;SACnB,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC/D,OAAO,UAAU;aACd,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,gBAAgB,EAAyB,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;aACpF,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAgB,CAAC;IAC3C,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAEb,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,WAAW,EAAE;aACV,IAAI,CAAC,CAAC,UAAuB,EAAE,EAAE;YAChC,kBAAkB,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE;YACZ,kBAAkB,CAAC,EAAE,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;IACP,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;IAElB,OAAO,CACL,iCAAK,SAAS,EAAC,8BAA8B,aAC3C,uBAAC,uBAAU,IACT,aAAa,EAAE,aAAa,CAAC,cAAc,EAC3C,aAAa,EAAE,aAAa,CAAC,aAAa,EAC1C,iBAAiB,EAAE,aAAa,CAAC,uBAAuB,EACxD,WAAW,EAAE,aAAa,CAAC,kBAAkB,EAC7C,aAAa,EAAE,aAAa,CAAC,gBAAgB,YAE5C,KAAK,CAAC,aAAa;oBAClB,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CACvB,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CACb,uBAAC,gBAAQ,cACN,GAAG,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,IAD9B,KAAK,CAET,CACd;oBACD,CAAC,CAAC;wBACA,uBAAC,aAAa,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,IAAM,cAAc,CAAG;wBACjF,uBAAC,aAAa,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,IAAM,cAAc,CAAG;wBACjF,uBAAC,YAAY,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,IAAM,gBAAgB,CAAG;wBAClF,uBAAC,YAAY,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,IAAM,aAAa,CAAG;wBAC/E,uBAAC,YAAY,IAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,IAAM,aAAa,CAAG;qBAChF,GAEQ,EACb,gCAAK,SAAS,EAAC,0BAA0B,YACvC,uBAAC,qBAAS,cACP,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CACtB,uBAAC,uBAAU,OACL,KAAK,EACT,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,QAAQ,EACpB,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,aAAa,CAAC,gBAAgB,EAAE,EACtF,eAAe,EAAE,eAAe,GAChC,CACH,GACS,GACR,IACF,CACP,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAAkC;IACvD,OAAO,CACL,uBAAC,0BAAU,IACT,IAAI,EAAC,OAAO,EACZ,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,SAAS,CAAC,EACtC,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,IAAA,uCAAa,EAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,YAExF,uBAAC,uCAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAAkC;IACvD,OAAO,CACL,uBAAC,0BAAU,IACT,IAAI,EAAC,OAAO,EACZ,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,SAAS,CAAC,EACtC,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,IAAA,uCAAa,EAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,YAExF,uBAAC,uCAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAkC;IACtD,OAAO,CACL,uBAAC,0BAAU,IACT,IAAI,EAAC,OAAO,EACZ,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,QAAQ,CAAC,EACrC,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,IAAA,yCAAe,EAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,YAE1F,uBAAC,uCAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAkC;IACtD,MAAM,QAAQ,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QAC5B,OAAO,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACzF,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnB,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,IAAA,gBAAQ,EAAC,KAAK,CAAC,CAAC;IAEhE,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,IAAA,6CAAmB,EAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QACnE,OAAO,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,EAAY,EAAE,EAAE,CAAC,mBAAmB,CAAC,IAAA,6CAAmB,EAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACpI,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAE/B,OAAO,CACL,uBAAC,sBAAM,IACL,IAAI,EAAC,OAAO,EACZ,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,eAAe,CAAC,EAC5C,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,IAAA,sCAAY,EAAC,QAAQ,EAAE,gBAAgB,EAAE,KAAK,CAAC,QAAQ,CAAC,EAC5E,QAAQ,EAAE,QAAQ,CAAC,MAAM,KAAK,CAAC,EAC/B,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,uBAAC,uCAAiB,KAAG,CAAC,CAAC,CAAC,uBAAC,uCAAiB,KAAG,YAExE,uBAAU,CAAC,SAAS,CAAC,SAAS,CAAC,GACzB,CACV,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAkC;IACtD,MAAM,QAAQ,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QAC5B,OAAO,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC1F,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnB,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,IAAA,gBAAQ,EAAC,KAAK,CAAC,CAAC;IAEhE,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,IAAA,6CAAmB,EAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QACnE,OAAO,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,EAAY,EAAE,EAAE,CAAC,mBAAmB,CAAC,IAAA,6CAAmB,EAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACpI,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAE/B,OAAO,CACL,uBAAC,sBAAM,IACL,IAAI,EAAC,OAAO,EACZ,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,eAAe,CAAC,EAC5C,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,IAAA,sCAAY,EAAC,QAAQ,EAAE,gBAAgB,EAAE,KAAK,CAAC,QAAQ,CAAC,EAC5E,QAAQ,EAAE,QAAQ,CAAC,MAAM,KAAK,CAAC,EAC/B,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,uBAAC,uCAAiB,KAAG,CAAC,CAAC,CAAC,uBAAC,uCAAiB,KAAG,YAExE,uBAAU,CAAC,SAAS,CAAC,SAAS,CAAC,GACzB,CACV,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n* See LICENSE.md in the project root for license terms and full copyright notice.\n*--------------------------------------------------------------------------------------------*/\n\nimport \"../VisibilityTreeBase.scss\";\nimport { Fragment, useCallback, useEffect, useMemo, useState } from \"react\";\nimport { useActiveIModelConnection, useActiveViewport } from \"@itwin/appui-react\";\nimport { SvgVisibilityHalf, SvgVisibilityHide, SvgVisibilityShow } from \"@itwin/itwinui-icons-react\";\nimport { Button, IconButton } from \"@itwin/itwinui-react\";\nimport { TreeWidget } from \"../../../TreeWidget\";\nimport { TreeHeader } from \"../../tree-header/TreeHeader\";\nimport { useTreeFilteringState } from \"../../TreeFilteringState\";\nimport { AutoSizer } from \"../../utils/AutoSizer\";\nimport { ModelsTree } from \"./ModelsTree\";\nimport { areAllModelsVisible, hideAllModels, invertAllModels, showAllModels, toggleModels } from \"./ModelsVisibilityHandler\";\n\nimport type { GeometricModel3dProps, ModelQueryParams } from \"@itwin/core-common\";\nimport type { IModelConnection, ScreenViewport, Viewport } from \"@itwin/core-frontend\";\nimport type { TreeHeaderButtonProps } from \"../../tree-header/TreeHeader\";\nimport type { ModelsTreeProps } from \"./ModelsTree\";\n\n/**\n * Information about a single Model.\n * @public\n */\nexport interface ModelInfo {\n  id: string;\n  isPlanProjection?: boolean;\n}\n\n/**\n * Props that get passed to [[ModelsTreeComponent]] header button renderer.\n * @see ModelTreeComponentProps.headerButtons\n * @public\n */\nexport interface ModelsTreeHeaderButtonProps extends TreeHeaderButtonProps {\n  /** A list of models available in the iModel. */\n  models: ModelInfo[];\n}\n\n/**\n * Props for [[ModelsTreeComponent]].\n * @public\n */\nexport interface ModelTreeComponentProps extends Omit<ModelsTreeProps,\n| \"iModel\"\n| \"activeView\"\n| \"width\"\n| \"height\"\n| \"filterInfo\"\n| \"onFilterApplied\"\n> {\n  /**\n   * Renderers of header buttons. Defaults to:\n   * ```ts\n   * [\n   *   ModelsTreeComponent.ShowAllButton,\n   *   ModelsTreeComponent.HideAllButton,\n   *   ModelsTreeComponent.InvertButton,\n   *   ModelsTreeComponent.View2DButton,\n   *   ModelsTreeComponent.View3DButton,\n   * ]\n   * ```\n   */\n  headerButtons?: Array<(props: ModelsTreeHeaderButtonProps) => React.ReactNode>;\n}\n\n/**\n * A component that renders [[ModelsTree]] and a header with filtering capabilities\n * and header buttons.\n * @public\n */\nexport const ModelsTreeComponent = (props: ModelTreeComponentProps) => {\n  const iModel = useActiveIModelConnection();\n  const viewport = useActiveViewport();\n\n  if (!iModel || !viewport)\n    return null;\n\n  return (\n    <ModelsTreeComponentImpl {...props} iModel={iModel} viewport={viewport} />\n  );\n};\n\n/**\n * Renders a \"Show all\" button that enables display of all models.\n * @public\n */\nModelsTreeComponent.ShowAllButton = ShowAllButton;\n\n/**\n * Renders a \"Hide all\" button that disables display of all models.\n * @public\n */\nModelsTreeComponent.HideAllButton = HideAllButton;\n\n/**\n * Renders an \"Invert all\" button that inverts display of all models.\n * @public\n */\nModelsTreeComponent.InvertButton = InvertButton;\n\n/**\n * Renders a \"View 2D\" button that enables display of all plan projection models and disables all others.\n * @public\n */\nModelsTreeComponent.View2DButton = View2DButton;\n\n/**\n * Renders a \"View 3D\" button that enables display of all non-plan projection models and disables all plan projection ones.\n * @public\n */\nModelsTreeComponent.View3DButton = View3DButton;\n\n/**\n * Id of the component. May be used when a creating a [[TreeDefinition]] for [[SelectableTree]].\n * @public\n */\nModelsTreeComponent.id = \"models-tree\";\n\n/**\n * Label of the component. May be used when a creating a [[TreeDefinition]] for [[SelectableTree]].\n * @public\n */\nModelsTreeComponent.getLabel = () => TreeWidget.translate(\"models\");\n\nfunction ModelsTreeComponentImpl(props: ModelTreeComponentProps & { iModel: IModelConnection, viewport: ScreenViewport }) {\n  const [availableModels, setAvailableModels] = useState<ModelInfo[]>([]);\n\n  const { viewport, iModel } = props;\n\n  const { searchOptions, filterString, onFilterApplied } = useTreeFilteringState();\n\n  const queryModels = useCallback(async (): Promise<ModelInfo[]> => {\n    const queryParams: ModelQueryParams = {\n      from: \"BisCore.GeometricModel3d\",\n      wantPrivate: false,\n    };\n    const modelProps = await iModel.models.queryProps(queryParams);\n    return modelProps\n      .map(({ id, isPlanProjection }: GeometricModel3dProps) => ({ id, isPlanProjection }))\n      .filter(({ id }) => id) as ModelInfo[];\n  }, [iModel]);\n\n  useEffect(() => {\n    queryModels()\n      .then((modelInfos: ModelInfo[]) => {\n        setAvailableModels(modelInfos);\n      })\n      .catch((_e) => {\n        setAvailableModels([]);\n      });\n  }, [queryModels]);\n\n  return (\n    <div className=\"tree-widget-tree-with-header\">\n      <TreeHeader\n        onFilterClear={searchOptions.onFilterCancel}\n        onFilterStart={searchOptions.onFilterStart}\n        onSelectedChanged={searchOptions.onResultSelectedChanged}\n        resultCount={searchOptions.matchedResultCount}\n        selectedIndex={searchOptions.activeMatchIndex}\n      >\n        {props.headerButtons\n          ? props.headerButtons.map(\n            (btn, index) =>\n              <Fragment key={index}>\n                {btn({ viewport, models: availableModels })}\n              </Fragment>\n          )\n          : [\n            <ShowAllButton viewport={viewport} models={availableModels} key=\"show-all-btn\" />,\n            <HideAllButton viewport={viewport} models={availableModels} key=\"hide-all-btn\" />,\n            <InvertButton viewport={viewport} models={availableModels} key=\"invert-all-btn\" />,\n            <View2DButton viewport={viewport} models={availableModels} key=\"view-2d-btn\" />,\n            <View3DButton viewport={viewport} models={availableModels} key=\"view-3d-btn\" />,\n          ]\n        }\n      </TreeHeader>\n      <div className=\"tree-widget-tree-content\">\n        <AutoSizer>\n          {({ width, height }) => (\n            <ModelsTree\n              {...props}\n              iModel={iModel}\n              activeView={viewport}\n              width={width}\n              height={height}\n              filterInfo={{ filter: filterString, activeMatchIndex: searchOptions.activeMatchIndex }}\n              onFilterApplied={onFilterApplied}\n            />\n          )}\n        </AutoSizer>\n      </div>\n    </div>\n  );\n}\n\nfunction ShowAllButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size=\"small\"\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"showAll\")}\n      onClick={() => void showAllModels(props.models.map((model) => model.id), props.viewport)}\n    >\n      <SvgVisibilityShow />\n    </IconButton>\n  );\n}\n\nfunction HideAllButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size=\"small\"\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"hideAll\")}\n      onClick={() => void hideAllModels(props.models.map((model) => model.id), props.viewport)}\n    >\n      <SvgVisibilityHide />\n    </IconButton>\n  );\n}\n\nfunction InvertButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size=\"small\"\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"invert\")}\n      onClick={() => void invertAllModels(props.models.map((model) => model.id), props.viewport)}\n    >\n      <SvgVisibilityHalf />\n    </IconButton>\n  );\n}\n\nfunction View2DButton(props: ModelsTreeHeaderButtonProps) {\n  const models2d = useMemo(() => {\n    return props.models.filter((model) => model.isPlanProjection).map((model) => model.id);\n  }, [props.models]);\n\n  const [is2dToggleActive, setIs2dToggleActive] = useState(false);\n\n  useEffect(() => {\n    setIs2dToggleActive(areAllModelsVisible(models2d, props.viewport));\n    return props.viewport.onViewedModelsChanged.addListener((vp: Viewport) => setIs2dToggleActive(areAllModelsVisible(models2d, vp)));\n  }, [models2d, props.viewport]);\n\n  return (\n    <Button\n      size=\"small\"\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"toggle2DViews\")}\n      onClick={() => void toggleModels(models2d, is2dToggleActive, props.viewport)}\n      disabled={models2d.length === 0}\n      endIcon={is2dToggleActive ? <SvgVisibilityShow /> : <SvgVisibilityHide />}\n    >\n      {TreeWidget.translate(\"label2D\")}\n    </Button>\n  );\n}\n\nfunction View3DButton(props: ModelsTreeHeaderButtonProps) {\n  const models3d = useMemo(() => {\n    return props.models.filter((model) => !model.isPlanProjection).map((model) => model.id);\n  }, [props.models]);\n\n  const [is3dToggleActive, setIs3dToggleActive] = useState(false);\n\n  useEffect(() => {\n    setIs3dToggleActive(areAllModelsVisible(models3d, props.viewport));\n    return props.viewport.onViewedModelsChanged.addListener((vp: Viewport) => setIs3dToggleActive(areAllModelsVisible(models3d, vp)));\n  }, [models3d, props.viewport]);\n\n  return (\n    <Button\n      size=\"small\"\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"toggle3DViews\")}\n      onClick={() => void toggleModels(models3d, is3dToggleActive, props.viewport)}\n      disabled={models3d.length === 0}\n      endIcon={is3dToggleActive ? <SvgVisibilityShow /> : <SvgVisibilityHide />}\n    >\n      {TreeWidget.translate(\"label3D\")}\n    </Button>\n  );\n}\n"]}

package/lib/cjs/components/trees/models-tree/Utils.d.ts

@@ -1,5 +1,6 @@
 import type { Id64String } from "@itwin/core-bentley";
-import type { Ruleset } from "@itwin/presentation-common";
+import type { DelayLoadedTreeNodeItem } from "@itwin/components-react";
+import type { Node, Ruleset } from "@itwin/presentation-common";
 import type { ModelsTreeHierarchyConfiguration } from "./ModelsTree";
 /** @internal */
 export declare class CachingElementIdsContainer {
@@ -19,4 +20,6 @@ export declare function createRuleset(props: CreateRulesetProps): Ruleset;
 export type CreateSearchRulesetProps = Omit<ModelsTreeHierarchyConfiguration, "enableElementsClassGrouping">;
 /** @internal */
 export declare function createSearchRuleset(props: CreateSearchRulesetProps): Ruleset;
+/** @internal */
+export declare function customizeModelsTreeNodeItem(item: Partial<DelayLoadedTreeNodeItem>, node: Partial<Node>): void;
 //# sourceMappingURL=Utils.d.ts.map

package/lib/cjs/components/trees/models-tree/Utils.js

@@ -4,7 +4,8 @@
 * See LICENSE.md in the project root for license terms and full copyright notice.
 *--------------------------------------------------------------------------------------------*/
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createSearchRuleset = exports.createRuleset = exports.CachingElementIdsContainer = void 0;
+exports.customizeModelsTreeNodeItem = exports.createSearchRuleset = exports.createRuleset = exports.CachingElementIdsContainer = void 0;
+const presentation_common_1 = require("@itwin/presentation-common");
 /** @internal */
 class CachingElementIdsContainer {
     constructor(_generator) {
@@ -59,6 +60,7 @@ function createRuleset(props) {
                         ruleType: "ExtendedData",
                         items: {
                             isSubject: "true",
+                            icon: "\"icon-imodel-hollow-2\"",
                         },
                     },
                 ],
@@ -113,6 +115,7 @@ function createRuleset(props) {
                         ruleType: "ExtendedData",
                         items: {
                             isSubject: "true",
+                            icon: "\"icon-folder\"",
                         },
                     },
                 ],
@@ -191,6 +194,7 @@ function createRuleset(props) {
                         ruleType: "ExtendedData",
                         items: {
                             isModel: "true",
+                            icon: "\"icon-model\"",
                         },
                     },
                 ],
@@ -221,6 +225,7 @@ function createRuleset(props) {
                         ruleType: "ExtendedData",
                         items: {
                             isModel: "true",
+                            icon: "\"icon-model\"",
                         },
                     },
                 ],
@@ -263,6 +268,7 @@ function createRuleset(props) {
                         items: {
                             isCategory: "true",
                             modelId: "ParentNode.InstanceId",
+                            icon: "\"icon-layers\"",
                         },
                     },
                 ],
@@ -294,6 +300,8 @@ function createRuleset(props) {
                         items: {
                             modelId: "this.Model.Id",
                             categoryId: "this.Category.Id",
+                            icon: "\"icon-item\"",
+                            groupIcon: "\"icon-ec-class\"",
                         },
                     },
                 ],
@@ -324,6 +332,8 @@ function createRuleset(props) {
                         items: {
                             modelId: "this.Model.Id",
                             categoryId: "this.Category.Id",
+                            icon: "\"icon-item\"",
+                            groupIcon: "\"icon-ec-class\"",
                         },
                     },
                 ],
@@ -354,31 +364,6 @@ function createRuleset(props) {
                     },
                 ],
             },
-            {
-                ruleType: "ImageIdOverride",
-                condition: `ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass("Subject", "BisCore")`,
-                imageIdExpression: `IIF(this.Parent.Id = NULL, "icon-imodel-hollow-2", "icon-folder")`,
-            },
-            {
-                ruleType: "ImageIdOverride",
-                condition: `ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass("Model", "BisCore")`,
-                imageIdExpression: `"icon-model"`,
-            },
-            {
-                ruleType: "ImageIdOverride",
-                condition: `ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass("Category", "BisCore")`,
-                imageIdExpression: `"icon-layers"`,
-            },
-            {
-                ruleType: "ImageIdOverride",
-                condition: `ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass("Element", "BisCore")`,
-                imageIdExpression: `"icon-item"`,
-            },
-            {
-                ruleType: "ImageIdOverride",
-                condition: `ThisNode.IsClassGroupingNode`,
-                imageIdExpression: `"icon-ec-class"`,
-            },
             {
                 ruleType: "Content",
                 condition: `ContentDisplayType = "AssemblyElementsRequest"`,
@@ -435,6 +420,7 @@ function createSearchRuleset(props) {
                         ruleType: "ExtendedData",
                         items: {
                             isSubject: "true",
+                            icon: "\"icon-imodel-hollow-2\"",
                         },
                     },
                 ],
@@ -489,6 +475,7 @@ function createSearchRuleset(props) {
                         ruleType: "ExtendedData",
                         items: {
                             isSubject: "true",
+                            icon: "\"icon-folder\"",
                         },
                     },
                     {
@@ -546,6 +533,7 @@ function createSearchRuleset(props) {
                         ruleType: "ExtendedData",
                         items: {
                             isModel: "true",
+                            icon: "\"icon-model\"",
                         },
                     },
                 ],
@@ -597,6 +585,7 @@ function createSearchRuleset(props) {
                         ruleType: "ExtendedData",
                         items: {
                             isModel: "true",
+                            icon: "\"icon-model\"",
                         },
                     },
                 ],
@@ -630,22 +619,25 @@ function createSearchRuleset(props) {
                         ruleType: "ExtendedData",
                         items: {
                             isModel: "true",
+                            icon: "\"icon-model\"",
                         },
                     },
                 ],
             },
-            {
-                ruleType: "ImageIdOverride",
-                condition: `ThisNode.IsOfClass("Subject", "BisCore")`,
-                imageIdExpression: `IIF(this.Parent.Id = NULL, "icon-imodel-hollow-2", "icon-folder")`,
-            },
-            {
-                ruleType: "ImageIdOverride",
-                condition: `ThisNode.IsOfClass("Model", "BisCore")`,
-                imageIdExpression: `"icon-model"`,
-            },
         ],
     };
 }
 exports.createSearchRuleset = createSearchRuleset;
+/** @internal */
+function customizeModelsTreeNodeItem(item, node) {
+    item.isCheckboxVisible = true;
+    item.isCheckboxDisabled = true;
+    item.icon = getIcon(node);
+}
+exports.customizeModelsTreeNodeItem = customizeModelsTreeNodeItem;
+function getIcon(node) {
+    return node.key && presentation_common_1.NodeKey.isClassGroupingNodeKey(node.key)
+        ? node.extendedData?.groupIcon
+        : node.extendedData?.icon;
+}
 //# sourceMappingURL=Utils.js.map

package/lib/cjs/components/trees/models-tree/Utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"Utils.js","sourceRoot":"","sources":["../../../../../src/components/trees/models-tree/Utils.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;;AAM/F,gBAAgB;AAChB,MAAa,0BAA0B;IAErC,YAAoB,UAAsC;QAAtC,eAAU,GAAV,UAAU,CAA4B;QADlD,SAAI,GAAG,IAAI,KAAK,EAAc,CAAC;IAEvC,CAAC;IAEO,KAAK,CAAC,IAAI,KAAK,OAAO,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IAE9D,KAAK,CAAA,CAAE,aAAa;QACzB,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,IAAI,EAAE;YAC1B,MAAM,EAAE,CAAC;SACV;QAED,IAAI,MAAM,CAAC;QACX,OAAO,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,EAAE;YACjC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,MAAM,MAAM,CAAC;SACd;IACH,CAAC;CACF;AAlBD,gEAkBC;AAOD,gBAAgB;AAChB,SAAgB,aAAa,CAAC,KAAyB;IACrD,MAAM,yBAAyB,GAAG,KAAK,CAAC,yBAAyB,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,oBAAoB,EAAE,CAAC;IAChI,OAAO;QACL,EAAE,EAAE,8BAA8B;QAClC,eAAe,EAAE;YACf;gBACE,IAAI,EAAE,SAAS;aAChB;SACF;QACD,KAAK,EAAE;YACL;gBACE,QAAQ,EAAE,WAAW;gBACrB,UAAU,EAAE,IAAI;gBAChB,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP;gCACE,UAAU,EAAE,SAAS;gCACrB,UAAU,EAAE;oCACV,SAAS;iCACV;6BACF;yBACF;wBACD,cAAc,EAAE,oBAAoB;wBACpC,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,SAAS,EAAE,MAAM;yBAClB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,4CAA4C;gBACvD,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,qBAAqB;iCACjC;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE;oCACX,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,SAAS;iCACrB;6BACF;yBACF;wBACD,cAAc,EAAE,0JAA0J;wBAC1K,oBAAoB,EAAE,IAAI;wBAC1B,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;oBACD;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,qBAAqB;iCACjC;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE;oCACX,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,SAAS;iCACrB;6BACF;yBACF;wBACD,cAAc,EAAE,2JAA2J;wBAC3K,gBAAgB,EAAE,IAAI;wBACtB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,SAAS,EAAE,MAAM;yBAClB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,4CAA4C;gBACvD,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP,UAAU,EAAE,SAAS;4BACrB,UAAU,EAAE;gCACV,kBAAkB;6BACnB;4BACD,cAAc,EAAE,IAAI;yBACrB;wBACD,gBAAgB,EAAE;4BAChB;gCACE,gBAAgB,EAAE;oCAChB,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,oBAAoB;qCAChC;oCACD,SAAS,EAAE,SAAS;oCACpB,WAAW,EAAE;wCACX,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,6BAA6B;qCACzC;iCACF;gCACD,KAAK,EAAE,WAAW;gCAClB,UAAU,EAAE,IAAI;6BACjB;yBACF;wBACD,cAAc,EAAE,mbAAmb,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBAClhB,WAAW,EAAE,QAAQ;wBACrB,gBAAgB,EAAE,IAAI;wBACtB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;oBACD;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP,UAAU,EAAE,SAAS;4BACrB,UAAU,EAAE;gCACV,kBAAkB;6BACnB;4BACD,cAAc,EAAE,IAAI;yBACrB;wBACD,gBAAgB,EAAE;4BAChB;gCACE,gBAAgB,EAAE;oCAChB,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,oBAAoB;qCAChC;oCACD,SAAS,EAAE,SAAS;oCACpB,WAAW,EAAE;wCACX,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,6BAA6B;qCACzC;iCACF;gCACD,KAAK,EAAE,WAAW;gCAClB,UAAU,EAAE,IAAI;6BACjB;yBACF;wBACD,cAAc,EAAE,sbAAsb,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBACrhB,WAAW,EAAE,QAAQ;wBACrB,oBAAoB,EAAE,IAAI;wBAC1B,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,MAAM;yBAChB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,uDAAuD;gBAClE,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,oBAAoB;iCAChC;gCACD,SAAS,EAAE,UAAU;6BACtB;yBACF;wBACD,cAAc,EAAE,+FAA+F,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBAC9L,oBAAoB,EAAE,IAAI;wBAC1B,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,MAAM;yBAChB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,qDAAqD;gBAChE,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE;oCACE,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,uBAAuB;qCACnC;oCACD,SAAS,EAAE,SAAS;oCACpB,WAAW,EAAE,yBAAyB;iCACvC;gCACD;oCACE,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,gCAAgC;qCAC5C;oCACD,SAAS,EAAE,SAAS;iCACrB;6BACF;yBACF;wBACD,cAAc,EAAE,oBAAoB;wBACpC,6BAA6B,EAAE,IAAI;wBACnC,gBAAgB,EAAE,IAAI;wBACtB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,UAAU,EAAE,MAAM;4BAClB,OAAO,EAAE,uBAAuB;yBACjC;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,oDAAoD;gBAC/D,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,gCAAgC;iCAC5C;gCACD,SAAS,EAAE,UAAU;gCACrB,WAAW,EAAE,yBAAyB;6BACvC;yBACF;wBACD,cAAc,EAAE,uEAAuE;wBACvF,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,2BAA2B;wBACjD,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,eAAe;4BACxB,UAAU,EAAE,kBAAkB;yBAC/B;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,uDAAuD;gBAClE,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,0BAA0B;iCACtC;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE,yBAAyB;6BACvC;yBACF;wBACD,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,2BAA2B;wBACjD,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,eAAe;4BACxB,UAAU,EAAE,kBAAkB;yBAC/B;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,UAAU;gBACpB,KAAK,EAAE;oBACL,UAAU,EAAE,SAAS;oBACrB,SAAS,EAAE,SAAS;iBACrB;gBACD,MAAM,EAAE;oBACN;wBACE,QAAQ,EAAE,mBAAmB;wBAC7B,gBAAgB,EAAE,aAAa;qBAChC;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,UAAU;gBACpB,KAAK,EAAE;oBACL,UAAU,EAAE,SAAS;oBACrB,SAAS,EAAE,iBAAiB;iBAC7B;gBACD,MAAM,EAAE;oBACN;wBACE,QAAQ,EAAE,mBAAmB;wBAC7B,gBAAgB,EAAE,aAAa;qBAChC;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,0EAA0E;gBACrF,iBAAiB,EAAE,mEAAmE;aACvF;YACD;gBACE,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,wEAAwE;gBACnF,iBAAiB,EAAE,cAAc;aAClC;YACD;gBACE,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,2EAA2E;gBACtF,iBAAiB,EAAE,eAAe;aACnC;YACD;gBACE,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,0EAA0E;gBACrF,iBAAiB,EAAE,aAAa;aACjC;YACD;gBACE,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,8BAA8B;gBACzC,iBAAiB,EAAE,iBAAiB;aACrC;YACD;gBACE,QAAQ,EAAE,SAAS;gBACnB,SAAS,EAAE,gDAAgD;gBAC3D,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,uBAAuB;qBAClC;oBACD;wBACE,QAAQ,EAAE,yBAAyB;wBACnC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,0BAA0B;iCACtC;gCACD,SAAS,EAAE,SAAS;gCACpB,KAAK,EAAE,GAAG;6BACX;yBACF;qBACF;iBACF;aACF;SACF;KACF,CAAC;AACJ,CAAC;AA1XD,sCA0XC;AAKD,gBAAgB;AAChB,SAAgB,mBAAmB,CAAC,KAA+B;IACjE,MAAM,yBAAyB,GAAG,KAAK,CAAC,yBAAyB,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,oBAAoB,EAAE,CAAC;IAChI,OAAO;QACL,EAAE,EAAE,oCAAoC;QACxC,KAAK,EAAE;YACL;gBACE,QAAQ,EAAE,WAAW;gBACrB,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP;gCACE,UAAU,EAAE,SAAS;gCACrB,UAAU,EAAE;oCACV,SAAS;iCACV;6BACF;yBACF;wBACD,cAAc,EAAE,oBAAoB;wBACpC,cAAc,EAAE,KAAK;wBACrB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,SAAS,EAAE,MAAM;yBAClB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,4CAA4C;gBACvD,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,qBAAqB;iCACjC;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE;oCACX,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,SAAS;iCACrB;6BACF;yBACF;wBACD,cAAc,EAAE,0JAA0J;wBAC1K,oBAAoB,EAAE,IAAI;wBAC1B,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;oBACD;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,qBAAqB;iCACjC;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE;oCACX,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,SAAS;iCACrB;6BACF;yBACF;wBACD,cAAc,EAAE,2JAA2J;wBAC3K,cAAc,EAAE,mGAAmG;wBACnH,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,SAAS,EAAE,MAAM;yBAClB;qBACF;oBACD;wBACE,QAAQ,EAAE,UAAU;wBACpB,KAAK,EAAE;4BACL,UAAU,EAAE,SAAS;4BACrB,SAAS,EAAE,SAAS;yBACrB;wBACD,MAAM,EAAE;4BACN;gCACE,QAAQ,EAAE,mBAAmB;gCAC7B,gBAAgB,EAAE,aAAa;6BAChC;yBACF;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,4CAA4C;gBACvD,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP,UAAU,EAAE,SAAS;4BACrB,UAAU,EAAE;gCACV,kBAAkB;6BACnB;yBACF;wBACD,cAAc,EAAE,IAAI;wBACpB,gBAAgB,EAAE;4BAChB;gCACE,gBAAgB,EAAE;oCAChB,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,oBAAoB;qCAChC;oCACD,SAAS,EAAE,SAAS;oCACpB,WAAW,EAAE;wCACX,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,6BAA6B;qCACzC;iCACF;gCACD,KAAK,EAAE,WAAW;gCAClB,UAAU,EAAE,IAAI;6BACjB;yBACF;wBACD,cAAc,EAAE,mbAAmb,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBAClhB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,MAAM;yBAChB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,4CAA4C;gBACvD,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP,UAAU,EAAE,SAAS;4BACrB,UAAU,EAAE;gCACV,kBAAkB;6BACnB;yBACF;wBACD,cAAc,EAAE,IAAI;wBACpB,gBAAgB,EAAE;4BAChB;gCACE,gBAAgB,EAAE;oCAChB,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,oBAAoB;qCAChC;oCACD,SAAS,EAAE,SAAS;oCACpB,WAAW,EAAE;wCACX,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,6BAA6B;qCACzC;iCACF;gCACD,KAAK,EAAE,WAAW;gCAClB,UAAU,EAAE,IAAI;6BACjB;yBACF;wBACD,cAAc,EAAE,sbAAsb,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBACrhB,oBAAoB,EAAE,IAAI;wBAC1B,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,eAAe;wBACzB,KAAK,EAAE;4BACL,cAAc,EAAE,MAAM;yBACvB;qBACF;oBACD;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,MAAM;yBAChB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,qDAAqD;gBAChE,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,mBAAmB;iCAC/B;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE;oCACX,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,kBAAkB;iCAC9B;6BACF;yBACF;wBACD,cAAc,EAAE,+FAA+F,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBAC9L,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,MAAM;yBAChB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,0CAA0C;gBACrD,iBAAiB,EAAE,mEAAmE;aACvF;YACD;gBACE,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,wCAAwC;gBACnD,iBAAiB,EAAE,cAAc;aAClC;SACF;KACF,CAAC;AACJ,CAAC;AAhPD,kDAgPC","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 type { Id64String } from \"@itwin/core-bentley\";\nimport type { Ruleset } from \"@itwin/presentation-common\";\nimport type { ModelsTreeHierarchyConfiguration } from \"./ModelsTree\";\n\n/** @internal */\nexport class CachingElementIdsContainer {\n  private _ids = new Array<Id64String>();\n  constructor(private _generator: AsyncGenerator<Id64String>) {\n  }\n\n  private async next() { return (await this._generator.next()).value; }\n\n  public async* getElementIds() {\n    for (const id of this._ids) {\n      yield id;\n    }\n\n    let nextId;\n    while (nextId = await this.next()) {\n      this._ids.push(nextId);\n      yield nextId;\n    }\n  }\n}\n\n/** @internal */\nexport type CreateRulesetProps = Omit<ModelsTreeHierarchyConfiguration, \"enableElementsClassGrouping\"> & {\n  enableElementsClassGrouping?: boolean;\n};\n\n/** @internal */\nexport function createRuleset(props: CreateRulesetProps): Ruleset {\n  const elementClassSpecification = props.elementClassSpecification ?? { schemaName: \"BisCore\", className: \"GeometricElement3d\" };\n  return {\n    id: \"tree-widget-react/ModelsTree\",\n    requiredSchemas: [\n      {\n        name: \"BisCore\",\n      },\n    ],\n    rules: [\n      {\n        ruleType: \"RootNodes\",\n        autoExpand: true,\n        specifications: [\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: [\n              {\n                schemaName: \"BisCore\",\n                classNames: [\n                  \"Subject\",\n                ],\n              },\n            ],\n            instanceFilter: `this.Parent = NULL`,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isSubject: \"true\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"SubjectOwnsSubjects\",\n                },\n                direction: \"Forward\",\n                targetClass: {\n                  schemaName: \"BisCore\",\n                  className: \"Subject\",\n                },\n              },\n            ],\n            instanceFilter: `json_extract(this.JsonProperties, \"$.Subject.Job.Bridge\") <> NULL OR ifnull(json_extract(this.JsonProperties, \"$.Subject.Model.Type\"), \"\") = \"Hierarchy\"`,\n            hideNodesInHierarchy: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"SubjectOwnsSubjects\",\n                },\n                direction: \"Forward\",\n                targetClass: {\n                  schemaName: \"BisCore\",\n                  className: \"Subject\",\n                },\n              },\n            ],\n            instanceFilter: `json_extract(this.JsonProperties, \"$.Subject.Job.Bridge\") = NULL AND ifnull(json_extract(this.JsonProperties, \"$.Subject.Model.Type\"), \"\") <> \"Hierarchy\"`,\n            hideIfNoChildren: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isSubject: \"true\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: {\n              schemaName: \"BisCore\",\n              classNames: [\n                \"GeometricModel3d\",\n              ],\n              arePolymorphic: true,\n            },\n            relatedInstances: [\n              {\n                relationshipPath: {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"ModelModelsElement\",\n                  },\n                  direction: \"Forward\",\n                  targetClass: {\n                    schemaName: \"BisCore\",\n                    className: \"InformationPartitionElement\",\n                  },\n                },\n                alias: \"partition\",\n                isRequired: true,\n              },\n            ],\n            instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, \"$.Subject.Model.TargetPartition\") = printf(\"0x%x\", partition.ECInstanceId)) AND NOT this.IsPrivate AND json_extract(partition.JsonProperties, \"$.PhysicalPartition.Model.Content\") = NULL AND json_extract(partition.JsonProperties, \"$.GraphicalPartition3d.Model.Content\") = NULL AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            hasChildren: \"Always\",\n            hideIfNoChildren: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: {\n              schemaName: \"BisCore\",\n              classNames: [\n                \"GeometricModel3d\",\n              ],\n              arePolymorphic: true,\n            },\n            relatedInstances: [\n              {\n                relationshipPath: {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"ModelModelsElement\",\n                  },\n                  direction: \"Forward\",\n                  targetClass: {\n                    schemaName: \"BisCore\",\n                    className: \"InformationPartitionElement\",\n                  },\n                },\n                alias: \"partition\",\n                isRequired: true,\n              },\n            ],\n            instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, \"$.Subject.Model.TargetPartition\") = printf(\"0x%x\", partition.ECInstanceId)) AND NOT this.IsPrivate AND (json_extract(partition.JsonProperties, \"$.PhysicalPartition.Model.Content\") <> NULL OR json_extract(partition.JsonProperties, \"$.GraphicalPartition3d.Model.Content\") <> NULL) AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            hasChildren: \"Always\",\n            hideNodesInHierarchy: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isModel: \"true\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"ISubModeledElement\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"ModelModelsElement\",\n                },\n                direction: \"Backward\",\n              },\n            ],\n            instanceFilter: `NOT this.IsPrivate AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            hideNodesInHierarchy: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isModel: \"true\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"GeometricModel3d\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              [\n                {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"ModelContainsElements\",\n                  },\n                  direction: \"Forward\",\n                  targetClass: elementClassSpecification,\n                },\n                {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"GeometricElement3dIsInCategory\",\n                  },\n                  direction: \"Forward\",\n                },\n              ],\n            ],\n            instanceFilter: `NOT this.IsPrivate`,\n            suppressSimilarAncestorsCheck: true,\n            hideIfNoChildren: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isCategory: \"true\",\n              modelId: \"ParentNode.InstanceId\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"SpatialCategory\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"GeometricElement3dIsInCategory\",\n                },\n                direction: \"Backward\",\n                targetClass: elementClassSpecification,\n              },\n            ],\n            instanceFilter: `this.Model.Id = parent.parent.ECInstanceId ANDALSO this.Parent = NULL`,\n            groupByClass: !!props.enableElementsClassGrouping,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              modelId: \"this.Model.Id\",\n              categoryId: \"this.Category.Id\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"GeometricElement3d\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"ElementOwnsChildElements\",\n                },\n                direction: \"Forward\",\n                targetClass: elementClassSpecification,\n              },\n            ],\n            groupByClass: !!props.enableElementsClassGrouping,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              modelId: \"this.Model.Id\",\n              categoryId: \"this.Category.Id\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"Grouping\",\n        class: {\n          schemaName: \"BisCore\",\n          className: \"Subject\",\n        },\n        groups: [\n          {\n            specType: \"SameLabelInstance\",\n            applicationStage: \"PostProcess\",\n          },\n        ],\n      },\n      {\n        ruleType: \"Grouping\",\n        class: {\n          schemaName: \"BisCore\",\n          className: \"SpatialCategory\",\n        },\n        groups: [\n          {\n            specType: \"SameLabelInstance\",\n            applicationStage: \"PostProcess\",\n          },\n        ],\n      },\n      {\n        ruleType: \"ImageIdOverride\",\n        condition: `ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        imageIdExpression: `IIF(this.Parent.Id = NULL, \"icon-imodel-hollow-2\", \"icon-folder\")`,\n      },\n      {\n        ruleType: \"ImageIdOverride\",\n        condition: `ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass(\"Model\", \"BisCore\")`,\n        imageIdExpression: `\"icon-model\"`,\n      },\n      {\n        ruleType: \"ImageIdOverride\",\n        condition: `ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass(\"Category\", \"BisCore\")`,\n        imageIdExpression: `\"icon-layers\"`,\n      },\n      {\n        ruleType: \"ImageIdOverride\",\n        condition: `ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass(\"Element\", \"BisCore\")`,\n        imageIdExpression: `\"icon-item\"`,\n      },\n      {\n        ruleType: \"ImageIdOverride\",\n        condition: `ThisNode.IsClassGroupingNode`,\n        imageIdExpression: `\"icon-ec-class\"`,\n      },\n      {\n        ruleType: \"Content\",\n        condition: `ContentDisplayType = \"AssemblyElementsRequest\"`,\n        specifications: [\n          {\n            specType: \"SelectedNodeInstances\",\n          },\n          {\n            specType: \"ContentRelatedInstances\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"ElementOwnsChildElements\",\n                },\n                direction: \"Forward\",\n                count: \"*\",\n              },\n            ],\n          },\n        ],\n      },\n    ],\n  };\n}\n\n/** @internal */\nexport type CreateSearchRulesetProps = Omit<ModelsTreeHierarchyConfiguration, \"enableElementsClassGrouping\">;\n\n/** @internal */\nexport function createSearchRuleset(props: CreateSearchRulesetProps): Ruleset {\n  const elementClassSpecification = props.elementClassSpecification ?? { schemaName: \"BisCore\", className: \"GeometricElement3d\" };\n  return {\n    id: \"tree-widget-react/ModelsTreeSearch\",\n    rules: [\n      {\n        ruleType: \"RootNodes\",\n        specifications: [\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: [\n              {\n                schemaName: \"BisCore\",\n                classNames: [\n                  \"Subject\",\n                ],\n              },\n            ],\n            instanceFilter: \"this.Parent = NULL\",\n            arePolymorphic: false,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isSubject: \"true\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"SubjectOwnsSubjects\",\n                },\n                direction: \"Forward\",\n                targetClass: {\n                  schemaName: \"BisCore\",\n                  className: \"Subject\",\n                },\n              },\n            ],\n            instanceFilter: `json_extract(this.JsonProperties, \"$.Subject.Job.Bridge\") <> NULL OR ifnull(json_extract(this.JsonProperties, \"$.Subject.Model.Type\"), \"\") = \"Hierarchy\"`,\n            hideNodesInHierarchy: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"SubjectOwnsSubjects\",\n                },\n                direction: \"Forward\",\n                targetClass: {\n                  schemaName: \"BisCore\",\n                  className: \"Subject\",\n                },\n              },\n            ],\n            instanceFilter: `json_extract(this.JsonProperties, \"$.Subject.Job.Bridge\") = NULL AND ifnull(json_extract(this.JsonProperties, \"$.Subject.Model.Type\"), \"\") <> \"Hierarchy\"`,\n            hideExpression: `NOT ThisNode.HasChildren ANDALSO NOT ThisNode.ChildrenArtifacts.AnyMatches(x => x.isContentModel)`,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isSubject: \"true\",\n            },\n          },\n          {\n            ruleType: \"Grouping\",\n            class: {\n              schemaName: \"BisCore\",\n              className: \"Subject\",\n            },\n            groups: [\n              {\n                specType: \"SameLabelInstance\",\n                applicationStage: \"PostProcess\",\n              },\n            ],\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: {\n              schemaName: \"BisCore\",\n              classNames: [\n                \"GeometricModel3d\",\n              ],\n            },\n            arePolymorphic: true,\n            relatedInstances: [\n              {\n                relationshipPath: {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"ModelModelsElement\",\n                  },\n                  direction: \"Forward\",\n                  targetClass: {\n                    schemaName: \"BisCore\",\n                    className: \"InformationPartitionElement\",\n                  },\n                },\n                alias: \"partition\",\n                isRequired: true,\n              },\n            ],\n            instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, \"$.Subject.Model.TargetPartition\") = printf(\"0x%x\", partition.ECInstanceId)) AND NOT this.IsPrivate AND json_extract(partition.JsonProperties, \"$.PhysicalPartition.Model.Content\") = NULL AND json_extract(partition.JsonProperties, \"$.GraphicalPartition3d.Model.Content\") = NULL AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isModel: \"true\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: {\n              schemaName: \"BisCore\",\n              classNames: [\n                \"GeometricModel3d\",\n              ],\n            },\n            arePolymorphic: true,\n            relatedInstances: [\n              {\n                relationshipPath: {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"ModelModelsElement\",\n                  },\n                  direction: \"Forward\",\n                  targetClass: {\n                    schemaName: \"BisCore\",\n                    className: \"InformationPartitionElement\",\n                  },\n                },\n                alias: \"partition\",\n                isRequired: true,\n              },\n            ],\n            instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, \"$.Subject.Model.TargetPartition\") = printf(\"0x%x\", partition.ECInstanceId)) AND NOT this.IsPrivate AND (json_extract(partition.JsonProperties, \"$.PhysicalPartition.Model.Content\") <> NULL OR json_extract(partition.JsonProperties, \"$.GraphicalPartition3d.Model.Content\") <> NULL) AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            hideNodesInHierarchy: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"NodeArtifacts\",\n            items: {\n              isContentModel: \"true\",\n            },\n          },\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isModel: \"true\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"GeometricModel3d\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"ModelOwnsSubModel\",\n                },\n                direction: \"Forward\",\n                targetClass: {\n                  schemaName: \"BisCore\",\n                  className: \"GeometricModel3d\",\n                },\n              },\n            ],\n            instanceFilter: `NOT this.IsPrivate AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isModel: \"true\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ImageIdOverride\",\n        condition: `ThisNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        imageIdExpression: `IIF(this.Parent.Id = NULL, \"icon-imodel-hollow-2\", \"icon-folder\")`,\n      },\n      {\n        ruleType: \"ImageIdOverride\",\n        condition: `ThisNode.IsOfClass(\"Model\", \"BisCore\")`,\n        imageIdExpression: `\"icon-model\"`,\n      },\n    ],\n  };\n}\n"]}
+{"version":3,"file":"Utils.js","sourceRoot":"","sources":["../../../../../src/components/trees/models-tree/Utils.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;;AAE/F,oEAAqD;AAOrD,gBAAgB;AAChB,MAAa,0BAA0B;IAErC,YAAoB,UAAsC;QAAtC,eAAU,GAAV,UAAU,CAA4B;QADlD,SAAI,GAAG,IAAI,KAAK,EAAc,CAAC;IAEvC,CAAC;IAEO,KAAK,CAAC,IAAI,KAAK,OAAO,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IAE9D,KAAK,CAAA,CAAE,aAAa;QACzB,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,IAAI,EAAE;YAC1B,MAAM,EAAE,CAAC;SACV;QAED,IAAI,MAAM,CAAC;QACX,OAAO,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,EAAE;YACjC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,MAAM,MAAM,CAAC;SACd;IACH,CAAC;CACF;AAlBD,gEAkBC;AAOD,gBAAgB;AAChB,SAAgB,aAAa,CAAC,KAAyB;IACrD,MAAM,yBAAyB,GAAG,KAAK,CAAC,yBAAyB,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,oBAAoB,EAAE,CAAC;IAChI,OAAO;QACL,EAAE,EAAE,8BAA8B;QAClC,eAAe,EAAE;YACf;gBACE,IAAI,EAAE,SAAS;aAChB;SACF;QACD,KAAK,EAAE;YACL;gBACE,QAAQ,EAAE,WAAW;gBACrB,UAAU,EAAE,IAAI;gBAChB,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP;gCACE,UAAU,EAAE,SAAS;gCACrB,UAAU,EAAE;oCACV,SAAS;iCACV;6BACF;yBACF;wBACD,cAAc,EAAE,oBAAoB;wBACpC,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,SAAS,EAAE,MAAM;4BACjB,IAAI,EAAE,0BAA0B;yBACjC;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,4CAA4C;gBACvD,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,qBAAqB;iCACjC;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE;oCACX,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,SAAS;iCACrB;6BACF;yBACF;wBACD,cAAc,EAAE,0JAA0J;wBAC1K,oBAAoB,EAAE,IAAI;wBAC1B,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;oBACD;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,qBAAqB;iCACjC;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE;oCACX,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,SAAS;iCACrB;6BACF;yBACF;wBACD,cAAc,EAAE,2JAA2J;wBAC3K,gBAAgB,EAAE,IAAI;wBACtB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,SAAS,EAAE,MAAM;4BACjB,IAAI,EAAE,iBAAiB;yBACxB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,4CAA4C;gBACvD,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP,UAAU,EAAE,SAAS;4BACrB,UAAU,EAAE;gCACV,kBAAkB;6BACnB;4BACD,cAAc,EAAE,IAAI;yBACrB;wBACD,gBAAgB,EAAE;4BAChB;gCACE,gBAAgB,EAAE;oCAChB,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,oBAAoB;qCAChC;oCACD,SAAS,EAAE,SAAS;oCACpB,WAAW,EAAE;wCACX,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,6BAA6B;qCACzC;iCACF;gCACD,KAAK,EAAE,WAAW;gCAClB,UAAU,EAAE,IAAI;6BACjB;yBACF;wBACD,cAAc,EAAE,mbAAmb,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBAClhB,WAAW,EAAE,QAAQ;wBACrB,gBAAgB,EAAE,IAAI;wBACtB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;oBACD;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP,UAAU,EAAE,SAAS;4BACrB,UAAU,EAAE;gCACV,kBAAkB;6BACnB;4BACD,cAAc,EAAE,IAAI;yBACrB;wBACD,gBAAgB,EAAE;4BAChB;gCACE,gBAAgB,EAAE;oCAChB,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,oBAAoB;qCAChC;oCACD,SAAS,EAAE,SAAS;oCACpB,WAAW,EAAE;wCACX,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,6BAA6B;qCACzC;iCACF;gCACD,KAAK,EAAE,WAAW;gCAClB,UAAU,EAAE,IAAI;6BACjB;yBACF;wBACD,cAAc,EAAE,sbAAsb,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBACrhB,WAAW,EAAE,QAAQ;wBACrB,oBAAoB,EAAE,IAAI;wBAC1B,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,MAAM;4BACf,IAAI,EAAE,gBAAgB;yBACvB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,uDAAuD;gBAClE,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,oBAAoB;iCAChC;gCACD,SAAS,EAAE,UAAU;6BACtB;yBACF;wBACD,cAAc,EAAE,+FAA+F,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBAC9L,oBAAoB,EAAE,IAAI;wBAC1B,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,MAAM;4BACf,IAAI,EAAE,gBAAgB;yBACvB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,qDAAqD;gBAChE,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE;oCACE,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,uBAAuB;qCACnC;oCACD,SAAS,EAAE,SAAS;oCACpB,WAAW,EAAE,yBAAyB;iCACvC;gCACD;oCACE,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,gCAAgC;qCAC5C;oCACD,SAAS,EAAE,SAAS;iCACrB;6BACF;yBACF;wBACD,cAAc,EAAE,oBAAoB;wBACpC,6BAA6B,EAAE,IAAI;wBACnC,gBAAgB,EAAE,IAAI;wBACtB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,UAAU,EAAE,MAAM;4BAClB,OAAO,EAAE,uBAAuB;4BAChC,IAAI,EAAE,iBAAiB;yBACxB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,oDAAoD;gBAC/D,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,gCAAgC;iCAC5C;gCACD,SAAS,EAAE,UAAU;gCACrB,WAAW,EAAE,yBAAyB;6BACvC;yBACF;wBACD,cAAc,EAAE,uEAAuE;wBACvF,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,2BAA2B;wBACjD,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,eAAe;4BACxB,UAAU,EAAE,kBAAkB;4BAC9B,IAAI,EAAE,eAAe;4BACrB,SAAS,EAAE,mBAAmB;yBAC/B;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,uDAAuD;gBAClE,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,0BAA0B;iCACtC;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE,yBAAyB;6BACvC;yBACF;wBACD,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,2BAA2B;wBACjD,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,eAAe;4BACxB,UAAU,EAAE,kBAAkB;4BAC9B,IAAI,EAAE,eAAe;4BACrB,SAAS,EAAE,mBAAmB;yBAC/B;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,UAAU;gBACpB,KAAK,EAAE;oBACL,UAAU,EAAE,SAAS;oBACrB,SAAS,EAAE,SAAS;iBACrB;gBACD,MAAM,EAAE;oBACN;wBACE,QAAQ,EAAE,mBAAmB;wBAC7B,gBAAgB,EAAE,aAAa;qBAChC;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,UAAU;gBACpB,KAAK,EAAE;oBACL,UAAU,EAAE,SAAS;oBACrB,SAAS,EAAE,iBAAiB;iBAC7B;gBACD,MAAM,EAAE;oBACN;wBACE,QAAQ,EAAE,mBAAmB;wBAC7B,gBAAgB,EAAE,aAAa;qBAChC;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,SAAS;gBACnB,SAAS,EAAE,gDAAgD;gBAC3D,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,uBAAuB;qBAClC;oBACD;wBACE,QAAQ,EAAE,yBAAyB;wBACnC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,0BAA0B;iCACtC;gCACD,SAAS,EAAE,SAAS;gCACpB,KAAK,EAAE,GAAG;6BACX;yBACF;qBACF;iBACF;aACF;SACF;KACF,CAAC;AACJ,CAAC;AA1WD,sCA0WC;AAKD,gBAAgB;AAChB,SAAgB,mBAAmB,CAAC,KAA+B;IACjE,MAAM,yBAAyB,GAAG,KAAK,CAAC,yBAAyB,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,oBAAoB,EAAE,CAAC;IAChI,OAAO;QACL,EAAE,EAAE,oCAAoC;QACxC,KAAK,EAAE;YACL;gBACE,QAAQ,EAAE,WAAW;gBACrB,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP;gCACE,UAAU,EAAE,SAAS;gCACrB,UAAU,EAAE;oCACV,SAAS;iCACV;6BACF;yBACF;wBACD,cAAc,EAAE,oBAAoB;wBACpC,cAAc,EAAE,KAAK;wBACrB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,SAAS,EAAE,MAAM;4BACjB,IAAI,EAAE,0BAA0B;yBACjC;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,4CAA4C;gBACvD,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,qBAAqB;iCACjC;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE;oCACX,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,SAAS;iCACrB;6BACF;yBACF;wBACD,cAAc,EAAE,0JAA0J;wBAC1K,oBAAoB,EAAE,IAAI;wBAC1B,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;oBACD;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,qBAAqB;iCACjC;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE;oCACX,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,SAAS;iCACrB;6BACF;yBACF;wBACD,cAAc,EAAE,2JAA2J;wBAC3K,cAAc,EAAE,mGAAmG;wBACnH,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,SAAS,EAAE,MAAM;4BACjB,IAAI,EAAE,iBAAiB;yBACxB;qBACF;oBACD;wBACE,QAAQ,EAAE,UAAU;wBACpB,KAAK,EAAE;4BACL,UAAU,EAAE,SAAS;4BACrB,SAAS,EAAE,SAAS;yBACrB;wBACD,MAAM,EAAE;4BACN;gCACE,QAAQ,EAAE,mBAAmB;gCAC7B,gBAAgB,EAAE,aAAa;6BAChC;yBACF;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,4CAA4C;gBACvD,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP,UAAU,EAAE,SAAS;4BACrB,UAAU,EAAE;gCACV,kBAAkB;6BACnB;yBACF;wBACD,cAAc,EAAE,IAAI;wBACpB,gBAAgB,EAAE;4BAChB;gCACE,gBAAgB,EAAE;oCAChB,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,oBAAoB;qCAChC;oCACD,SAAS,EAAE,SAAS;oCACpB,WAAW,EAAE;wCACX,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,6BAA6B;qCACzC;iCACF;gCACD,KAAK,EAAE,WAAW;gCAClB,UAAU,EAAE,IAAI;6BACjB;yBACF;wBACD,cAAc,EAAE,mbAAmb,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBAClhB,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,MAAM;4BACf,IAAI,EAAE,gBAAgB;yBACvB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,4CAA4C;gBACvD,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,gCAAgC;wBAC1C,OAAO,EAAE;4BACP,UAAU,EAAE,SAAS;4BACrB,UAAU,EAAE;gCACV,kBAAkB;6BACnB;yBACF;wBACD,cAAc,EAAE,IAAI;wBACpB,gBAAgB,EAAE;4BAChB;gCACE,gBAAgB,EAAE;oCAChB,YAAY,EAAE;wCACZ,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,oBAAoB;qCAChC;oCACD,SAAS,EAAE,SAAS;oCACpB,WAAW,EAAE;wCACX,UAAU,EAAE,SAAS;wCACrB,SAAS,EAAE,6BAA6B;qCACzC;iCACF;gCACD,KAAK,EAAE,WAAW;gCAClB,UAAU,EAAE,IAAI;6BACjB;yBACF;wBACD,cAAc,EAAE,sbAAsb,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBACrhB,oBAAoB,EAAE,IAAI;wBAC1B,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,eAAe;wBACzB,KAAK,EAAE;4BACL,cAAc,EAAE,MAAM;yBACvB;qBACF;oBACD;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,MAAM;4BACf,IAAI,EAAE,gBAAgB;yBACvB;qBACF;iBACF;aACF;YACD;gBACE,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,qDAAqD;gBAChE,cAAc,EAAE;oBACd;wBACE,QAAQ,EAAE,sBAAsB;wBAChC,iBAAiB,EAAE;4BACjB;gCACE,YAAY,EAAE;oCACZ,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,mBAAmB;iCAC/B;gCACD,SAAS,EAAE,SAAS;gCACpB,WAAW,EAAE;oCACX,UAAU,EAAE,SAAS;oCACrB,SAAS,EAAE,kBAAkB;iCAC9B;6BACF;yBACF;wBACD,cAAc,EAAE,+FAA+F,yBAAyB,CAAC,UAAU,IAAI,yBAAyB,CAAC,SAAS,IAAI;wBAC9L,YAAY,EAAE,KAAK;wBACnB,YAAY,EAAE,KAAK;qBACpB;iBACF;gBACD,kBAAkB,EAAE;oBAClB;wBACE,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE;4BACL,OAAO,EAAE,MAAM;4BACf,IAAI,EAAE,gBAAgB;yBACvB;qBACF;iBACF;aACF;SACF;KACF,CAAC;AACJ,CAAC;AA3OD,kDA2OC;AAED,gBAAgB;AAChB,SAAgB,2BAA2B,CAAC,IAAsC,EAAE,IAAmB;IACrG,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;IAC/B,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC;AAJD,kEAIC;AAED,SAAS,OAAO,CAAC,IAAmB;IAClC,OAAO,IAAI,CAAC,GAAG,IAAI,6BAAO,CAAC,sBAAsB,CAAC,IAAI,CAAC,GAAG,CAAC;QACzD,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,SAAS;QAC9B,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC;AAC9B,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 { NodeKey } from \"@itwin/presentation-common\";\n\nimport type { Id64String } from \"@itwin/core-bentley\";\nimport type { DelayLoadedTreeNodeItem } from \"@itwin/components-react\";\nimport type { Node, Ruleset } from \"@itwin/presentation-common\";\nimport type { ModelsTreeHierarchyConfiguration } from \"./ModelsTree\";\n\n/** @internal */\nexport class CachingElementIdsContainer {\n  private _ids = new Array<Id64String>();\n  constructor(private _generator: AsyncGenerator<Id64String>) {\n  }\n\n  private async next() { return (await this._generator.next()).value; }\n\n  public async* getElementIds() {\n    for (const id of this._ids) {\n      yield id;\n    }\n\n    let nextId;\n    while (nextId = await this.next()) {\n      this._ids.push(nextId);\n      yield nextId;\n    }\n  }\n}\n\n/** @internal */\nexport type CreateRulesetProps = Omit<ModelsTreeHierarchyConfiguration, \"enableElementsClassGrouping\"> & {\n  enableElementsClassGrouping?: boolean;\n};\n\n/** @internal */\nexport function createRuleset(props: CreateRulesetProps): Ruleset {\n  const elementClassSpecification = props.elementClassSpecification ?? { schemaName: \"BisCore\", className: \"GeometricElement3d\" };\n  return {\n    id: \"tree-widget-react/ModelsTree\",\n    requiredSchemas: [\n      {\n        name: \"BisCore\",\n      },\n    ],\n    rules: [\n      {\n        ruleType: \"RootNodes\",\n        autoExpand: true,\n        specifications: [\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: [\n              {\n                schemaName: \"BisCore\",\n                classNames: [\n                  \"Subject\",\n                ],\n              },\n            ],\n            instanceFilter: `this.Parent = NULL`,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isSubject: \"true\",\n              icon: \"\\\"icon-imodel-hollow-2\\\"\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"SubjectOwnsSubjects\",\n                },\n                direction: \"Forward\",\n                targetClass: {\n                  schemaName: \"BisCore\",\n                  className: \"Subject\",\n                },\n              },\n            ],\n            instanceFilter: `json_extract(this.JsonProperties, \"$.Subject.Job.Bridge\") <> NULL OR ifnull(json_extract(this.JsonProperties, \"$.Subject.Model.Type\"), \"\") = \"Hierarchy\"`,\n            hideNodesInHierarchy: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"SubjectOwnsSubjects\",\n                },\n                direction: \"Forward\",\n                targetClass: {\n                  schemaName: \"BisCore\",\n                  className: \"Subject\",\n                },\n              },\n            ],\n            instanceFilter: `json_extract(this.JsonProperties, \"$.Subject.Job.Bridge\") = NULL AND ifnull(json_extract(this.JsonProperties, \"$.Subject.Model.Type\"), \"\") <> \"Hierarchy\"`,\n            hideIfNoChildren: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isSubject: \"true\",\n              icon: \"\\\"icon-folder\\\"\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: {\n              schemaName: \"BisCore\",\n              classNames: [\n                \"GeometricModel3d\",\n              ],\n              arePolymorphic: true,\n            },\n            relatedInstances: [\n              {\n                relationshipPath: {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"ModelModelsElement\",\n                  },\n                  direction: \"Forward\",\n                  targetClass: {\n                    schemaName: \"BisCore\",\n                    className: \"InformationPartitionElement\",\n                  },\n                },\n                alias: \"partition\",\n                isRequired: true,\n              },\n            ],\n            instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, \"$.Subject.Model.TargetPartition\") = printf(\"0x%x\", partition.ECInstanceId)) AND NOT this.IsPrivate AND json_extract(partition.JsonProperties, \"$.PhysicalPartition.Model.Content\") = NULL AND json_extract(partition.JsonProperties, \"$.GraphicalPartition3d.Model.Content\") = NULL AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            hasChildren: \"Always\",\n            hideIfNoChildren: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: {\n              schemaName: \"BisCore\",\n              classNames: [\n                \"GeometricModel3d\",\n              ],\n              arePolymorphic: true,\n            },\n            relatedInstances: [\n              {\n                relationshipPath: {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"ModelModelsElement\",\n                  },\n                  direction: \"Forward\",\n                  targetClass: {\n                    schemaName: \"BisCore\",\n                    className: \"InformationPartitionElement\",\n                  },\n                },\n                alias: \"partition\",\n                isRequired: true,\n              },\n            ],\n            instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, \"$.Subject.Model.TargetPartition\") = printf(\"0x%x\", partition.ECInstanceId)) AND NOT this.IsPrivate AND (json_extract(partition.JsonProperties, \"$.PhysicalPartition.Model.Content\") <> NULL OR json_extract(partition.JsonProperties, \"$.GraphicalPartition3d.Model.Content\") <> NULL) AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            hasChildren: \"Always\",\n            hideNodesInHierarchy: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isModel: \"true\",\n              icon: \"\\\"icon-model\\\"\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"ISubModeledElement\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"ModelModelsElement\",\n                },\n                direction: \"Backward\",\n              },\n            ],\n            instanceFilter: `NOT this.IsPrivate AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            hideNodesInHierarchy: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isModel: \"true\",\n              icon: \"\\\"icon-model\\\"\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"GeometricModel3d\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              [\n                {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"ModelContainsElements\",\n                  },\n                  direction: \"Forward\",\n                  targetClass: elementClassSpecification,\n                },\n                {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"GeometricElement3dIsInCategory\",\n                  },\n                  direction: \"Forward\",\n                },\n              ],\n            ],\n            instanceFilter: `NOT this.IsPrivate`,\n            suppressSimilarAncestorsCheck: true,\n            hideIfNoChildren: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isCategory: \"true\",\n              modelId: \"ParentNode.InstanceId\",\n              icon: \"\\\"icon-layers\\\"\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"SpatialCategory\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"GeometricElement3dIsInCategory\",\n                },\n                direction: \"Backward\",\n                targetClass: elementClassSpecification,\n              },\n            ],\n            instanceFilter: `this.Model.Id = parent.parent.ECInstanceId ANDALSO this.Parent = NULL`,\n            groupByClass: !!props.enableElementsClassGrouping,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              modelId: \"this.Model.Id\",\n              categoryId: \"this.Category.Id\",\n              icon: \"\\\"icon-item\\\"\",\n              groupIcon: \"\\\"icon-ec-class\\\"\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"GeometricElement3d\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"ElementOwnsChildElements\",\n                },\n                direction: \"Forward\",\n                targetClass: elementClassSpecification,\n              },\n            ],\n            groupByClass: !!props.enableElementsClassGrouping,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              modelId: \"this.Model.Id\",\n              categoryId: \"this.Category.Id\",\n              icon: \"\\\"icon-item\\\"\",\n              groupIcon: \"\\\"icon-ec-class\\\"\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"Grouping\",\n        class: {\n          schemaName: \"BisCore\",\n          className: \"Subject\",\n        },\n        groups: [\n          {\n            specType: \"SameLabelInstance\",\n            applicationStage: \"PostProcess\",\n          },\n        ],\n      },\n      {\n        ruleType: \"Grouping\",\n        class: {\n          schemaName: \"BisCore\",\n          className: \"SpatialCategory\",\n        },\n        groups: [\n          {\n            specType: \"SameLabelInstance\",\n            applicationStage: \"PostProcess\",\n          },\n        ],\n      },\n      {\n        ruleType: \"Content\",\n        condition: `ContentDisplayType = \"AssemblyElementsRequest\"`,\n        specifications: [\n          {\n            specType: \"SelectedNodeInstances\",\n          },\n          {\n            specType: \"ContentRelatedInstances\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"ElementOwnsChildElements\",\n                },\n                direction: \"Forward\",\n                count: \"*\",\n              },\n            ],\n          },\n        ],\n      },\n    ],\n  };\n}\n\n/** @internal */\nexport type CreateSearchRulesetProps = Omit<ModelsTreeHierarchyConfiguration, \"enableElementsClassGrouping\">;\n\n/** @internal */\nexport function createSearchRuleset(props: CreateSearchRulesetProps): Ruleset {\n  const elementClassSpecification = props.elementClassSpecification ?? { schemaName: \"BisCore\", className: \"GeometricElement3d\" };\n  return {\n    id: \"tree-widget-react/ModelsTreeSearch\",\n    rules: [\n      {\n        ruleType: \"RootNodes\",\n        specifications: [\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: [\n              {\n                schemaName: \"BisCore\",\n                classNames: [\n                  \"Subject\",\n                ],\n              },\n            ],\n            instanceFilter: \"this.Parent = NULL\",\n            arePolymorphic: false,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isSubject: \"true\",\n              icon: \"\\\"icon-imodel-hollow-2\\\"\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"SubjectOwnsSubjects\",\n                },\n                direction: \"Forward\",\n                targetClass: {\n                  schemaName: \"BisCore\",\n                  className: \"Subject\",\n                },\n              },\n            ],\n            instanceFilter: `json_extract(this.JsonProperties, \"$.Subject.Job.Bridge\") <> NULL OR ifnull(json_extract(this.JsonProperties, \"$.Subject.Model.Type\"), \"\") = \"Hierarchy\"`,\n            hideNodesInHierarchy: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"SubjectOwnsSubjects\",\n                },\n                direction: \"Forward\",\n                targetClass: {\n                  schemaName: \"BisCore\",\n                  className: \"Subject\",\n                },\n              },\n            ],\n            instanceFilter: `json_extract(this.JsonProperties, \"$.Subject.Job.Bridge\") = NULL AND ifnull(json_extract(this.JsonProperties, \"$.Subject.Model.Type\"), \"\") <> \"Hierarchy\"`,\n            hideExpression: `NOT ThisNode.HasChildren ANDALSO NOT ThisNode.ChildrenArtifacts.AnyMatches(x => x.isContentModel)`,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isSubject: \"true\",\n              icon: \"\\\"icon-folder\\\"\",\n            },\n          },\n          {\n            ruleType: \"Grouping\",\n            class: {\n              schemaName: \"BisCore\",\n              className: \"Subject\",\n            },\n            groups: [\n              {\n                specType: \"SameLabelInstance\",\n                applicationStage: \"PostProcess\",\n              },\n            ],\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: {\n              schemaName: \"BisCore\",\n              classNames: [\n                \"GeometricModel3d\",\n              ],\n            },\n            arePolymorphic: true,\n            relatedInstances: [\n              {\n                relationshipPath: {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"ModelModelsElement\",\n                  },\n                  direction: \"Forward\",\n                  targetClass: {\n                    schemaName: \"BisCore\",\n                    className: \"InformationPartitionElement\",\n                  },\n                },\n                alias: \"partition\",\n                isRequired: true,\n              },\n            ],\n            instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, \"$.Subject.Model.TargetPartition\") = printf(\"0x%x\", partition.ECInstanceId)) AND NOT this.IsPrivate AND json_extract(partition.JsonProperties, \"$.PhysicalPartition.Model.Content\") = NULL AND json_extract(partition.JsonProperties, \"$.GraphicalPartition3d.Model.Content\") = NULL AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isModel: \"true\",\n              icon: \"\\\"icon-model\\\"\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"Subject\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"InstanceNodesOfSpecificClasses\",\n            classes: {\n              schemaName: \"BisCore\",\n              classNames: [\n                \"GeometricModel3d\",\n              ],\n            },\n            arePolymorphic: true,\n            relatedInstances: [\n              {\n                relationshipPath: {\n                  relationship: {\n                    schemaName: \"BisCore\",\n                    className: \"ModelModelsElement\",\n                  },\n                  direction: \"Forward\",\n                  targetClass: {\n                    schemaName: \"BisCore\",\n                    className: \"InformationPartitionElement\",\n                  },\n                },\n                alias: \"partition\",\n                isRequired: true,\n              },\n            ],\n            instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, \"$.Subject.Model.TargetPartition\") = printf(\"0x%x\", partition.ECInstanceId)) AND NOT this.IsPrivate AND (json_extract(partition.JsonProperties, \"$.PhysicalPartition.Model.Content\") <> NULL OR json_extract(partition.JsonProperties, \"$.GraphicalPartition3d.Model.Content\") <> NULL) AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            hideNodesInHierarchy: true,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"NodeArtifacts\",\n            items: {\n              isContentModel: \"true\",\n            },\n          },\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isModel: \"true\",\n              icon: \"\\\"icon-model\\\"\",\n            },\n          },\n        ],\n      },\n      {\n        ruleType: \"ChildNodes\",\n        condition: `ParentNode.IsOfClass(\"GeometricModel3d\", \"BisCore\")`,\n        specifications: [\n          {\n            specType: \"RelatedInstanceNodes\",\n            relationshipPaths: [\n              {\n                relationship: {\n                  schemaName: \"BisCore\",\n                  className: \"ModelOwnsSubModel\",\n                },\n                direction: \"Forward\",\n                targetClass: {\n                  schemaName: \"BisCore\",\n                  className: \"GeometricModel3d\",\n                },\n              },\n            ],\n            instanceFilter: `NOT this.IsPrivate AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"${elementClassSpecification.schemaName}:${elementClassSpecification.className}\")`,\n            groupByClass: false,\n            groupByLabel: false,\n          },\n        ],\n        customizationRules: [\n          {\n            ruleType: \"ExtendedData\",\n            items: {\n              isModel: \"true\",\n              icon: \"\\\"icon-model\\\"\",\n            },\n          },\n        ],\n      },\n    ],\n  };\n}\n\n/** @internal */\nexport function customizeModelsTreeNodeItem(item: Partial<DelayLoadedTreeNodeItem>, node: Partial<Node>) {\n  item.isCheckboxVisible = true;\n  item.isCheckboxDisabled = true;\n  item.icon = getIcon(node);\n}\n\nfunction getIcon(node: Partial<Node>) {\n  return node.key && NodeKey.isClassGroupingNodeKey(node.key)\n    ? node.extendedData?.groupIcon\n    : node.extendedData?.icon;\n}\n"]}

package/lib/cjs/components/utils/UseTreeTransientState.d.ts

@@ -0,0 +1,19 @@
+/// <reference types="react" />
+/**
+ * Allows to persist tree scroll position when tree is used in `@itwin/appui-react` Widgets.
+ * Returned `ref` should be passed to container element containing tree component.
+ *
+ * Example:
+ * ```typescript
+ * function ModelsTreeWidget() {
+ *   const ref = useTreeTransientState<HTMLDivElement>();
+ *   return <div ref={ref}>
+ *     <ModelsTree />
+ *   </div>
+ * }
+ * ```
+ *
+ * @public
+ */
+export declare function useTreeTransientState<T extends Element>(): import("react").Ref<T>;
+//# sourceMappingURL=UseTreeTransientState.d.ts.map

package/lib/cjs/components/utils/UseTreeTransientState.js

@@ -0,0 +1,33 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useTreeTransientState = void 0;
+const appui_react_1 = require("@itwin/appui-react");
+const components_react_1 = require("@itwin/components-react");
+// istanbul ignore file
+/**
+ * Allows to persist tree scroll position when tree is used in `@itwin/appui-react` Widgets.
+ * Returned `ref` should be passed to container element containing tree component.
+ *
+ * Example:
+ * ```typescript
+ * function ModelsTreeWidget() {
+ *   const ref = useTreeTransientState<HTMLDivElement>();
+ *   return <div ref={ref}>
+ *     <ModelsTree />
+ *   </div>
+ * }
+ * ```
+ *
+ * @public
+ */
+function useTreeTransientState() {
+    const { ref, persist, restore } = (0, components_react_1.useControlledTreeLayoutStorage)();
+    (0, appui_react_1.useTransientState)(persist, restore);
+    return ref;
+}
+exports.useTreeTransientState = useTreeTransientState;
+//# sourceMappingURL=UseTreeTransientState.js.map

package/lib/cjs/components/utils/UseTreeTransientState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UseTreeTransientState.js","sourceRoot":"","sources":["../../../../src/components/utils/UseTreeTransientState.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;;AAE/F,oDAAuD;AACvD,8DAAyE;AAEzE,uBAAuB;AAEvB;;;;;;;;;;;;;;;GAeG;AACH,SAAgB,qBAAqB;IACnC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,IAAA,iDAA8B,GAAK,CAAC;IACtE,IAAA,+BAAiB,EAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACpC,OAAO,GAAG,CAAC;AACb,CAAC;AAJD,sDAIC","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 { useTransientState } from \"@itwin/appui-react\";\nimport { useControlledTreeLayoutStorage } from \"@itwin/components-react\";\n\n// istanbul ignore file\n\n/**\n * Allows to persist tree scroll position when tree is used in `@itwin/appui-react` Widgets.\n * Returned `ref` should be passed to container element containing tree component.\n *\n * Example:\n * ```typescript\n * function ModelsTreeWidget() {\n *   const ref = useTreeTransientState<HTMLDivElement>();\n *   return <div ref={ref}>\n *     <ModelsTree />\n *   </div>\n * }\n * ```\n *\n * @public\n */\nexport function useTreeTransientState<T extends Element>() {\n  const { ref, persist, restore } = useControlledTreeLayoutStorage<T>();\n  useTransientState(persist, restore);\n  return ref;\n}\n"]}

package/lib/cjs/tree-widget-react.d.ts

@@ -1,5 +1,6 @@
 export * from "./TreeWidget";
-export * from "./components/TreeWidgetComponent";
+export * from "./components/SelectableTree";
 export * from "./components/trees";
 export * from "./components/TreeWidgetUiItemsProvider";
+export * from "./components/utils/UseTreeTransientState";
 //# sourceMappingURL=tree-widget-react.d.ts.map

package/lib/cjs/tree-widget-react.js

@@ -19,7 +19,8 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./TreeWidget"), exports);
-__exportStar(require("./components/TreeWidgetComponent"), exports);
+__exportStar(require("./components/SelectableTree"), exports);
 __exportStar(require("./components/trees"), exports);
 __exportStar(require("./components/TreeWidgetUiItemsProvider"), exports);
+__exportStar(require("./components/utils/UseTreeTransientState"), exports);
 //# sourceMappingURL=tree-widget-react.js.map

package/lib/cjs/tree-widget-react.js.map

@@ -1 +1 @@
-{"version":3,"file":"tree-widget-react.js","sourceRoot":"","sources":["../../src/tree-widget-react.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;;;;;;;;;;;;;;;AAE/F,+CAA6B;AAC7B,mEAAiD;AACjD,qDAAmC;AACnC,yEAAuD","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\nexport * from \"./TreeWidget\";\nexport * from \"./components/TreeWidgetComponent\";\nexport * from \"./components/trees\";\nexport * from \"./components/TreeWidgetUiItemsProvider\";\n"]}
+{"version":3,"file":"tree-widget-react.js","sourceRoot":"","sources":["../../src/tree-widget-react.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;;;;;;;;;;;;;;;AAE/F,+CAA6B;AAC7B,8DAA4C;AAC5C,qDAAmC;AACnC,yEAAuD;AACvD,2EAAyD","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\nexport * from \"./TreeWidget\";\nexport * from \"./components/SelectableTree\";\nexport * from \"./components/trees\";\nexport * from \"./components/TreeWidgetUiItemsProvider\";\nexport * from \"./components/utils/UseTreeTransientState\";\n"]}

package/lib/esm/components/{TreeWidgetComponent.d.ts → SelectableTree.d.ts}

@@ -1,8 +1,8 @@
 /// <reference types="react" />
-import "./TreeWidgetComponent.scss";
+import "./SelectableTree.scss";
 import type { IModelConnection } from "@itwin/core-frontend";
 /**
- * Definition of a tree component displayed in [[TreeWidgetComponent]]
+ * Definition of a tree component displayed in [[SelectableTree]]
  * @public
  */
 export interface TreeDefinition {
@@ -19,15 +19,15 @@ export interface TreeDefinition {
     shouldShow?: (imodel: IModelConnection) => Promise<boolean>;
 }
 /**
- * Props for [[TreeWidgetComponent]]
+ * Props for [[SelectableTree]]
  * @public
  */
-export interface TreeWidgetComponentProps {
+export interface SelectableTreeProps {
     trees: TreeDefinition[];
 }
 /**
  * A component that renders a tree (combo box) selector and the selected tree component.
  * @public
  */
-export declare function TreeWidgetComponent(props: TreeWidgetComponentProps): JSX.Element | null;
-//# sourceMappingURL=TreeWidgetComponent.d.ts.map
+export declare function SelectableTree(props: SelectableTreeProps): JSX.Element | null;
+//# sourceMappingURL=SelectableTree.d.ts.map

package/lib/esm/components/{TreeWidgetComponent.js → SelectableTree.js}

@@ -3,7 +3,7 @@ import { jsx as _jsx, Fragment as _Fragment } from "react/jsx-runtime";
 * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
 * See LICENSE.md in the project root for license terms and full copyright notice.
 *--------------------------------------------------------------------------------------------*/
-import "./TreeWidgetComponent.scss";
+import "./SelectableTree.scss";
 import { useEffect, useState } from "react";
 import { useActiveIModelConnection } from "@itwin/appui-react";
 import { SelectableContent } from "@itwin/components-react";
@@ -14,16 +14,16 @@ import { TreeWidget } from "../TreeWidget";
  * A component that renders a tree (combo box) selector and the selected tree component.
  * @public
  */
-export function TreeWidgetComponent(props) {
+export function SelectableTree(props) {
     const imodel = useActiveIModelConnection();
     if (!imodel)
         return null;
-    return _jsx(TreeWidgetComponentContent, { ...props, imodel: imodel });
+    return _jsx(SelectableTreeContent, { ...props, imodel: imodel });
 }
-function TreeWidgetComponentContent(props) {
+function SelectableTreeContent(props) {
     const { trees: treeDefinitions, imodel } = props;
     const trees = useActiveTrees(treeDefinitions, imodel);
-    return (_jsx("div", { className: "tree-widget-visibility-widget", children: _jsx(SelectableContent, { ...getSelectableContentProps(trees) }) }));
+    return (_jsx("div", { className: "tree-widget-selectable-tree", children: _jsx(SelectableContent, { ...getSelectableContentProps(trees) }) }));
 }
 function useActiveTrees(treeDefinitions, imodel) {
     const [trees, setTrees] = useState();
@@ -90,4 +90,4 @@ function Delayed({ delay = 200, children }) {
     }
     return _jsx(_Fragment, { children: children });
 }
-//# sourceMappingURL=TreeWidgetComponent.js.map
+//# sourceMappingURL=SelectableTree.js.map

package/lib/esm/components/SelectableTree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SelectableTree.js","sourceRoot":"","sources":["../../../src/components/SelectableTree.tsx"],"names":[],"mappings":";AAAA;;;+FAG+F;AAE/F,OAAO,uBAAuB,CAAC;AAC/B,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAgC3C;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,KAA0B;IACvD,MAAM,MAAM,GAAG,yBAAyB,EAAE,CAAC;IAE3C,IAAI,CAAC,MAAM;QACT,OAAO,IAAI,CAAC;IAEd,OAAO,KAAC,qBAAqB,OAAK,KAAK,EAAE,MAAM,EAAE,MAAM,GAAI,CAAC;AAC9D,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAyD;IACtF,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IACjD,MAAM,KAAK,GAAG,cAAc,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;IAEtD,OAAO,CACL,cAAK,SAAS,EAAC,6BAA6B,YAC1C,KAAC,iBAAiB,OAAK,yBAAyB,CAAC,KAAK,CAAC,GAAI,GACvD,CACP,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,eAAiC,EAAE,MAAwB;IACjF,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,EAAiC,CAAC;IAEpE,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,CAAC,KAAK,IAAI,EAAE;YACV,MAAM,YAAY,GAAG,MAAM,cAAc,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YACnE,uBAAuB;YACvB,IAAI,CAAC,QAAQ,EAAE;gBACb,QAAQ,CAAC,YAAY,CAAC,CAAC;aACxB;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,OAAO,GAAG,EAAE,GAAG,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IACpC,CAAC,EAAE,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC,CAAC;IAE9B,OAAO,KAAK,CAAC;AACf,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,eAAiC,EAAE,MAAwB;IACvF,MAAM,gBAAgB,GAAG,KAAK,EAAE,OAAuB,EAAE,EAAE;QACzD,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE;YAC3E,OAAO,SAAS,CAAC;SAClB;QACD,OAAO;YACL,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,KAAK,EAAE,OAAO,CAAC,QAAQ,EAAE;YACzB,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,SAAS,CAAkC,CAAC;AAC1I,CAAC;AAED,SAAS,yBAAyB,CAAC,KAAqC;IACtE,IAAI,KAAK,KAAK,SAAS,EAAE;QACvB,OAAO;YACL,wBAAwB,EAAE,SAAS;YACnC,QAAQ,EAAE,CAAC;oBACT,EAAE,EAAE,SAAS;oBACb,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,GAAG,EAAE,CAAC,CACZ,KAAC,OAAO,cACN,KAAC,cAAc,IAAC,aAAa,EAAE,IAAI,GAAI,GAC/B,CACX;iBACF,CAAC;SACH,CAAC;KACH;IAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE;QACtB,OAAO;YACL,wBAAwB,EAAE,UAAU;YACpC,QAAQ,EAAE,CAAC;oBACT,EAAE,EAAE,UAAU;oBACd,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,GAAG,EAAE,CAAC,CACZ,KAAC,YAAY,cACV,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,GACnB,CAChB;iBACF,CAAC;SACH,CAAC;KACH;IAED,OAAO;QACL,wBAAwB,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE;QACrC,QAAQ,EAAE,KAAK;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,OAAO,CAAC,EAAE,KAAK,GAAG,GAAG,EAAE,QAAQ,EAAyC;IAC/E,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAExC,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACvD,OAAO,GAAG,EAAE,GAAG,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;IAEZ,IAAI,CAAC,IAAI,EAAE;QACT,OAAO,IAAI,CAAC;KACb;IAED,OAAO,4BAAG,QAAQ,GAAI,CAAC;AACzB,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 \"./SelectableTree.scss\";\nimport { useEffect, useState } from \"react\";\nimport { useActiveIModelConnection } from \"@itwin/appui-react\";\nimport { SelectableContent } from \"@itwin/components-react\";\nimport { FillCentered } from \"@itwin/core-react\";\nimport { ProgressLinear } from \"@itwin/itwinui-react\";\nimport { TreeWidget } from \"../TreeWidget\";\n\nimport type { PropsWithChildren } from \"react\";\nimport type { SelectableContentDefinition, SelectableContentProps } from \"@itwin/components-react\";\nimport type { IModelConnection } from \"@itwin/core-frontend\";\n\n/**\n * Definition of a tree component displayed in [[SelectableTree]]\n * @public\n */\nexport interface TreeDefinition {\n  /** Id of the tree */\n  id: string;\n  /** Callback that is used to get tree label */\n  getLabel: () => string;\n  /** Callback that is used to render tree component */\n  render: () => React.ReactNode;\n  /**\n   * Callback that is used to determine if tree should be shown for current active iModel connection.\n   * If callback is `undefined` tree is shown for all iModel connections.\n   */\n  shouldShow?: (imodel: IModelConnection) => Promise<boolean>;\n}\n\n/**\n * Props for [[SelectableTree]]\n * @public\n */\nexport interface SelectableTreeProps {\n  trees: TreeDefinition[];\n}\n\n/**\n * A component that renders a tree (combo box) selector and the selected tree component.\n * @public\n */\nexport function SelectableTree(props: SelectableTreeProps) {\n  const imodel = useActiveIModelConnection();\n\n  if (!imodel)\n    return null;\n\n  return <SelectableTreeContent {...props} imodel={imodel} />;\n}\n\nfunction SelectableTreeContent(props: SelectableTreeProps & { imodel: IModelConnection }) {\n  const { trees: treeDefinitions, imodel } = props;\n  const trees = useActiveTrees(treeDefinitions, imodel);\n\n  return (\n    <div className=\"tree-widget-selectable-tree\">\n      <SelectableContent {...getSelectableContentProps(trees)} />\n    </div>\n  );\n}\n\nfunction useActiveTrees(treeDefinitions: TreeDefinition[], imodel: IModelConnection) {\n  const [trees, setTrees] = useState<SelectableContentDefinition[]>();\n\n  useEffect(() => {\n    let disposed = false;\n    (async () => {\n      const visibleTrees = await getActiveTrees(treeDefinitions, imodel);\n      // istanbul ignore else\n      if (!disposed) {\n        setTrees(visibleTrees);\n      }\n    })();\n\n    return () => { disposed = true; };\n  }, [treeDefinitions, imodel]);\n\n  return trees;\n}\n\nasync function getActiveTrees(treeDefinitions: TreeDefinition[], imodel: IModelConnection): Promise<SelectableContentDefinition[]> {\n  const handleDefinition = async (treeDef: TreeDefinition) => {\n    if (treeDef.shouldShow !== undefined && !(await treeDef.shouldShow(imodel))) {\n      return undefined;\n    }\n    return {\n      id: treeDef.id,\n      label: treeDef.getLabel(),\n      render: treeDef.render,\n    };\n  };\n\n  return (await Promise.all(treeDefinitions.map(handleDefinition))).filter((tree) => tree !== undefined) as SelectableContentDefinition[];\n}\n\nfunction getSelectableContentProps(trees?: SelectableContentDefinition[]): SelectableContentProps {\n  if (trees === undefined) {\n    return {\n      defaultSelectedContentId: \"loading\",\n      children: [{\n        id: \"loading\",\n        label: \"\",\n        render: () => (\n          <Delayed>\n            <ProgressLinear indeterminate={true} />\n          </Delayed>\n        ),\n      }],\n    };\n  }\n\n  if (trees.length === 0) {\n    return {\n      defaultSelectedContentId: \"no-trees\",\n      children: [{\n        id: \"no-trees\",\n        label: \"\",\n        render: () => (\n          <FillCentered>\n            {TreeWidget.translate(\"noTrees\")}\n          </FillCentered>\n        ),\n      }],\n    };\n  }\n\n  return {\n    defaultSelectedContentId: trees[0].id,\n    children: trees,\n  };\n}\n\nfunction Delayed({ delay = 200, children }: PropsWithChildren<{ delay?: number }>) {\n  const [show, setShow] = useState(false);\n\n  useEffect(() => {\n    const id = setTimeout(() => { setShow(true); }, delay);\n    return () => { clearTimeout(id); };\n  }, [delay]);\n\n  if (!show) {\n    return null;\n  }\n\n  return <>{children}</>;\n}\n"]}