@itwin/tree-widget-react 3.14.2 → 3.15.0

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

@@ -11,11 +11,28 @@ export function TreeNodeCheckbox({ node, onCheckboxClicked, getCheckboxState, ..
         return null;
     }
     const checkboxState = getCheckboxState(node);
-    return (_jsx(TooltipWrapper, { content: checkboxState.tooltip, children: _jsx(Checkbox, { ...props, checked: checkboxState.state === "on", onClick: (e) => {
-                e.stopPropagation();
-            }, onChange: (e) => {
-                onCheckboxClicked(node, e.currentTarget.checked);
-            }, indeterminate: checkboxState.state === "partial", disabled: checkboxState.isDisabled, "aria-label": checkboxState.tooltip }) }));
+    const checkboxProps = {
+        ...props,
+        "aria-label": checkboxState.tooltip,
+        onClick: (e) => {
+            e.stopPropagation();
+        },
+        onChange: (e) => {
+            onCheckboxClicked(node, e.currentTarget.checked);
+        },
+    };
+    if ("isLoading" in checkboxState) {
+        checkboxProps.style = {
+            ...checkboxProps.style,
+            visibility: "hidden",
+        };
+    }
+    else {
+        checkboxProps.checked = checkboxState.state === "on";
+        checkboxProps.indeterminate = checkboxState.state === "partial";
+        checkboxProps.disabled = checkboxState.isDisabled;
+    }
+    return (_jsx(TooltipWrapper, { content: checkboxState.tooltip, children: _jsx(Checkbox, { ...checkboxProps }) }));
 }
 function TooltipWrapper({ content, children }) {
     return !!content ? (_jsx(Tooltip, { content: content, placement: "left", children: children })) : (_jsx(_Fragment, { children: children }));

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

@@ -1 +1 @@
-{"version":3,"file":"TreeNodeCheckbox.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/TreeNodeCheckbox.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,2BAA2B,EAAE,MAAM,uCAAuC,CAAC;AA2BpF,gBAAgB;AAChB,MAAM,UAAU,gBAAgB,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,GAAG,KAAK,EAAsD;IAC1I,IAAI,MAAM,IAAI,IAAI,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,EAAE,CAAC;QACzD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,aAAa,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAC7C,OAAO,CACL,KAAC,cAAc,IAAC,OAAO,EAAE,aAAa,CAAC,OAAO,YAC5C,KAAC,QAAQ,OACH,KAAK,EACT,OAAO,EAAE,aAAa,CAAC,KAAK,KAAK,IAAI,EACrC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;gBACb,CAAC,CAAC,eAAe,EAAE,CAAC;YACtB,CAAC,EACD,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE;gBACd,iBAAiB,CAAC,IAAI,EAAE,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YACnD,CAAC,EACD,aAAa,EAAE,aAAa,CAAC,KAAK,KAAK,SAAS,EAChD,QAAQ,EAAE,aAAa,CAAC,UAAU,gBACtB,aAAa,CAAC,OAAO,GACjC,GACa,CAClB,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAoD;IAC7F,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CACjB,KAAC,OAAO,IAAC,OAAO,EAAE,OAAO,EAAE,SAAS,EAAC,MAAM,YACxC,QAAQ,GACD,CACX,CAAC,CAAC,CAAC,CACF,4BAAG,QAAQ,GAAI,CAChB,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 { Checkbox, Tooltip } from \"@itwin/itwinui-react\";\nimport { isPresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\n\nimport type { PresentationHierarchyNode, RenderedTreeNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { ComponentPropsWithoutRef } from \"react\";\n\n/**\n * Data structure that describes tree node checkbox state.\n * @beta\n */\ninterface TreeNodeCheckboxState {\n  state: \"on\" | \"off\" | \"partial\";\n  isDisabled?: boolean;\n  tooltip?: string;\n}\n\n/** @beta */\nexport interface TreeCheckboxProps {\n  /** Callback that should be invoked when checkbox is clicked. */\n  onCheckboxClicked: (node: PresentationHierarchyNode, checked: boolean) => void;\n  /** Callback that should be used to determine current checkbox state. */\n  getCheckboxState: (node: PresentationHierarchyNode) => TreeNodeCheckboxState;\n}\n\n/** @internal */\ntype TreeNodeCheckboxProps = TreeCheckboxProps &\n  Omit<ComponentPropsWithoutRef<typeof Checkbox>, \"onClick\" | \"checked\" | \"onChange\" | \"indeterminate\" | \"disabled\" | \"title\">;\n\n/** @internal */\nexport function TreeNodeCheckbox({ node, onCheckboxClicked, getCheckboxState, ...props }: TreeNodeCheckboxProps & { node: RenderedTreeNode }) {\n  if (\"type\" in node || !isPresentationHierarchyNode(node)) {\n    return null;\n  }\n\n  const checkboxState = getCheckboxState(node);\n  return (\n    <TooltipWrapper content={checkboxState.tooltip}>\n      <Checkbox\n        {...props}\n        checked={checkboxState.state === \"on\"}\n        onClick={(e) => {\n          e.stopPropagation();\n        }}\n        onChange={(e) => {\n          onCheckboxClicked(node, e.currentTarget.checked);\n        }}\n        indeterminate={checkboxState.state === \"partial\"}\n        disabled={checkboxState.isDisabled}\n        aria-label={checkboxState.tooltip}\n      />\n    </TooltipWrapper>\n  );\n}\n\nfunction TooltipWrapper({ content, children }: { content?: string; children?: React.ReactNode }) {\n  return !!content ? (\n    <Tooltip content={content} placement=\"left\">\n      {children}\n    </Tooltip>\n  ) : (\n    <>{children}</>\n  );\n}\n"]}
+{"version":3,"file":"TreeNodeCheckbox.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/TreeNodeCheckbox.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,2BAA2B,EAAE,MAAM,uCAAuC,CAAC;AAiCpF,gBAAgB;AAChB,MAAM,UAAU,gBAAgB,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,GAAG,KAAK,EAAsD;IAC1I,IAAI,MAAM,IAAI,IAAI,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,EAAE,CAAC;QACzD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,aAAa,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAC7C,MAAM,aAAa,GAA8C;QAC/D,GAAG,KAAK;QACR,YAAY,EAAE,aAAa,CAAC,OAAO;QACnC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;YACb,CAAC,CAAC,eAAe,EAAE,CAAC;QACtB,CAAC;QACD,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE;YACd,iBAAiB,CAAC,IAAI,EAAE,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACnD,CAAC;KACF,CAAC;IACF,IAAI,WAAW,IAAI,aAAa,EAAE,CAAC;QACjC,aAAa,CAAC,KAAK,GAAG;YACpB,GAAG,aAAa,CAAC,KAAK;YACtB,UAAU,EAAE,QAAQ;SACrB,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,aAAa,CAAC,OAAO,GAAG,aAAa,CAAC,KAAK,KAAK,IAAI,CAAC;QACrD,aAAa,CAAC,aAAa,GAAG,aAAa,CAAC,KAAK,KAAK,SAAS,CAAC;QAChE,aAAa,CAAC,QAAQ,GAAG,aAAa,CAAC,UAAU,CAAC;IACpD,CAAC;IACD,OAAO,CACL,KAAC,cAAc,IAAC,OAAO,EAAE,aAAa,CAAC,OAAO,YAC5C,KAAC,QAAQ,OAAK,aAAa,GAAI,GAChB,CAClB,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAoD;IAC7F,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CACjB,KAAC,OAAO,IAAC,OAAO,EAAE,OAAO,EAAE,SAAS,EAAC,MAAM,YACxC,QAAQ,GACD,CACX,CAAC,CAAC,CAAC,CACF,4BAAG,QAAQ,GAAI,CAChB,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 { Checkbox, Tooltip } from \"@itwin/itwinui-react\";\nimport { isPresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\n\nimport type { PresentationHierarchyNode, RenderedTreeNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { ComponentPropsWithoutRef } from \"react\";\n\n/**\n * Data structure that describes tree node checkbox state.\n * @beta\n */\ntype TreeNodeCheckboxState = (\n  | {\n      state: \"on\" | \"off\" | \"partial\";\n      isDisabled?: boolean;\n    }\n  | {\n      isLoading: true;\n    }\n) & {\n  tooltip?: string;\n};\n\n/** @beta */\nexport interface TreeCheckboxProps {\n  /** Callback that should be invoked when checkbox is clicked. */\n  onCheckboxClicked: (node: PresentationHierarchyNode, checked: boolean) => void;\n  /** Callback that should be used to determine current checkbox state. */\n  getCheckboxState: (node: PresentationHierarchyNode) => TreeNodeCheckboxState;\n}\n\n/** @internal */\ntype TreeNodeCheckboxProps = TreeCheckboxProps &\n  Omit<ComponentPropsWithoutRef<typeof Checkbox>, \"onClick\" | \"checked\" | \"onChange\" | \"indeterminate\" | \"disabled\" | \"title\">;\n\n/** @internal */\nexport function TreeNodeCheckbox({ node, onCheckboxClicked, getCheckboxState, ...props }: TreeNodeCheckboxProps & { node: RenderedTreeNode }) {\n  if (\"type\" in node || !isPresentationHierarchyNode(node)) {\n    return null;\n  }\n\n  const checkboxState = getCheckboxState(node);\n  const checkboxProps: ComponentPropsWithoutRef<typeof Checkbox> = {\n    ...props,\n    \"aria-label\": checkboxState.tooltip,\n    onClick: (e) => {\n      e.stopPropagation();\n    },\n    onChange: (e) => {\n      onCheckboxClicked(node, e.currentTarget.checked);\n    },\n  };\n  if (\"isLoading\" in checkboxState) {\n    checkboxProps.style = {\n      ...checkboxProps.style,\n      visibility: \"hidden\",\n    };\n  } else {\n    checkboxProps.checked = checkboxState.state === \"on\";\n    checkboxProps.indeterminate = checkboxState.state === \"partial\";\n    checkboxProps.disabled = checkboxState.isDisabled;\n  }\n  return (\n    <TooltipWrapper content={checkboxState.tooltip}>\n      <Checkbox {...checkboxProps} />\n    </TooltipWrapper>\n  );\n}\n\nfunction TooltipWrapper({ content, children }: { content?: string; children?: React.ReactNode }) {\n  return !!content ? (\n    <Tooltip content={content} placement=\"left\">\n      {children}\n    </Tooltip>\n  ) : (\n    <>{children}</>\n  );\n}\n"]}

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

@@ -336,33 +336,38 @@ export class ModelsTreeIdsCache {
             const reader = this._queryExecutor.createQueryReader({
                 ctes: [
                     `
-                    CategoryElements(id, modelId, categoryId) AS (
-                      SELECT ECInstanceId, Model.Id, Category.Id
-                      FROM ${this._hierarchyConfig.elementClassSpecification}
-                      WHERE
-                        Parent.Id IS NULL
-                        AND (
-                          ${whereClauses.join(" OR ")}
-                        )
+                  CategoryElements(id, modelId, categoryId) AS (
+                    SELECT ECInstanceId, Model.Id, Category.Id
+                    FROM ${this._hierarchyConfig.elementClassSpecification}
+                    WHERE
+                      Parent.Id IS NULL
+                      AND (
+                        ${whereClauses.join(" OR ")}
+                      )
 
-                      UNION ALL
+                    UNION ALL
 
-                      SELECT c.ECInstanceId, p.modelId, p.categoryId
-                      FROM ${this._hierarchyConfig.elementClassSpecification} c
-                      JOIN CategoryElements p ON c.Parent.Id = p.id
-                    )
-                  `,
+                    SELECT c.ECInstanceId, p.modelId, p.categoryId
+                    FROM ${this._hierarchyConfig.elementClassSpecification} c
+                    JOIN CategoryElements p ON c.Parent.Id = p.id
+                  )
+                `,
                 ],
                 ecsql: `
-                  SELECT modelId, categoryId, COUNT(id) elementsCount
-                  FROM CategoryElements
-                  GROUP BY modelId, categoryId
-                `,
+                SELECT modelId, categoryId, COUNT(id) elementsCount
+                FROM CategoryElements
+                GROUP BY modelId, categoryId
+              `,
             }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" });
             const result = new Array();
             for await (const row of reader) {
                 result.push({ modelId: row.modelId, categoryId: row.categoryId, elementsCount: row.elementsCount });
             }
+            input.forEach(({ modelId, categoryId }) => {
+                if (!result.some((queriedResult) => queriedResult.categoryId === categoryId && queriedResult.modelId === modelId)) {
+                    result.push({ categoryId, modelId, elementsCount: 0 });
+                }
+            });
             return result;
         }), mergeAll()));
     }

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

@@ -1 +1 @@
-{"version":3,"file":"ModelsTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/ModelsTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACtI,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAsBlD,gBAAgB;AAChB,MAAM,OAAO,kBAAkB;IAWnB;IACA;IAXO,sBAAsB,CAAkC;IACjE,aAAa,CAAoD;IACjE,iBAAiB,CAAiC,CAAC,sFAAsF;IACzI,WAAW,CAAkD;IAC7D,iCAAiC,CAA4C;IAC7E,cAAc,CAA2D;IACzE,gBAAgB,CAAyD;IACzE,iBAAiB,CAA2D;IAEpF,YACU,cAA0C,EAC1C,gBAAkD;QADlD,mBAAc,GAAd,cAAc,CAA4B;QAC1C,qBAAgB,GAAhB,gBAAgB,CAAkC;QAE1D,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3H,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC,gBAAgB,GAAG,IAAI,GAAG,EAAE,CAAC;QAClC,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;IACrC,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,aAAa;QAC1B,MAAM,aAAa,GAAG;;;;;;;;;wCASc,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;;;;;;;KAUlF,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACjJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,iBAAiB,EAAE,GAAG,CAAC,iBAAiB,EAAE,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;QACjI,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,WAAW;QACxB,MAAM,WAAW,GAAG;;;;;;UAMd,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,6BAA6B,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,mCAAmC;KACjK,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YAC/I,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC/C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,aAAa,KAAK,CAAC,KAAK,IAAI,EAAE;YACjC,MAAM,CAAC,YAAY,EAAE,uBAAuB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChE,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;oBAClD,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;wBACjD,MAAM,WAAW,GAAgB;4BAC/B,aAAa,EAAE,OAAO,CAAC,QAAQ;4BAC/B,eAAe,EAAE,OAAO,CAAC,eAAe;4BACxC,aAAa,EAAE,IAAI,GAAG,EAAE;4BACxB,WAAW,EAAE,IAAI,GAAG,EAAE;yBACvB,CAAC;wBACF,IAAI,OAAO,CAAC,iBAAiB,EAAE,CAAC;4BAC9B,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;wBACzD,CAAC;wBACD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;oBACtC,CAAC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA+B,CAAC;oBACtD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;wBAC7C,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;oBAC9C,CAAC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YAEH,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;gBACrF,IAAI,eAAe,EAAE,CAAC;oBACpB,MAAM,iBAAiB,GAAG,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;oBAC5D,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC;oBAC5B,iBAAiB,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBACjD,CAAC;YACH,CAAC;YAED,KAAK,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,uBAAuB,EAAE,CAAC;gBAChE,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;oBAC/B,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChD,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;oBACtB,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC3C,CAAC,CAAC,CAAC;YACL,CAAC;YAED,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,kHAAkH;IAC3G,KAAK,CAAC,mBAAmB;QAC9B,IAAI,CAAC,iBAAiB,KAAK,CAAC,KAAK,IAAI,EAAE;YACrC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAClD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;YAC/C,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE;gBAC9C,IAAI,WAAW,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;oBACrC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChC,IAAI,YAAY,GAAG,WAAW,CAAC,aAAa,CAAC;oBAC7C,OAAO,YAAY,EAAE,CAAC;wBACpB,gBAAgB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;wBACnC,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,aAAa,CAAC;oBAC/D,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,gBAAgB,CAAC,CAAC;QAC/B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,kBAAkB,CAAC,gBAA8B;QAC5D,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;QAChD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE,CAAC;oBACtC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACrC,OAAO,OAAO,CAAC;gBACjB,CAAC;gBACD,OAAO,UAAU,CAAC;YACpB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,+GAA+G;IACxG,KAAK,CAAC,kBAAkB,CAAC,UAAqB;QACnD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,MAAM,YAAY,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,EAAE,CAAC;YACrC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,MAAM;YACR,CAAC;YACD,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,SAAS;YACX,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;YACxC,YAAY,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,aAAa,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,6GAA6G;IACtG,KAAK,CAAC,uBAAuB,CAAC,gBAA8B;QACjE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAElD,MAAM,gBAAgB,GAAG,IAAI,KAAK,EAAc,CAAC;QACjD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,gBAAgB,CAAC,eAAe,EAAE,CAAC;oBACrC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACtC,OAAO,UAAU,CAAC;gBACpB,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,KAAK,EAAc,CAAC;QACzC,CAAC,GAAG,gBAAgB,EAAE,GAAG,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YAC/D,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,WAAW,IAAI,QAAQ,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEM,KAAK,CAAC,6BAA6B,CAAC,eAA2B;QACpE,IAAI,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QACvD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;gBAClD,MAAM,MAAM,GAAG,IAAI,KAAK,EAAe,CAAC;gBACxC,IAAI,YAAY,GAA2B,eAAe,CAAC;gBAC3D,OAAO,YAAY,EAAE,CAAC;oBACpB,IAAI,IAAI,CAAC,gBAAgB,CAAC,eAAe,IAAI,YAAY,KAAK,MAAM,CAAC,aAAa,EAAE,CAAC;wBACnF,MAAM;oBACR,CAAC;oBACD,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;oBAClD,IAAI,CAAC,UAAU,EAAE,eAAe,EAAE,CAAC;wBACjC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;oBAClE,CAAC;oBACD,YAAY,GAAG,UAAU,EAAE,aAAa,CAAC;gBAC3C,CAAC;gBACD,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QAMjC,MAAM,KAAK,GAAG;;;;;;;aAOL,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;KAEvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACzI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,cAAc,EAAE,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE,qBAAqB,EAAE,CAAC,CAAC,GAAG,CAAC,qBAAqB,EAAE,CAAC;QACvJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QACjC,MAAM,KAAK,GAAG;;;;;;aAML,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;sDAGN,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;KAChG,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACzI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACrG,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mCAAmC;QAC/C,IAAI,CAAC,iCAAiC,KAAK,CAAC,KAAK,IAAI,EAAE;YACrD,MAAM,gCAAgC,GAAG,IAAI,GAAG,EAAuB,CAAC;YACxE,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBAC1F,MAAM,GAAG,GAAG,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;gBACvC,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACxD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,gCAAgC,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;gBACzE,CAAC;qBAAM,CAAC;oBACN,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,OAAO,gCAAgC,CAAC;QAC1C,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iCAAiC,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,IAAI,CAAC,WAAW,KAAK,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,EAA4G,CAAC;YACvI,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,qBAAqB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBAC/G,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACtC,IAAI,KAAK,EAAE,CAAC;oBACV,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,qBAAqB,EAAE,CAAC,CAAC;oBAC5D,KAAK,CAAC,cAAc,GAAG,cAAc,CAAC;gBACxC,CAAC;qBAAM,CAAC;oBACN,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,EAAE,qBAAqB,EAAE,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,CAAC,CAAC;gBAC9G,CAAC;YACH,CAAC;YACD,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,OAAmB;QACjD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC;QAC9D,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,SAAqB;QAC5C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,MAAM,kBAAkB,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACxB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,CAAC,kBAAkB,CAAC,cAAc,CAAC;IAC5C,CAAC;IAEM,KAAK,CAAC,4BAA4B,CAAC,OAAmB,EAAE,WAAoB;QACjF,MAAM,gCAAgC,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;QAC1F,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YACpD,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;YAC/E,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,2BAA2B,CAAC,OAAmB;QAC1D,IAAI,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC7C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAG,IAAI,KAAK,EAAgC,CAAC;gBACzD,MAAM,YAAY,GAAG,CAAC,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC9D,KAAK,MAAM,CAAC,cAAc,EAAE,WAAW,CAAC,IAAI,YAAY,EAAE,CAAC;oBACzD,IAAI,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;wBACzC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,6BAA6B,CAAC,cAAc,CAAC,CAAC;wBAC7E,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,WAAW,EAAE,EAAE,SAAS,EAAE,0BAA0B,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;oBACxF,CAAC;gBACH,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,EAAE,CAAC;YAEL,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACtC,KAA6D;QAE7D,OAAO,OAAO,CACZ,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CACd,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE;YACtC,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC/B,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC1C,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACxB,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAuB,CAAC,EAClC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,EAC1D,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,cAAc,OAAO,wBAAwB,CAAC,GAAG,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;QAC5G,wGAAwG;QACxG,oDAAoD;QACpD,WAAW,CAAC,GAAG,CAAC,EAChB,QAAQ,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE;YAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAClD;gBACE,IAAI,EAAE;oBACJ;;;6BAGa,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;4BAIhD,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC;;;;;;6BAMxB,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;mBAGzD;iBACJ;gBACD,KAAK,EAAE;;;;iBAIJ;aACJ,EACD,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CACxD,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,KAAK,EAAkE,CAAC;YAC3F,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;gBAC/B,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,aAAa,EAAE,GAAG,CAAC,aAAa,EAAE,CAAC,CAAC;YACtG,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EACF,QAAQ,EAAE,CACX,CACF,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,KAAK,CAAC,8BAA8B,CAAC,UAAsB;QAChE,IAAI,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACnD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAG,IAAI,GAAG,EAAc,CAAC;gBACrC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;gBAC9C,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,EAAE,OAAO,EAAE,EAAE;oBACzC,MAAM,aAAa,GAAG,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC3D,IAAI,aAAa,EAAE,qBAAqB,EAAE,CAAC;wBACzC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBACtB,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,MAAM,aAAa,GAAG,IAAI,KAAK,EAAgC,CAAC;gBAChE,KAAK,MAAM,eAAe,IAAI,CAAC,GAAG,MAAM,CAAC,EAAE,CAAC;oBAC1C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,eAAe,CAAC,CAAC;oBAC3E,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;wBACnC,aAAa,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,EAAE,EAAE,SAAS,EAAE,yBAAyB,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;oBAC/F,CAAC;gBACH,CAAC;gBACD,OAAO,aAAa,CAAC;YACvB,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED,SAAS,mBAAmB,CAC1B,YAA0C,EAC1C,aAAuC,EACvC,EAAuF;IAEvF,MAAM,iBAAiB,GAAG,OAAO,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;IAC9G,iBAAiB;QACf,iBAAiB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YACzD,MAAM,gBAAgB,GAAG,YAAY,CAAC,GAAG,CAAC,cAAc,CAAE,CAAC;YAC3D,IAAI,EAAE,CAAC,cAAc,EAAE,gBAAgB,CAAC,KAAK,OAAO,EAAE,CAAC;gBACrD,OAAO;YACT,CAAC;YACD,mBAAmB,CAAC,YAAY,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,+BAA+B;IAMzB;IALF,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;IAC5C,eAAe,GAAG,IAAI,OAAO,EAAmD,CAAC;IACjF,aAAa,CAAe;IAEpC,YACU,OAE2E;QAF3E,YAAO,GAAP,OAAO,CAEoE;QAEnF,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,eAAe;aACtC,IAAI,CACH,UAAU,CAAC,EAAE,CAAC,EACd,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EACzC,QAAQ,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EACpD,QAAQ,EAAE,CACX;aACA,SAAS,CAAC;YACT,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,OAAO,GAAG,UAAU,EAAE,CAAC,CAAC;gBAC3D,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9B,CAAC;SACF,CAAC,CAAC;IACP,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;IACnC,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,MAAM,QAAQ,GAAG,GAAG,OAAO,GAAG,UAAU,EAAE,CAAC;QAC3C,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC;QAED,MAAM,GAAG,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAClC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { bufferCount, bufferTime, filter, firstValueFrom, from, map, mergeAll, mergeMap, reduce, ReplaySubject, Subject } from \"rxjs\";\nimport { assert, Id64 } from \"@itwin/core-bentley\";\nimport { IModel } from \"@itwin/core-common\";\nimport { collect } from \"../../common/Rxjs.js\";\nimport { pushToMap } from \"../../common/Utils.js\";\n\nimport type { Subscription } from \"rxjs\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\nimport type { ModelsTreeDefinition } from \"../ModelsTreeDefinition.js\";\nimport type { Id64Arg, Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\n\ninterface SubjectInfo {\n  parentSubject: Id64String | undefined;\n  hideInHierarchy: boolean;\n  childSubjects: Id64Set;\n  childModels: Id64Set;\n}\n\ninterface ModelInfo {\n  isModelPrivate: boolean;\n  categories: Map<Id64String, { isRootElementCategory: boolean }>;\n}\n\ntype ModelsTreeHierarchyConfiguration = ConstructorParameters<typeof ModelsTreeDefinition>[0][\"hierarchyConfig\"];\n\n/** @internal */\nexport class ModelsTreeIdsCache implements Disposable {\n  private readonly _categoryElementCounts: ModelCategoryElementsCountCache;\n  private _subjectInfos: Promise<Map<Id64String, SubjectInfo>> | undefined;\n  private _parentSubjectIds: Promise<Id64Array> | undefined; // the list should contain a subject id if its node should be shown as having children\n  private _modelInfos: Promise<Map<Id64String, ModelInfo>> | undefined;\n  private _modelWithCategoryModeledElements: Promise<Map<string, Id64Set>> | undefined;\n  private _modelKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath[]>>;\n  private _subjectKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath>>;\n  private _categoryKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath[]>>;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    private _hierarchyConfig: ModelsTreeHierarchyConfiguration,\n  ) {\n    this._categoryElementCounts = new ModelCategoryElementsCountCache(async (input) => this.queryCategoryElementCounts(input));\n    this._modelKeyPaths = new Map();\n    this._subjectKeyPaths = new Map();\n    this._categoryKeyPaths = new Map();\n  }\n\n  public [Symbol.dispose]() {\n    this._categoryElementCounts[Symbol.dispose]();\n  }\n\n  private async *querySubjects(): AsyncIterableIterator<{ id: Id64String; parentId?: Id64String; targetPartitionId?: Id64String; hideInHierarchy: boolean }> {\n    const subjectsQuery = `\n      SELECT\n        s.ECInstanceId id,\n        s.Parent.Id parentId,\n        (\n          SELECT m.ECInstanceId\n          FROM bis.GeometricModel3d m\n          WHERE m.ECInstanceId = HexToId(json_extract(s.JsonProperties, '$.Subject.Model.TargetPartition'))\n            AND NOT m.IsPrivate\n            AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)\n        ) targetPartitionId,\n        CASE\n          WHEN (\n            json_extract(s.JsonProperties, '$.Subject.Job.Bridge') IS NOT NULL\n            OR json_extract(s.JsonProperties, '$.Subject.Model.Type') = 'Hierarchy'\n          ) THEN 1\n          ELSE 0\n        END hideInHierarchy\n      FROM bis.Subject s\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: subjectsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId, targetPartitionId: row.targetPartitionId, hideInHierarchy: !!row.hideInHierarchy };\n    }\n  }\n\n  private async *queryModels(): AsyncIterableIterator<{ id: Id64String; parentId: Id64String }> {\n    const modelsQuery = `\n      SELECT p.ECInstanceId id, p.Parent.Id parentId\n      FROM bis.InformationPartitionElement p\n      INNER JOIN bis.GeometricModel3d m ON m.ModeledElement.Id = p.ECInstanceId\n      WHERE\n        NOT m.IsPrivate\n        ${this._hierarchyConfig.showEmptyModels ? \"\" : `AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)`}\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: modelsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId };\n    }\n  }\n\n  private async getSubjectInfos() {\n    this._subjectInfos ??= (async () => {\n      const [subjectInfos, targetPartitionSubjects] = await Promise.all([\n        (async () => {\n          const result = new Map<Id64String, SubjectInfo>();\n          for await (const subject of this.querySubjects()) {\n            const subjectInfo: SubjectInfo = {\n              parentSubject: subject.parentId,\n              hideInHierarchy: subject.hideInHierarchy,\n              childSubjects: new Set(),\n              childModels: new Set(),\n            };\n            if (subject.targetPartitionId) {\n              subjectInfo.childModels.add(subject.targetPartitionId);\n            }\n            result.set(subject.id, subjectInfo);\n          }\n          return result;\n        })(),\n        (async () => {\n          const result = new Map<Id64String, Set<Id64String>>();\n          for await (const model of this.queryModels()) {\n            pushToMap(result, model.id, model.parentId);\n          }\n          return result;\n        })(),\n      ]);\n\n      for (const [subjectId, { parentSubject: parentSubjectId }] of subjectInfos.entries()) {\n        if (parentSubjectId) {\n          const parentSubjectInfo = subjectInfos.get(parentSubjectId);\n          assert(!!parentSubjectInfo);\n          parentSubjectInfo.childSubjects.add(subjectId);\n        }\n      }\n\n      for (const [partitionId, subjectIds] of targetPartitionSubjects) {\n        subjectIds.forEach((subjectId) => {\n          const subjectInfo = subjectInfos.get(subjectId);\n          assert(!!subjectInfo);\n          subjectInfo.childModels.add(partitionId);\n        });\n      }\n\n      return subjectInfos;\n    })();\n    return this._subjectInfos;\n  }\n\n  /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */\n  public async getParentSubjectIds(): Promise<Id64String[]> {\n    this._parentSubjectIds ??= (async () => {\n      const subjectInfos = await this.getSubjectInfos();\n      const parentSubjectIds = new Set<Id64String>();\n      subjectInfos.forEach((subjectInfo, subjectId) => {\n        if (subjectInfo.childModels.size > 0) {\n          parentSubjectIds.add(subjectId);\n          let currParentId = subjectInfo.parentSubject;\n          while (currParentId) {\n            parentSubjectIds.add(currParentId);\n            currParentId = subjectInfos.get(currParentId)?.parentSubject;\n          }\n        }\n      });\n      return [...parentSubjectIds];\n    })();\n    return this._parentSubjectIds;\n  }\n\n  /**\n   * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into\n   * account `hideInHierarchy` flag.\n   */\n  public async getChildSubjectIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const childSubjectIds = new Array<Id64String>();\n    const subjectInfos = await this.getSubjectInfos();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (!childSubjectInfo.hideInHierarchy) {\n          childSubjectIds.push(childSubjectId);\n          return \"break\";\n        }\n        return \"continue\";\n      });\n    });\n    return childSubjectIds;\n  }\n\n  /** Returns ECInstanceIDs of all Models under specific parent Subjects, including their child Subjects, etc. */\n  public async getSubjectModelIds(subjectIds: Id64Array): Promise<Id64Array> {\n    const subjectInfos = await this.getSubjectInfos();\n    const subjectStack = [...subjectIds];\n    const result = new Array<Id64String>();\n    while (true) {\n      const subjectId = subjectStack.pop();\n      if (subjectId === undefined) {\n        break;\n      }\n      const subjectInfo = subjectInfos.get(subjectId);\n      if (!subjectInfo) {\n        continue;\n      }\n      result.push(...subjectInfo.childModels);\n      subjectStack.push(...subjectInfo.childSubjects);\n    }\n    return result;\n  }\n\n  /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */\n  public async getChildSubjectModelIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const subjectInfos = await this.getSubjectInfos();\n\n    const hiddenSubjectIds = new Array<Id64String>();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (childSubjectInfo.hideInHierarchy) {\n          hiddenSubjectIds.push(childSubjectId);\n          return \"continue\";\n        }\n        return \"break\";\n      });\n    });\n\n    const modelIds = new Array<Id64String>();\n    [...parentSubjectIds, ...hiddenSubjectIds].forEach((subjectId) => {\n      const subjectInfo = subjectInfos.get(subjectId);\n      subjectInfo && modelIds.push(...subjectInfo.childModels);\n    });\n    return modelIds;\n  }\n\n  public async createSubjectInstanceKeysPath(targetSubjectId: Id64String): Promise<HierarchyNodeIdentifiersPath> {\n    let entry = this._subjectKeyPaths.get(targetSubjectId);\n    if (!entry) {\n      entry = (async () => {\n        const subjectInfos = await this.getSubjectInfos();\n        const result = new Array<InstanceKey>();\n        let currParentId: Id64String | undefined = targetSubjectId;\n        while (currParentId) {\n          if (this._hierarchyConfig.hideRootSubject && currParentId === IModel.rootSubjectId) {\n            break;\n          }\n          const parentInfo = subjectInfos.get(currParentId);\n          if (!parentInfo?.hideInHierarchy) {\n            result.push({ className: \"BisCore.Subject\", id: currParentId });\n          }\n          currParentId = parentInfo?.parentSubject;\n        }\n        return result.reverse();\n      })();\n      this._subjectKeyPaths.set(targetSubjectId, entry);\n    }\n    return entry;\n  }\n\n  private async *queryModelCategories(): AsyncIterableIterator<{\n    modelId: Id64String;\n    categoryId: Id64String;\n    isModelPrivate: boolean;\n    isRootElementCategory: boolean;\n  }> {\n    const query = `\n      SELECT\n        this.Model.Id modelId,\n        this.Category.Id categoryId,\n        m.IsPrivate isModelPrivate,\n        MAX(IIF(this.Parent.Id IS NULL, 1, 0)) isRootElementCategory\n      FROM BisCore.Model m\n      JOIN ${this._hierarchyConfig.elementClassSpecification} this ON m.ECInstanceId = this.Model.Id\n      GROUP BY modelId, categoryId, isModelPrivate\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, isModelPrivate: !!row.isModelPrivate, isRootElementCategory: !!row.isRootElementCategory };\n    }\n  }\n\n  private async *queryModeledElements() {\n    const query = `\n      SELECT\n        pe.ECInstanceId modeledElementId,\n        pe.Category.Id categoryId,\n        pe.Model.Id modelId\n      FROM BisCore.Model m\n      JOIN ${this._hierarchyConfig.elementClassSpecification} pe ON pe.ECInstanceId = m.ModeledElement.Id\n      WHERE\n        m.IsPrivate = false\n        AND m.ECInstanceId IN (SELECT Model.Id FROM ${this._hierarchyConfig.elementClassSpecification})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId };\n    }\n  }\n\n  private async getModelWithCategoryModeledElements() {\n    this._modelWithCategoryModeledElements ??= (async () => {\n      const modelWithCategoryModeledElements = new Map<Id64String, Id64Set>();\n      for await (const { modelId, categoryId, modeledElementId } of this.queryModeledElements()) {\n        const key = `${modelId}-${categoryId}`;\n        const entry = modelWithCategoryModeledElements.get(key);\n        if (entry === undefined) {\n          modelWithCategoryModeledElements.set(key, new Set([modeledElementId]));\n        } else {\n          entry.add(modeledElementId);\n        }\n      }\n      return modelWithCategoryModeledElements;\n    })();\n    return this._modelWithCategoryModeledElements;\n  }\n\n  private async getModelInfos() {\n    this._modelInfos ??= (async () => {\n      const modelInfos = new Map<Id64String, { categories: Map<Id64String, { isRootElementCategory: boolean }>; isModelPrivate: boolean }>();\n      for await (const { modelId, categoryId, isModelPrivate, isRootElementCategory } of this.queryModelCategories()) {\n        const entry = modelInfos.get(modelId);\n        if (entry) {\n          entry.categories.set(categoryId, { isRootElementCategory });\n          entry.isModelPrivate = isModelPrivate;\n        } else {\n          modelInfos.set(modelId, { categories: new Map([[categoryId, { isRootElementCategory }]]), isModelPrivate });\n        }\n      }\n      return modelInfos;\n    })();\n    return this._modelInfos;\n  }\n\n  public async getModelCategories(modelId: Id64String): Promise<Id64Array> {\n    const modelInfos = await this.getModelInfos();\n    const categories = modelInfos.get(modelId)?.categories.keys();\n    return categories ? [...categories] : [];\n  }\n\n  public async hasSubModel(elementId: Id64String): Promise<boolean> {\n    const modelInfos = await this.getModelInfos();\n    const modeledElementInfo = modelInfos.get(elementId);\n    if (!modeledElementInfo) {\n      return false;\n    }\n    return !modeledElementInfo.isModelPrivate;\n  }\n\n  public async getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Arg): Promise<Id64Array> {\n    const modelWithCategoryModeledElements = await this.getModelWithCategoryModeledElements();\n    const result = new Array<Id64String>();\n    for (const categoryId of Id64.iterable(categoryIds)) {\n      const entry = modelWithCategoryModeledElements.get(`${modelId}-${categoryId}`);\n      if (entry !== undefined) {\n        result.push(...entry);\n      }\n    }\n    return result;\n  }\n\n  public async createModelInstanceKeyPaths(modelId: Id64String): Promise<HierarchyNodeIdentifiersPath[]> {\n    let entry = this._modelKeyPaths.get(modelId);\n    if (!entry) {\n      entry = (async () => {\n        const result = new Array<HierarchyNodeIdentifiersPath>();\n        const subjectInfos = (await this.getSubjectInfos()).entries();\n        for (const [modelSubjectId, subjectInfo] of subjectInfos) {\n          if (subjectInfo.childModels.has(modelId)) {\n            const subjectPath = await this.createSubjectInstanceKeysPath(modelSubjectId);\n            result.push([...subjectPath, { className: \"BisCore.GeometricModel3d\", id: modelId }]);\n          }\n        }\n        return result;\n      })();\n\n      this._modelKeyPaths.set(modelId, entry);\n    }\n    return entry;\n  }\n\n  private async queryCategoryElementCounts(\n    input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n  ): Promise<Array<{ modelId: number; categoryId: number; elementsCount: number }>> {\n    return collect(\n      from(input).pipe(\n        reduce((acc, { modelId, categoryId }) => {\n          const entry = acc.get(modelId);\n          if (!entry) {\n            acc.set(modelId, new Set([categoryId]));\n          } else {\n            entry.add(categoryId);\n          }\n          return acc;\n        }, new Map<Id64String, Id64Set>()),\n        mergeMap((modelCategoryMap) => modelCategoryMap.entries()),\n        map(([modelId, categoryIds]) => `Model.Id = ${modelId} AND Category.Id IN (${[...categoryIds].join(\", \")})`),\n        // we may have thousands of where clauses here, and sending a single query with all of them could take a\n        // long time - instead, split it into smaller chunks\n        bufferCount(100),\n        mergeMap(async (whereClauses) => {\n          const reader = this._queryExecutor.createQueryReader(\n            {\n              ctes: [\n                `\n                    CategoryElements(id, modelId, categoryId) AS (\n                      SELECT ECInstanceId, Model.Id, Category.Id\n                      FROM ${this._hierarchyConfig.elementClassSpecification}\n                      WHERE\n                        Parent.Id IS NULL\n                        AND (\n                          ${whereClauses.join(\" OR \")}\n                        )\n\n                      UNION ALL\n\n                      SELECT c.ECInstanceId, p.modelId, p.categoryId\n                      FROM ${this._hierarchyConfig.elementClassSpecification} c\n                      JOIN CategoryElements p ON c.Parent.Id = p.id\n                    )\n                  `,\n              ],\n              ecsql: `\n                  SELECT modelId, categoryId, COUNT(id) elementsCount\n                  FROM CategoryElements\n                  GROUP BY modelId, categoryId\n                `,\n            },\n            { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" },\n          );\n\n          const result = new Array<{ modelId: number; categoryId: number; elementsCount: number }>();\n          for await (const row of reader) {\n            result.push({ modelId: row.modelId, categoryId: row.categoryId, elementsCount: row.elementsCount });\n          }\n          return result;\n        }),\n        mergeAll(),\n      ),\n    );\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  public async createCategoryInstanceKeyPaths(categoryId: Id64String): Promise<HierarchyNodeIdentifiersPath[]> {\n    let entry = this._categoryKeyPaths.get(categoryId);\n    if (!entry) {\n      entry = (async () => {\n        const result = new Set<Id64String>();\n        const modelInfos = await this.getModelInfos();\n        modelInfos?.forEach((modelInfo, modelId) => {\n          const categoryEntry = modelInfo.categories.get(categoryId);\n          if (categoryEntry?.isRootElementCategory) {\n            result.add(modelId);\n          }\n        });\n\n        const categoryPaths = new Array<HierarchyNodeIdentifiersPath>();\n        for (const categoryModelId of [...result]) {\n          const modelPaths = await this.createModelInstanceKeyPaths(categoryModelId);\n          for (const modelPath of modelPaths) {\n            categoryPaths.push([...modelPath, { className: \"BisCore.SpatialCategory\", id: categoryId }]);\n          }\n        }\n        return categoryPaths;\n      })();\n      this._categoryKeyPaths.set(categoryId, entry);\n    }\n    return entry;\n  }\n}\n\nfunction forEachChildSubject(\n  subjectInfos: Map<Id64String, SubjectInfo>,\n  parentSubject: Id64String | SubjectInfo,\n  cb: (childSubjectId: Id64String, childSubjectInfo: SubjectInfo) => \"break\" | \"continue\",\n) {\n  const parentSubjectInfo = typeof parentSubject === \"string\" ? subjectInfos.get(parentSubject) : parentSubject;\n  parentSubjectInfo &&\n    parentSubjectInfo.childSubjects.forEach((childSubjectId) => {\n      const childSubjectInfo = subjectInfos.get(childSubjectId)!;\n      if (cb(childSubjectId, childSubjectInfo) === \"break\") {\n        return;\n      }\n      forEachChildSubject(subjectInfos, childSubjectInfo, cb);\n    });\n}\n\nclass ModelCategoryElementsCountCache {\n  private _cache = new Map<string, Subject<number>>();\n  private _requestsStream = new Subject<{ modelId: Id64String; categoryId: Id64String }>();\n  private _subscription: Subscription;\n\n  public constructor(\n    private _loader: (\n      input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n    ) => Promise<Array<{ modelId: number; categoryId: number; elementsCount: number }>>,\n  ) {\n    this._subscription = this._requestsStream\n      .pipe(\n        bufferTime(20),\n        filter((requests) => requests.length > 0),\n        mergeMap(async (requests) => this._loader(requests)),\n        mergeAll(),\n      )\n      .subscribe({\n        next: ({ modelId, categoryId, elementsCount }) => {\n          const subject = this._cache.get(`${modelId}${categoryId}`);\n          assert(!!subject);\n          subject.next(elementsCount);\n        },\n      });\n  }\n\n  public [Symbol.dispose]() {\n    this._subscription.unsubscribe();\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    const cacheKey = `${modelId}${categoryId}`;\n    let result = this._cache.get(cacheKey);\n    if (result !== undefined) {\n      return firstValueFrom(result);\n    }\n\n    result = new ReplaySubject(1);\n    this._cache.set(cacheKey, result);\n    this._requestsStream.next({ modelId, categoryId });\n    return firstValueFrom(result);\n  }\n}\n"]}
+{"version":3,"file":"ModelsTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/ModelsTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACtI,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAsBlD,gBAAgB;AAChB,MAAM,OAAO,kBAAkB;IAWnB;IACA;IAXO,sBAAsB,CAAkC;IACjE,aAAa,CAAoD;IACjE,iBAAiB,CAAiC,CAAC,sFAAsF;IACzI,WAAW,CAAkD;IAC7D,iCAAiC,CAA4C;IAC7E,cAAc,CAA2D;IACzE,gBAAgB,CAAyD;IACzE,iBAAiB,CAA2D;IAEpF,YACU,cAA0C,EAC1C,gBAAkD;QADlD,mBAAc,GAAd,cAAc,CAA4B;QAC1C,qBAAgB,GAAhB,gBAAgB,CAAkC;QAE1D,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3H,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC,gBAAgB,GAAG,IAAI,GAAG,EAAE,CAAC;QAClC,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;IACrC,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,aAAa;QAC1B,MAAM,aAAa,GAAG;;;;;;;;;wCASc,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;;;;;;;KAUlF,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACjJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,iBAAiB,EAAE,GAAG,CAAC,iBAAiB,EAAE,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;QACjI,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,WAAW;QACxB,MAAM,WAAW,GAAG;;;;;;UAMd,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,6BAA6B,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,mCAAmC;KACjK,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YAC/I,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC/C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,aAAa,KAAK,CAAC,KAAK,IAAI,EAAE;YACjC,MAAM,CAAC,YAAY,EAAE,uBAAuB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChE,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;oBAClD,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;wBACjD,MAAM,WAAW,GAAgB;4BAC/B,aAAa,EAAE,OAAO,CAAC,QAAQ;4BAC/B,eAAe,EAAE,OAAO,CAAC,eAAe;4BACxC,aAAa,EAAE,IAAI,GAAG,EAAE;4BACxB,WAAW,EAAE,IAAI,GAAG,EAAE;yBACvB,CAAC;wBACF,IAAI,OAAO,CAAC,iBAAiB,EAAE,CAAC;4BAC9B,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;wBACzD,CAAC;wBACD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;oBACtC,CAAC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA+B,CAAC;oBACtD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;wBAC7C,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;oBAC9C,CAAC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YAEH,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;gBACrF,IAAI,eAAe,EAAE,CAAC;oBACpB,MAAM,iBAAiB,GAAG,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;oBAC5D,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC;oBAC5B,iBAAiB,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBACjD,CAAC;YACH,CAAC;YAED,KAAK,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,uBAAuB,EAAE,CAAC;gBAChE,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;oBAC/B,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChD,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;oBACtB,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC3C,CAAC,CAAC,CAAC;YACL,CAAC;YAED,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,kHAAkH;IAC3G,KAAK,CAAC,mBAAmB;QAC9B,IAAI,CAAC,iBAAiB,KAAK,CAAC,KAAK,IAAI,EAAE;YACrC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAClD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;YAC/C,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE;gBAC9C,IAAI,WAAW,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;oBACrC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChC,IAAI,YAAY,GAAG,WAAW,CAAC,aAAa,CAAC;oBAC7C,OAAO,YAAY,EAAE,CAAC;wBACpB,gBAAgB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;wBACnC,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,aAAa,CAAC;oBAC/D,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,gBAAgB,CAAC,CAAC;QAC/B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,kBAAkB,CAAC,gBAA8B;QAC5D,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;QAChD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE,CAAC;oBACtC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACrC,OAAO,OAAO,CAAC;gBACjB,CAAC;gBACD,OAAO,UAAU,CAAC;YACpB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,+GAA+G;IACxG,KAAK,CAAC,kBAAkB,CAAC,UAAqB;QACnD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,MAAM,YAAY,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,EAAE,CAAC;YACrC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,MAAM;YACR,CAAC;YACD,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,SAAS;YACX,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;YACxC,YAAY,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,aAAa,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,6GAA6G;IACtG,KAAK,CAAC,uBAAuB,CAAC,gBAA8B;QACjE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAElD,MAAM,gBAAgB,GAAG,IAAI,KAAK,EAAc,CAAC;QACjD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,gBAAgB,CAAC,eAAe,EAAE,CAAC;oBACrC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACtC,OAAO,UAAU,CAAC;gBACpB,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,KAAK,EAAc,CAAC;QACzC,CAAC,GAAG,gBAAgB,EAAE,GAAG,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YAC/D,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,WAAW,IAAI,QAAQ,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEM,KAAK,CAAC,6BAA6B,CAAC,eAA2B;QACpE,IAAI,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QACvD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;gBAClD,MAAM,MAAM,GAAG,IAAI,KAAK,EAAe,CAAC;gBACxC,IAAI,YAAY,GAA2B,eAAe,CAAC;gBAC3D,OAAO,YAAY,EAAE,CAAC;oBACpB,IAAI,IAAI,CAAC,gBAAgB,CAAC,eAAe,IAAI,YAAY,KAAK,MAAM,CAAC,aAAa,EAAE,CAAC;wBACnF,MAAM;oBACR,CAAC;oBACD,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;oBAClD,IAAI,CAAC,UAAU,EAAE,eAAe,EAAE,CAAC;wBACjC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;oBAClE,CAAC;oBACD,YAAY,GAAG,UAAU,EAAE,aAAa,CAAC;gBAC3C,CAAC;gBACD,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QAMjC,MAAM,KAAK,GAAG;;;;;;;aAOL,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;KAEvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACzI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,cAAc,EAAE,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE,qBAAqB,EAAE,CAAC,CAAC,GAAG,CAAC,qBAAqB,EAAE,CAAC;QACvJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QACjC,MAAM,KAAK,GAAG;;;;;;aAML,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;sDAGN,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;KAChG,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YACzI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACrG,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mCAAmC;QAC/C,IAAI,CAAC,iCAAiC,KAAK,CAAC,KAAK,IAAI,EAAE;YACrD,MAAM,gCAAgC,GAAG,IAAI,GAAG,EAAuB,CAAC;YACxE,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBAC1F,MAAM,GAAG,GAAG,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;gBACvC,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACxD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,gCAAgC,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;gBACzE,CAAC;qBAAM,CAAC;oBACN,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,OAAO,gCAAgC,CAAC;QAC1C,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iCAAiC,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,IAAI,CAAC,WAAW,KAAK,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,EAA4G,CAAC;YACvI,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,qBAAqB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBAC/G,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACtC,IAAI,KAAK,EAAE,CAAC;oBACV,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,qBAAqB,EAAE,CAAC,CAAC;oBAC5D,KAAK,CAAC,cAAc,GAAG,cAAc,CAAC;gBACxC,CAAC;qBAAM,CAAC;oBACN,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,EAAE,qBAAqB,EAAE,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,CAAC,CAAC;gBAC9G,CAAC;YACH,CAAC;YACD,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,OAAmB;QACjD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC;QAC9D,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,SAAqB;QAC5C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,MAAM,kBAAkB,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACxB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,CAAC,kBAAkB,CAAC,cAAc,CAAC;IAC5C,CAAC;IAEM,KAAK,CAAC,4BAA4B,CAAC,OAAmB,EAAE,WAAoB;QACjF,MAAM,gCAAgC,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;QAC1F,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YACpD,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;YAC/E,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,2BAA2B,CAAC,OAAmB;QAC1D,IAAI,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC7C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAG,IAAI,KAAK,EAAgC,CAAC;gBACzD,MAAM,YAAY,GAAG,CAAC,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC9D,KAAK,MAAM,CAAC,cAAc,EAAE,WAAW,CAAC,IAAI,YAAY,EAAE,CAAC;oBACzD,IAAI,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;wBACzC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,6BAA6B,CAAC,cAAc,CAAC,CAAC;wBAC7E,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,WAAW,EAAE,EAAE,SAAS,EAAE,0BAA0B,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;oBACxF,CAAC;gBACH,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,EAAE,CAAC;YAEL,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACtC,KAA6D;QAE7D,OAAO,OAAO,CACZ,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CACd,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE;YACtC,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC/B,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC1C,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACxB,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAuB,CAAC,EAClC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,EAC1D,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,cAAc,OAAO,wBAAwB,CAAC,GAAG,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;QAC5G,wGAAwG;QACxG,oDAAoD;QACpD,WAAW,CAAC,GAAG,CAAC,EAChB,QAAQ,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE;YAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAClD;gBACE,IAAI,EAAE;oBACJ;;;2BAGW,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;0BAIhD,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC;;;;;;2BAMxB,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;iBAGzD;iBACF;gBACD,KAAK,EAAE;;;;eAIN;aACF,EACD,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CACxD,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,KAAK,EAA0E,CAAC;YACnG,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;gBAC/B,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,aAAa,EAAE,GAAG,CAAC,aAAa,EAAE,CAAC,CAAC;YACtG,CAAC;YAED,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE;gBACxC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,UAAU,KAAK,UAAU,IAAI,aAAa,CAAC,OAAO,KAAK,OAAO,CAAC,EAAE,CAAC;oBAClH,MAAM,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EACF,QAAQ,EAAE,CACX,CACF,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,KAAK,CAAC,8BAA8B,CAAC,UAAsB;QAChE,IAAI,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACnD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAG,IAAI,GAAG,EAAc,CAAC;gBACrC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;gBAC9C,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,EAAE,OAAO,EAAE,EAAE;oBACzC,MAAM,aAAa,GAAG,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC3D,IAAI,aAAa,EAAE,qBAAqB,EAAE,CAAC;wBACzC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBACtB,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,MAAM,aAAa,GAAG,IAAI,KAAK,EAAgC,CAAC;gBAChE,KAAK,MAAM,eAAe,IAAI,CAAC,GAAG,MAAM,CAAC,EAAE,CAAC;oBAC1C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,eAAe,CAAC,CAAC;oBAC3E,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;wBACnC,aAAa,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,EAAE,EAAE,SAAS,EAAE,yBAAyB,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;oBAC/F,CAAC;gBACH,CAAC;gBACD,OAAO,aAAa,CAAC;YACvB,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED,SAAS,mBAAmB,CAC1B,YAA0C,EAC1C,aAAuC,EACvC,EAAuF;IAEvF,MAAM,iBAAiB,GAAG,OAAO,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;IAC9G,iBAAiB;QACf,iBAAiB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YACzD,MAAM,gBAAgB,GAAG,YAAY,CAAC,GAAG,CAAC,cAAc,CAAE,CAAC;YAC3D,IAAI,EAAE,CAAC,cAAc,EAAE,gBAAgB,CAAC,KAAK,OAAO,EAAE,CAAC;gBACrD,OAAO;YACT,CAAC;YACD,mBAAmB,CAAC,YAAY,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,+BAA+B;IAMzB;IALF,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;IAC5C,eAAe,GAAG,IAAI,OAAO,EAAmD,CAAC;IACjF,aAAa,CAAe;IAEpC,YACU,OAEmF;QAFnF,YAAO,GAAP,OAAO,CAE4E;QAE3F,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,eAAe;aACtC,IAAI,CACH,UAAU,CAAC,EAAE,CAAC,EACd,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EACzC,QAAQ,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EACpD,QAAQ,EAAE,CACX;aACA,SAAS,CAAC;YACT,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,OAAO,GAAG,UAAU,EAAE,CAAC,CAAC;gBAC3D,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9B,CAAC;SACF,CAAC,CAAC;IACP,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;IACnC,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,MAAM,QAAQ,GAAG,GAAG,OAAO,GAAG,UAAU,EAAE,CAAC;QAC3C,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC;QAED,MAAM,GAAG,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAClC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { bufferCount, bufferTime, filter, firstValueFrom, from, map, mergeAll, mergeMap, reduce, ReplaySubject, Subject } from \"rxjs\";\nimport { assert, Id64 } from \"@itwin/core-bentley\";\nimport { IModel } from \"@itwin/core-common\";\nimport { collect } from \"../../common/Rxjs.js\";\nimport { pushToMap } from \"../../common/Utils.js\";\n\nimport type { Subscription } from \"rxjs\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\nimport type { ModelsTreeDefinition } from \"../ModelsTreeDefinition.js\";\nimport type { Id64Arg, Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\n\ninterface SubjectInfo {\n  parentSubject: Id64String | undefined;\n  hideInHierarchy: boolean;\n  childSubjects: Id64Set;\n  childModels: Id64Set;\n}\n\ninterface ModelInfo {\n  isModelPrivate: boolean;\n  categories: Map<Id64String, { isRootElementCategory: boolean }>;\n}\n\ntype ModelsTreeHierarchyConfiguration = ConstructorParameters<typeof ModelsTreeDefinition>[0][\"hierarchyConfig\"];\n\n/** @internal */\nexport class ModelsTreeIdsCache implements Disposable {\n  private readonly _categoryElementCounts: ModelCategoryElementsCountCache;\n  private _subjectInfos: Promise<Map<Id64String, SubjectInfo>> | undefined;\n  private _parentSubjectIds: Promise<Id64Array> | undefined; // the list should contain a subject id if its node should be shown as having children\n  private _modelInfos: Promise<Map<Id64String, ModelInfo>> | undefined;\n  private _modelWithCategoryModeledElements: Promise<Map<string, Id64Set>> | undefined;\n  private _modelKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath[]>>;\n  private _subjectKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath>>;\n  private _categoryKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath[]>>;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    private _hierarchyConfig: ModelsTreeHierarchyConfiguration,\n  ) {\n    this._categoryElementCounts = new ModelCategoryElementsCountCache(async (input) => this.queryCategoryElementCounts(input));\n    this._modelKeyPaths = new Map();\n    this._subjectKeyPaths = new Map();\n    this._categoryKeyPaths = new Map();\n  }\n\n  public [Symbol.dispose]() {\n    this._categoryElementCounts[Symbol.dispose]();\n  }\n\n  private async *querySubjects(): AsyncIterableIterator<{ id: Id64String; parentId?: Id64String; targetPartitionId?: Id64String; hideInHierarchy: boolean }> {\n    const subjectsQuery = `\n      SELECT\n        s.ECInstanceId id,\n        s.Parent.Id parentId,\n        (\n          SELECT m.ECInstanceId\n          FROM bis.GeometricModel3d m\n          WHERE m.ECInstanceId = HexToId(json_extract(s.JsonProperties, '$.Subject.Model.TargetPartition'))\n            AND NOT m.IsPrivate\n            AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)\n        ) targetPartitionId,\n        CASE\n          WHEN (\n            json_extract(s.JsonProperties, '$.Subject.Job.Bridge') IS NOT NULL\n            OR json_extract(s.JsonProperties, '$.Subject.Model.Type') = 'Hierarchy'\n          ) THEN 1\n          ELSE 0\n        END hideInHierarchy\n      FROM bis.Subject s\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: subjectsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId, targetPartitionId: row.targetPartitionId, hideInHierarchy: !!row.hideInHierarchy };\n    }\n  }\n\n  private async *queryModels(): AsyncIterableIterator<{ id: Id64String; parentId: Id64String }> {\n    const modelsQuery = `\n      SELECT p.ECInstanceId id, p.Parent.Id parentId\n      FROM bis.InformationPartitionElement p\n      INNER JOIN bis.GeometricModel3d m ON m.ModeledElement.Id = p.ECInstanceId\n      WHERE\n        NOT m.IsPrivate\n        ${this._hierarchyConfig.showEmptyModels ? \"\" : `AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)`}\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: modelsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId };\n    }\n  }\n\n  private async getSubjectInfos() {\n    this._subjectInfos ??= (async () => {\n      const [subjectInfos, targetPartitionSubjects] = await Promise.all([\n        (async () => {\n          const result = new Map<Id64String, SubjectInfo>();\n          for await (const subject of this.querySubjects()) {\n            const subjectInfo: SubjectInfo = {\n              parentSubject: subject.parentId,\n              hideInHierarchy: subject.hideInHierarchy,\n              childSubjects: new Set(),\n              childModels: new Set(),\n            };\n            if (subject.targetPartitionId) {\n              subjectInfo.childModels.add(subject.targetPartitionId);\n            }\n            result.set(subject.id, subjectInfo);\n          }\n          return result;\n        })(),\n        (async () => {\n          const result = new Map<Id64String, Set<Id64String>>();\n          for await (const model of this.queryModels()) {\n            pushToMap(result, model.id, model.parentId);\n          }\n          return result;\n        })(),\n      ]);\n\n      for (const [subjectId, { parentSubject: parentSubjectId }] of subjectInfos.entries()) {\n        if (parentSubjectId) {\n          const parentSubjectInfo = subjectInfos.get(parentSubjectId);\n          assert(!!parentSubjectInfo);\n          parentSubjectInfo.childSubjects.add(subjectId);\n        }\n      }\n\n      for (const [partitionId, subjectIds] of targetPartitionSubjects) {\n        subjectIds.forEach((subjectId) => {\n          const subjectInfo = subjectInfos.get(subjectId);\n          assert(!!subjectInfo);\n          subjectInfo.childModels.add(partitionId);\n        });\n      }\n\n      return subjectInfos;\n    })();\n    return this._subjectInfos;\n  }\n\n  /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */\n  public async getParentSubjectIds(): Promise<Id64String[]> {\n    this._parentSubjectIds ??= (async () => {\n      const subjectInfos = await this.getSubjectInfos();\n      const parentSubjectIds = new Set<Id64String>();\n      subjectInfos.forEach((subjectInfo, subjectId) => {\n        if (subjectInfo.childModels.size > 0) {\n          parentSubjectIds.add(subjectId);\n          let currParentId = subjectInfo.parentSubject;\n          while (currParentId) {\n            parentSubjectIds.add(currParentId);\n            currParentId = subjectInfos.get(currParentId)?.parentSubject;\n          }\n        }\n      });\n      return [...parentSubjectIds];\n    })();\n    return this._parentSubjectIds;\n  }\n\n  /**\n   * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into\n   * account `hideInHierarchy` flag.\n   */\n  public async getChildSubjectIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const childSubjectIds = new Array<Id64String>();\n    const subjectInfos = await this.getSubjectInfos();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (!childSubjectInfo.hideInHierarchy) {\n          childSubjectIds.push(childSubjectId);\n          return \"break\";\n        }\n        return \"continue\";\n      });\n    });\n    return childSubjectIds;\n  }\n\n  /** Returns ECInstanceIDs of all Models under specific parent Subjects, including their child Subjects, etc. */\n  public async getSubjectModelIds(subjectIds: Id64Array): Promise<Id64Array> {\n    const subjectInfos = await this.getSubjectInfos();\n    const subjectStack = [...subjectIds];\n    const result = new Array<Id64String>();\n    while (true) {\n      const subjectId = subjectStack.pop();\n      if (subjectId === undefined) {\n        break;\n      }\n      const subjectInfo = subjectInfos.get(subjectId);\n      if (!subjectInfo) {\n        continue;\n      }\n      result.push(...subjectInfo.childModels);\n      subjectStack.push(...subjectInfo.childSubjects);\n    }\n    return result;\n  }\n\n  /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */\n  public async getChildSubjectModelIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const subjectInfos = await this.getSubjectInfos();\n\n    const hiddenSubjectIds = new Array<Id64String>();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (childSubjectInfo.hideInHierarchy) {\n          hiddenSubjectIds.push(childSubjectId);\n          return \"continue\";\n        }\n        return \"break\";\n      });\n    });\n\n    const modelIds = new Array<Id64String>();\n    [...parentSubjectIds, ...hiddenSubjectIds].forEach((subjectId) => {\n      const subjectInfo = subjectInfos.get(subjectId);\n      subjectInfo && modelIds.push(...subjectInfo.childModels);\n    });\n    return modelIds;\n  }\n\n  public async createSubjectInstanceKeysPath(targetSubjectId: Id64String): Promise<HierarchyNodeIdentifiersPath> {\n    let entry = this._subjectKeyPaths.get(targetSubjectId);\n    if (!entry) {\n      entry = (async () => {\n        const subjectInfos = await this.getSubjectInfos();\n        const result = new Array<InstanceKey>();\n        let currParentId: Id64String | undefined = targetSubjectId;\n        while (currParentId) {\n          if (this._hierarchyConfig.hideRootSubject && currParentId === IModel.rootSubjectId) {\n            break;\n          }\n          const parentInfo = subjectInfos.get(currParentId);\n          if (!parentInfo?.hideInHierarchy) {\n            result.push({ className: \"BisCore.Subject\", id: currParentId });\n          }\n          currParentId = parentInfo?.parentSubject;\n        }\n        return result.reverse();\n      })();\n      this._subjectKeyPaths.set(targetSubjectId, entry);\n    }\n    return entry;\n  }\n\n  private async *queryModelCategories(): AsyncIterableIterator<{\n    modelId: Id64String;\n    categoryId: Id64String;\n    isModelPrivate: boolean;\n    isRootElementCategory: boolean;\n  }> {\n    const query = `\n      SELECT\n        this.Model.Id modelId,\n        this.Category.Id categoryId,\n        m.IsPrivate isModelPrivate,\n        MAX(IIF(this.Parent.Id IS NULL, 1, 0)) isRootElementCategory\n      FROM BisCore.Model m\n      JOIN ${this._hierarchyConfig.elementClassSpecification} this ON m.ECInstanceId = this.Model.Id\n      GROUP BY modelId, categoryId, isModelPrivate\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, isModelPrivate: !!row.isModelPrivate, isRootElementCategory: !!row.isRootElementCategory };\n    }\n  }\n\n  private async *queryModeledElements() {\n    const query = `\n      SELECT\n        pe.ECInstanceId modeledElementId,\n        pe.Category.Id categoryId,\n        pe.Model.Id modelId\n      FROM BisCore.Model m\n      JOIN ${this._hierarchyConfig.elementClassSpecification} pe ON pe.ECInstanceId = m.ModeledElement.Id\n      WHERE\n        m.IsPrivate = false\n        AND m.ECInstanceId IN (SELECT Model.Id FROM ${this._hierarchyConfig.elementClassSpecification})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId };\n    }\n  }\n\n  private async getModelWithCategoryModeledElements() {\n    this._modelWithCategoryModeledElements ??= (async () => {\n      const modelWithCategoryModeledElements = new Map<Id64String, Id64Set>();\n      for await (const { modelId, categoryId, modeledElementId } of this.queryModeledElements()) {\n        const key = `${modelId}-${categoryId}`;\n        const entry = modelWithCategoryModeledElements.get(key);\n        if (entry === undefined) {\n          modelWithCategoryModeledElements.set(key, new Set([modeledElementId]));\n        } else {\n          entry.add(modeledElementId);\n        }\n      }\n      return modelWithCategoryModeledElements;\n    })();\n    return this._modelWithCategoryModeledElements;\n  }\n\n  private async getModelInfos() {\n    this._modelInfos ??= (async () => {\n      const modelInfos = new Map<Id64String, { categories: Map<Id64String, { isRootElementCategory: boolean }>; isModelPrivate: boolean }>();\n      for await (const { modelId, categoryId, isModelPrivate, isRootElementCategory } of this.queryModelCategories()) {\n        const entry = modelInfos.get(modelId);\n        if (entry) {\n          entry.categories.set(categoryId, { isRootElementCategory });\n          entry.isModelPrivate = isModelPrivate;\n        } else {\n          modelInfos.set(modelId, { categories: new Map([[categoryId, { isRootElementCategory }]]), isModelPrivate });\n        }\n      }\n      return modelInfos;\n    })();\n    return this._modelInfos;\n  }\n\n  public async getModelCategories(modelId: Id64String): Promise<Id64Array> {\n    const modelInfos = await this.getModelInfos();\n    const categories = modelInfos.get(modelId)?.categories.keys();\n    return categories ? [...categories] : [];\n  }\n\n  public async hasSubModel(elementId: Id64String): Promise<boolean> {\n    const modelInfos = await this.getModelInfos();\n    const modeledElementInfo = modelInfos.get(elementId);\n    if (!modeledElementInfo) {\n      return false;\n    }\n    return !modeledElementInfo.isModelPrivate;\n  }\n\n  public async getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Arg): Promise<Id64Array> {\n    const modelWithCategoryModeledElements = await this.getModelWithCategoryModeledElements();\n    const result = new Array<Id64String>();\n    for (const categoryId of Id64.iterable(categoryIds)) {\n      const entry = modelWithCategoryModeledElements.get(`${modelId}-${categoryId}`);\n      if (entry !== undefined) {\n        result.push(...entry);\n      }\n    }\n    return result;\n  }\n\n  public async createModelInstanceKeyPaths(modelId: Id64String): Promise<HierarchyNodeIdentifiersPath[]> {\n    let entry = this._modelKeyPaths.get(modelId);\n    if (!entry) {\n      entry = (async () => {\n        const result = new Array<HierarchyNodeIdentifiersPath>();\n        const subjectInfos = (await this.getSubjectInfos()).entries();\n        for (const [modelSubjectId, subjectInfo] of subjectInfos) {\n          if (subjectInfo.childModels.has(modelId)) {\n            const subjectPath = await this.createSubjectInstanceKeysPath(modelSubjectId);\n            result.push([...subjectPath, { className: \"BisCore.GeometricModel3d\", id: modelId }]);\n          }\n        }\n        return result;\n      })();\n\n      this._modelKeyPaths.set(modelId, entry);\n    }\n    return entry;\n  }\n\n  private async queryCategoryElementCounts(\n    input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n  ): Promise<Array<{ modelId: Id64String; categoryId: Id64String; elementsCount: number }>> {\n    return collect(\n      from(input).pipe(\n        reduce((acc, { modelId, categoryId }) => {\n          const entry = acc.get(modelId);\n          if (!entry) {\n            acc.set(modelId, new Set([categoryId]));\n          } else {\n            entry.add(categoryId);\n          }\n          return acc;\n        }, new Map<Id64String, Id64Set>()),\n        mergeMap((modelCategoryMap) => modelCategoryMap.entries()),\n        map(([modelId, categoryIds]) => `Model.Id = ${modelId} AND Category.Id IN (${[...categoryIds].join(\", \")})`),\n        // we may have thousands of where clauses here, and sending a single query with all of them could take a\n        // long time - instead, split it into smaller chunks\n        bufferCount(100),\n        mergeMap(async (whereClauses) => {\n          const reader = this._queryExecutor.createQueryReader(\n            {\n              ctes: [\n                `\n                  CategoryElements(id, modelId, categoryId) AS (\n                    SELECT ECInstanceId, Model.Id, Category.Id\n                    FROM ${this._hierarchyConfig.elementClassSpecification}\n                    WHERE\n                      Parent.Id IS NULL\n                      AND (\n                        ${whereClauses.join(\" OR \")}\n                      )\n\n                    UNION ALL\n\n                    SELECT c.ECInstanceId, p.modelId, p.categoryId\n                    FROM ${this._hierarchyConfig.elementClassSpecification} c\n                    JOIN CategoryElements p ON c.Parent.Id = p.id\n                  )\n                `,\n              ],\n              ecsql: `\n                SELECT modelId, categoryId, COUNT(id) elementsCount\n                FROM CategoryElements\n                GROUP BY modelId, categoryId\n              `,\n            },\n            { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" },\n          );\n\n          const result = new Array<{ modelId: Id64String; categoryId: Id64String; elementsCount: number }>();\n          for await (const row of reader) {\n            result.push({ modelId: row.modelId, categoryId: row.categoryId, elementsCount: row.elementsCount });\n          }\n\n          input.forEach(({ modelId, categoryId }) => {\n            if (!result.some((queriedResult) => queriedResult.categoryId === categoryId && queriedResult.modelId === modelId)) {\n              result.push({ categoryId, modelId, elementsCount: 0 });\n            }\n          });\n\n          return result;\n        }),\n        mergeAll(),\n      ),\n    );\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  public async createCategoryInstanceKeyPaths(categoryId: Id64String): Promise<HierarchyNodeIdentifiersPath[]> {\n    let entry = this._categoryKeyPaths.get(categoryId);\n    if (!entry) {\n      entry = (async () => {\n        const result = new Set<Id64String>();\n        const modelInfos = await this.getModelInfos();\n        modelInfos?.forEach((modelInfo, modelId) => {\n          const categoryEntry = modelInfo.categories.get(categoryId);\n          if (categoryEntry?.isRootElementCategory) {\n            result.add(modelId);\n          }\n        });\n\n        const categoryPaths = new Array<HierarchyNodeIdentifiersPath>();\n        for (const categoryModelId of [...result]) {\n          const modelPaths = await this.createModelInstanceKeyPaths(categoryModelId);\n          for (const modelPath of modelPaths) {\n            categoryPaths.push([...modelPath, { className: \"BisCore.SpatialCategory\", id: categoryId }]);\n          }\n        }\n        return categoryPaths;\n      })();\n      this._categoryKeyPaths.set(categoryId, entry);\n    }\n    return entry;\n  }\n}\n\nfunction forEachChildSubject(\n  subjectInfos: Map<Id64String, SubjectInfo>,\n  parentSubject: Id64String | SubjectInfo,\n  cb: (childSubjectId: Id64String, childSubjectInfo: SubjectInfo) => \"break\" | \"continue\",\n) {\n  const parentSubjectInfo = typeof parentSubject === \"string\" ? subjectInfos.get(parentSubject) : parentSubject;\n  parentSubjectInfo &&\n    parentSubjectInfo.childSubjects.forEach((childSubjectId) => {\n      const childSubjectInfo = subjectInfos.get(childSubjectId)!;\n      if (cb(childSubjectId, childSubjectInfo) === \"break\") {\n        return;\n      }\n      forEachChildSubject(subjectInfos, childSubjectInfo, cb);\n    });\n}\n\nclass ModelCategoryElementsCountCache {\n  private _cache = new Map<string, Subject<number>>();\n  private _requestsStream = new Subject<{ modelId: Id64String; categoryId: Id64String }>();\n  private _subscription: Subscription;\n\n  public constructor(\n    private _loader: (\n      input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n    ) => Promise<Array<{ modelId: Id64String; categoryId: Id64String; elementsCount: number }>>,\n  ) {\n    this._subscription = this._requestsStream\n      .pipe(\n        bufferTime(20),\n        filter((requests) => requests.length > 0),\n        mergeMap(async (requests) => this._loader(requests)),\n        mergeAll(),\n      )\n      .subscribe({\n        next: ({ modelId, categoryId, elementsCount }) => {\n          const subject = this._cache.get(`${modelId}${categoryId}`);\n          assert(!!subject);\n          subject.next(elementsCount);\n        },\n      });\n  }\n\n  public [Symbol.dispose]() {\n    this._subscription.unsubscribe();\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    const cacheKey = `${modelId}${categoryId}`;\n    let result = this._cache.get(cacheKey);\n    if (result !== undefined) {\n      return firstValueFrom(result);\n    }\n\n    result = new ReplaySubject(1);\n    this._cache.set(cacheKey, result);\n    this._requestsStream.next({ modelId, categoryId });\n    return firstValueFrom(result);\n  }\n}\n"]}

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

@@ -1,9 +1,9 @@
-import type { GroupingHierarchyNode, HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
-import type { HierarchyVisibilityHandler, HierarchyVisibilityHandlerOverridableMethod, VisibilityStatus } from "../../common/UseHierarchyVisibility.js";
-import type { ModelsTreeIdsCache } from "./ModelsTreeIdsCache.js";
 import type { Id64Arg, Id64Array, Id64Set, Id64String } from "@itwin/core-bentley";
 import type { Viewport } from "@itwin/core-frontend";
+import type { GroupingHierarchyNode, HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
 import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
+import type { HierarchyVisibilityHandler, HierarchyVisibilityHandlerOverridableMethod, VisibilityStatus } from "../../common/UseHierarchyVisibility.js";
+import type { ModelsTreeIdsCache } from "./ModelsTreeIdsCache.js";
 /** @beta */
 interface GetCategoryVisibilityStatusProps {
     categoryIds: Id64Arg;

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

@@ -586,6 +586,9 @@ class ModelsTreeVisibilityHandlerImpl {
     }
     getVisibilityFromAlwaysAndNeverDrawnElementsImpl(props) {
         const { alwaysDrawn, neverDrawn, totalCount, ignoreTooltip } = props;
+        if (totalCount === 0) {
+            return props.defaultStatus();
+        }
         if (neverDrawn?.size === totalCount) {
             return createVisibilityStatus("hidden", getTooltipOptions(props.tooltips.allElementsInNeverDrawnList, ignoreTooltip));
         }
@@ -630,7 +633,16 @@ class ModelsTreeVisibilityHandlerImpl {
                 totalCount,
                 alwaysDrawn: this.getAlwaysDrawnElements({ categoryIds: categoryId, modelId }),
                 neverDrawn: this.getNeverDrawnElements({ categoryIds: categoryId, modelId }),
-            }).pipe(map((state) => {
+            }).pipe(
+            // There is a known bug:
+            // Categories that don't have root elements will make visibility result incorrect
+            // E.g.:
+            // - CategoryA
+            //  - ElementA (CategoryA is visible)
+            //    - ChildElementB (CategoryB is hidden) ChildElementB is in always drawn list
+            // Result will be "partial" because CategoryB will return hidden visibility, even though all elements are visible
+            // TODO fix with: https://github.com/iTwin/viewer-components-react/issues/1100
+            map((state) => {
                 return this.getVisibilityFromAlwaysAndNeverDrawnElementsImpl({
                     ...props,
                     ...state,