@itwin/unified-selection 1.7.1 → 1.7.3

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # @itwin/unified-selection
 
+## 1.7.3
+
+### Patch Changes
+
+- [#1338](https://github.com/iTwin/presentation/pull/1338): Bump iTwin.js core dependencies to `^5.9.1`.
+- Updated dependencies:
+  - @itwin/presentation-shared@1.2.13
+
+## 1.7.2
+
+### Patch Changes
+
+- [#1330](https://github.com/iTwin/presentation/pull/1330): `computeSelection`: Yield transient element IDs as selectable instance keys with the `TRANSIENT_ELEMENT_CLASSNAME` class name, instead of silently dropping them. This ensures transient selections are preserved in the computed selection result.
+
 ## 1.7.1
 
 ### Patch Changes

package/lib/cjs/unified-selection/SelectionScope.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionScope.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/SelectionScope.ts"],"names":[],"mappings":"AAKA,OAAO,EAAc,OAAO,EAAa,MAAM,qBAAqB,CAAC;AACrE,OAAO,EAGL,kBAAkB,EAGnB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAGxD;;;GAGG;AACH,KAAK,wBAAwB,GAAG,SAAS,GAAG,OAAO,GAAG,UAAU,GAAG,YAAY,CAAC;AAEhF;;;GAGG;AACH,MAAM,WAAW,0BAA0B;IACzC,0EAA0E;IAC1E,EAAE,EAAE,SAAS,GAAG,YAAY,CAAC;IAC7B;;;;;;;OAOG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,0BAA0B,GAAG;IAAE,EAAE,EAAE,wBAAwB,CAAA;CAAE,GAAG,wBAAwB,CAAC;AAEtH;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,6BAA6B;IAC7B,aAAa,EAAE,kBAAkB,CAAC;IAClC,gDAAgD;IAChD,UAAU,EAAE,OAAO,CAAC;IACpB,iDAAiD;IACjD,KAAK,EAAE,cAAc,CAAC;IACtB,sJAAsJ;IACtJ,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;;GAGG;AACH,wBAAuB,gBAAgB,CAAC,KAAK,EAAE,qBAAqB,GAAG,qBAAqB,CAAC,qBAAqB,CAAC,CA0ClH"}
+{"version":3,"file":"SelectionScope.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/SelectionScope.ts"],"names":[],"mappings":"AAKA,OAAO,EAAc,OAAO,EAAsB,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EAGL,kBAAkB,EAGnB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,qBAAqB,EAA+B,MAAM,iBAAiB,CAAC;AAGrF;;;GAGG;AACH,KAAK,wBAAwB,GAAG,SAAS,GAAG,OAAO,GAAG,UAAU,GAAG,YAAY,CAAC;AAEhF;;;GAGG;AACH,MAAM,WAAW,0BAA0B;IACzC,0EAA0E;IAC1E,EAAE,EAAE,SAAS,GAAG,YAAY,CAAC;IAC7B;;;;;;;OAOG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,0BAA0B,GAAG;IAAE,EAAE,EAAE,wBAAwB,CAAA;CAAE,GAAG,wBAAwB,CAAC;AAEtH;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,6BAA6B;IAC7B,aAAa,EAAE,kBAAkB,CAAC;IAClC,gDAAgD;IAChD,UAAU,EAAE,OAAO,CAAC;IACpB,iDAAiD;IACjD,KAAK,EAAE,cAAc,CAAC;IACtB,sJAAsJ;IACtJ,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;;GAGG;AACH,wBAAuB,gBAAgB,CAAC,KAAK,EAAE,qBAAqB,GAAG,qBAAqB,CAAC,qBAAqB,CAAC,CAyClH"}

package/lib/cjs/unified-selection/SelectionScope.js

@@ -6,6 +6,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.computeSelection = computeSelection;
 const core_bentley_1 = require("@itwin/core-bentley");
+const Selectable_js_1 = require("./Selectable.js");
 const Utils_js_1 = require("./Utils.js");
 /**
  * Computes selection from given element IDs.
@@ -19,28 +20,26 @@ async function* computeSelection(props) {
         yield* computeSelection({ queryExecutor, elementIds, scope: { id: scope }, componentId });
         return;
     }
-    const nonTransientIds = [];
-    for (const id of core_bentley_1.Id64.iterable(elementIds)) {
-        if (!core_bentley_1.Id64.isTransient(id)) {
-            nonTransientIds.push(id);
-        }
+    const { persistent: persistentIds, transient: transientIds } = parseElementIds(elementIds);
+    for (const id of transientIds) {
+        yield { id, className: Selectable_js_1.TRANSIENT_ELEMENT_CLASSNAME };
     }
     switch (scope.id) {
         case "element":
-            yield* computeElementSelection(queryExecutor, nonTransientIds, scope.ancestorLevel ?? 0, componentId, componentName);
+            yield* computeElementSelection(queryExecutor, persistentIds, scope.ancestorLevel ?? 0, componentId, componentName);
             return;
         case "category":
-            yield* computeCategorySelection(queryExecutor, nonTransientIds, componentId, componentName);
+            yield* computeCategorySelection(queryExecutor, persistentIds, componentId, componentName);
             return;
         case "model":
-            yield* computeModelSelection(queryExecutor, nonTransientIds, componentId, componentName);
+            yield* computeModelSelection(queryExecutor, persistentIds, componentId, componentName);
             return;
         case "functional":
-            yield* computeFunctionalElementSelection(queryExecutor, nonTransientIds, scope.ancestorLevel ?? 0, componentId, componentName);
+            yield* computeFunctionalElementSelection(queryExecutor, persistentIds, scope.ancestorLevel ?? 0, componentId, componentName);
             return;
     }
 }
-async function* computeElementSelection(queryExecutor, elementIds, ancestorLevel, componentId, componentName) {
+async function* computeElementSelection(queryExecutor, ids, ancestorLevel, componentId, componentName) {
     const bindings = [];
     const recurseUntilRoot = ancestorLevel < 0;
     const ctes = [
@@ -48,7 +47,7 @@ async function* computeElementSelection(queryExecutor, elementIds, ancestorLevel
       AncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (
         SELECT ECInstanceId, ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, Parent.Id
         FROM BisCore.Element
-        WHERE ${(0, Utils_js_1.formIdBindings)("ECInstanceId", elementIds, bindings)}
+        WHERE ${(0, Utils_js_1.formIdBindings)("ECInstanceId", ids, bindings)}
         UNION ALL
         SELECT pe.ECInstanceId, pe.ECClassId, a.Depth - 1, pe.Parent.Id
         FROM AncestorElements a
@@ -211,4 +210,43 @@ async function* executeQuery(props) {
         parseQueryRow: (row) => ({ className: row.ClassName, id: row.ECInstanceId }),
     });
 }
+function parseElementIds(ids) {
+    if (typeof ids === "string") {
+        ids = [ids];
+    }
+    // detect if all ids are either transient or persistent to avoid unnecessary copying
+    let allPersistent = true;
+    let allTransient = true;
+    for (const id of ids) {
+        if (core_bentley_1.Id64.isTransient(id)) {
+            allPersistent = false;
+        }
+        else {
+            allTransient = false;
+        }
+        if (!allPersistent && !allTransient) {
+            break;
+        }
+    }
+    // avoid making a copy if ids are only persistent or only transient
+    if (allPersistent) {
+        return { persistent: ids, transient: [] };
+    }
+    else if (allTransient) {
+        return { persistent: [], transient: ids };
+    }
+    // if `ids` contain mixed ids, we have to copy.. use Array instead of
+    // a Set for performance
+    const persistentElementIds = [];
+    const transientElementIds = [];
+    for (const id of ids) {
+        if (core_bentley_1.Id64.isTransient(id)) {
+            transientElementIds.push(id);
+        }
+        else {
+            persistentElementIds.push(id);
+        }
+    }
+    return { persistent: persistentElementIds, transient: transientElementIds };
+}
 //# sourceMappingURL=SelectionScope.js.map

package/lib/cjs/unified-selection/SelectionScope.js.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionScope.js","sourceRoot":"","sources":["../../../src/unified-selection/SelectionScope.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;AA+DhG,4CA0CC;AAvGD,sDAAqE;AASrE,yCAAiE;AAgDjE;;;GAGG;AACI,KAAK,SAAS,CAAC,CAAC,gBAAgB,CAAC,KAA4B;IAClE,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;IACnD,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,mBAAI,CAAC,WAAW,EAAE,CAAC;IAC5D,MAAM,aAAa,GAAG,kBAAkB,CAAC;IACzC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,KAAK,CAAC,CAAC,gBAAgB,CAAC,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;QAC1F,OAAO;IACT,CAAC;IAED,MAAM,eAAe,GAAc,EAAE,CAAC;IACtC,KAAK,MAAM,EAAE,IAAI,mBAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QAC3C,IAAI,CAAC,mBAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YAC1B,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,QAAQ,KAAK,CAAC,EAAE,EAAE,CAAC;QACjB,KAAK,SAAS;YACZ,KAAK,CAAC,CAAC,uBAAuB,CAC5B,aAAa,EACb,eAAe,EACd,KAAoC,CAAC,aAAa,IAAI,CAAC,EACxD,WAAW,EACX,aAAa,CACd,CAAC;YACF,OAAO;QACT,KAAK,UAAU;YACb,KAAK,CAAC,CAAC,wBAAwB,CAAC,aAAa,EAAE,eAAe,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;YAC5F,OAAO;QACT,KAAK,OAAO;YACV,KAAK,CAAC,CAAC,qBAAqB,CAAC,aAAa,EAAE,eAAe,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;YACzF,OAAO;QACT,KAAK,YAAY;YACf,KAAK,CAAC,CAAC,iCAAiC,CACtC,aAAa,EACb,eAAe,EACd,KAAoC,CAAC,aAAa,IAAI,CAAC,EACxD,WAAW,EACX,aAAa,CACd,CAAC;YACF,OAAO;IACX,CAAC;AACH,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,uBAAuB,CACrC,aAAiC,EACjC,UAAoB,EACpB,aAAqB,EACrB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,gBAAgB,GAAG,aAAa,GAAG,CAAC,CAAC;IAC3C,MAAM,IAAI,GAAG;QACX;;0CAEsC,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;gBAE3E,IAAA,yBAAc,EAAC,cAAc,EAAE,UAAU,EAAE,QAAQ,CAAC;;;;;gBAKpD,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;KACF,CAAC;IACF,MAAM,KAAK,GAAG;;;YAGJ,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc;GAC/C,CAAC;IACF,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;QAChC,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,YAAY,mBAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KAC1F,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,wBAAwB,CACtC,aAAiC,EACjC,GAAa,EACb,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,KAAK,GAAG;QACZ;;;;cAIU,IAAA,yBAAc,EAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;KACzD;QACD;;;;cAIU,IAAA,yBAAc,EAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;KACzD;KACF,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACtB,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;QAC1B,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,aAAa,mBAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KAC3F,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,qBAAqB,CACnC,aAAiC,EACjC,GAAa,EACb,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,KAAK,GAAG;;;;YAIJ,IAAA,yBAAc,EAAC,gBAAgB,EAAE,GAAG,EAAE,QAAQ,CAAC;GACxD,CAAC;IACF,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;QAC1B,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,UAAU,mBAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KACxF,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,iCAAiC,CAC/C,aAAiC,EACjC,GAAa,EACb,aAAqB,EACrB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,gBAAgB,GAAG,aAAa,GAAG,CAAC,CAAC;IAC3C,iBAAiB;IACjB,MAAM,IAAI,GAAG;QACX;;;;gBAIY,IAAA,yBAAc,EAAC,cAAc,EAAE,GAAG,EAAE,QAAQ,CAAC;;;;;;;;;;;;;KAaxD;QACD;;;;;;KAMC;QACD;;;;;;;KAOC;QACD;;;;;;KAMC;QACD;;8CAE0C,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;;;;;;gBAO/E,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;QACD;;gDAE4C,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;gBAEjF,IAAA,yBAAc,EAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;;;;;gBAKhD,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;QACD;;;;;;;gBAOY,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,mBAAmB;;KAEtD;KACF,CAAC;IACF,MAAM,KAAK,GAAG;QACZ;;;eAGW,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc;KAChD;QACD;;KAEC;KACF,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAClB,gBAAgB;IAChB,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;QAChC,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,uBAAuB,mBAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KACrG,CAAC,CAAC;AACL,CAAC;AAED,SAAS,wBAAwB,CAAC,aAAqB,EAAE,QAAwB;IAC/E,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;IACrD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,YAAY,CAAC,KAI5B;IACC,KAAK,CAAC,CAAC,IAAA,8BAAmB,EAAC;QACzB,GAAG,KAAK;QACR,aAAa,EAAE,CAAC,GAAkB,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,EAAE,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC;KAC5F,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { Guid, Id64, Id64Arg, Id64Array } from \"@itwin/core-bentley\";\nimport {\n  ECSqlBinding,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n} from \"@itwin/presentation-shared\";\nimport { SelectableInstanceKey } from \"./Selectable.js\";\nimport { formIdBindings, genericExecuteQuery } from \"./Utils.js\";\n\n/**\n * Available selection scopes.\n * @public\n */\ntype SelectionScopeIdentifier = \"element\" | \"model\" | \"category\" | \"functional\";\n\n/**\n * Props for computing element selection.\n * @public\n */\nexport interface ElementSelectionScopeProps {\n  /** Identifies this as either \"element\" or \"functional\" selection scope */\n  id: \"element\" | \"functional\";\n  /**\n   * Specifies how far \"up\" we should walk to find the target element. When not specified or `0`,\n   * the target element matches the request element. When set to `1`, the target element matches the direct parent element.\n   * When set to `2`, the target element is parent of the parent element and so on. In all situations when this is `> 0`,\n   * we're not walking further than the last existing element, for example when `ancestorLevel = 1` (direct parent\n   * element is requested), but the request element doesn't have a parent, the request element is returned as the result.\n   * A negative value would result in the top-most element to be returned.\n   */\n  ancestorLevel?: number;\n}\n\n/**\n * A union of types that can be used to define a selection scope.\n * @public\n * @since 1.4.0\n */\nexport type SelectionScope = ElementSelectionScopeProps | { id: SelectionScopeIdentifier } | SelectionScopeIdentifier;\n\n/**\n * Props for `computeSelection`.\n * @public\n */\nexport interface ComputeSelectionProps {\n  /** iModel query executor. */\n  queryExecutor: ECSqlQueryExecutor;\n  /** IDs of elements to compute selection for. */\n  elementIds: Id64Arg;\n  /** Selection scope to compute selection with. */\n  scope: SelectionScope;\n  /** Optional unique id of the component that consumes this function. It can be any string, as long as it is not shared across different components. */\n  componentId?: string;\n}\n\n/**\n * Computes selection from given element IDs.\n * @public\n */\nexport async function* computeSelection(props: ComputeSelectionProps): AsyncIterableIterator<SelectableInstanceKey> {\n  const { queryExecutor, elementIds, scope } = props;\n  const componentId = props.componentId ?? Guid.createValue();\n  const componentName = \"computeSelection\";\n  if (typeof scope === \"string\") {\n    yield* computeSelection({ queryExecutor, elementIds, scope: { id: scope }, componentId });\n    return;\n  }\n\n  const nonTransientIds: Id64Array = [];\n  for (const id of Id64.iterable(elementIds)) {\n    if (!Id64.isTransient(id)) {\n      nonTransientIds.push(id);\n    }\n  }\n\n  switch (scope.id) {\n    case \"element\":\n      yield* computeElementSelection(\n        queryExecutor,\n        nonTransientIds,\n        (scope as ElementSelectionScopeProps).ancestorLevel ?? 0,\n        componentId,\n        componentName,\n      );\n      return;\n    case \"category\":\n      yield* computeCategorySelection(queryExecutor, nonTransientIds, componentId, componentName);\n      return;\n    case \"model\":\n      yield* computeModelSelection(queryExecutor, nonTransientIds, componentId, componentName);\n      return;\n    case \"functional\":\n      yield* computeFunctionalElementSelection(\n        queryExecutor,\n        nonTransientIds,\n        (scope as ElementSelectionScopeProps).ancestorLevel ?? 0,\n        componentId,\n        componentName,\n      );\n      return;\n  }\n}\n\nasync function* computeElementSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  elementIds: string[],\n  ancestorLevel: number,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const recurseUntilRoot = ancestorLevel < 0;\n  const ctes = [\n    `\n      AncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT ECInstanceId, ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, Parent.Id\n        FROM BisCore.Element\n        WHERE ${formIdBindings(\"ECInstanceId\", elementIds, bindings)}\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, a.Depth - 1, pe.Parent.Id\n        FROM AncestorElements a\n        JOIN BisCore.Element pe ON pe.ECInstanceId = a.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n  ];\n  const ecsql = `\n    SELECT DISTINCT ECInstanceId, ec_classname(ECClassId, 's.c') AS ClassName\n    FROM AncestorElements\n    WHERE ${recurseUntilRoot ? \"\" : \"Depth = 0 OR\"} ParentId IS NULL\n  `;\n  yield* executeQuery({\n    queryExecutor,\n    query: { ctes, ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/element/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeCategorySelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: string[],\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const ecsql = [\n    `\n      SELECT DISTINCT c.ECInstanceId, ec_classname(c.ECClassId, 's.c') AS ClassName\n      FROM BisCore.Category c\n      JOIN BisCore.GeometricElement2d ge ON ge.Category.Id = c.ECInstanceId\n      WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n    `,\n    `\n      SELECT DISTINCT c.ECInstanceId, ec_classname(c.ECClassId, 's.c') AS ClassName\n      FROM BisCore.Category c\n      JOIN BisCore.GeometricElement3d ge ON ge.Category.Id = c.ECInstanceId\n      WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n    `,\n  ].join(\" UNION ALL \");\n  yield* executeQuery({\n    queryExecutor,\n    query: { ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/category/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeModelSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: string[],\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const ecsql = `\n    SELECT DISTINCT m.ECInstanceId, ec_classname(m.ECClassId, 's.c') AS ClassName\n    FROM BisCore.Model m\n    JOIN BisCore.Element e ON e.Model.Id = m.ECInstanceId\n    WHERE ${formIdBindings(\"e.ECInstanceId\", ids, bindings)}\n  `;\n  yield* executeQuery({\n    queryExecutor,\n    query: { ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/model/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeFunctionalElementSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: string[],\n  ancestorLevel: number,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const recurseUntilRoot = ancestorLevel < 0;\n  // cspell:disable\n  const ctes = [\n    `\n      Elements2dOrNearestFunctionalElements(OriginalECInstanceId, OriginalECClassId, ECInstanceId, ECClassId, ParentId) AS (\n        SELECT ECInstanceId, ECClassId, ECInstanceId, ECClassId, Parent.Id\n        FROM BisCore.Element\n        WHERE ${formIdBindings(\"ECInstanceId\", ids, bindings)} AND ECClassId IS NOT (BisCore.GeometricElement3d)\n        UNION ALL\n        SELECT\n          e2onfe.OriginalECInstanceId,\n          e2onfe.OriginalECClassId,\n          COALESCE(dgrfe.TargetECInstanceId, pe.ECInstanceId),\n          COALESCE(dgrfe.TargetECClassId, pe.ECClassId),\n          pe.Parent.Id\n        FROM Elements2dOrNearestFunctionalElements e2onfe\n        LEFT JOIN BisCore.Element pe ON pe.ECInstanceId = e2onfe.ParentId\n        LEFT JOIN Functional.DrawingGraphicRepresentsFunctionalElement dgrfe ON dgrfe.SourceECInstanceId = e2onfe.ECInstanceId\n        WHERE e2onfe.ECClassId IS NOT (Functional.FunctionalElement) AND (e2onfe.ParentId IS NOT NULL OR dgrfe.TargetECInstanceId IS NOT NULL)\n      )\n    `,\n    `\n      Element2dNearestFunctionalElements(OriginalECInstanceId, ECInstanceId, ECClassId) AS (\n        SELECT OriginalECInstanceId, ECInstanceId, ECClassId\n        FROM Elements2dOrNearestFunctionalElements\n        WHERE ECClassId IS (Functional.FunctionalElement)\n      )\n    `,\n    `\n      Elements2dWithoutFunctionalElement(ECInstanceId, ECClassId) AS (\n        SELECT e2wfe.OriginalECInstanceId, OriginalECClassId\n        FROM Elements2dOrNearestFunctionalElements e2wfe\n        LEFT JOIN Element2dNearestFunctionalElements e2nfe ON e2nfe.OriginalECInstanceId = e2wfe.OriginalECInstanceId\n        WHERE e2wfe.ParentId IS NULL AND e2nfe.ECInstanceId IS NULL\n      )\n    `,\n    `\n      Elements2d(ECInstanceId, ECClassId) AS (\n        SELECT ECInstanceId, ECClassId FROM Element2dNearestFunctionalElements\n        UNION\n        SELECT ECInstanceId, ECClassId FROM Elements2dWithoutFunctionalElement\n      )\n    `,\n    `\n      Element2dAncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT e.ECInstanceId, e.ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, e.Parent.Id\n        FROM BisCore.Element e\n        JOIN Elements2d e2d ON e2d.ECInstanceId = e.ECInstanceId\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, e2ae.Depth - 1, pe.Parent.Id\n        FROM Element2dAncestorElements e2ae\n        JOIN BisCore.Element pe ON pe.ECInstanceId = e2ae.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n    `\n      Element3dAncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT ge.ECInstanceId, ge.ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, ge.Parent.Id\n        FROM BisCore.GeometricElement3d ge\n        WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, e3ae.Depth - 1, pe.Parent.Id\n        FROM Element3dAncestorElements e3ae\n        JOIN BisCore.Element pe ON pe.ECInstanceId = e3ae.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n    `\n      Element3dAncestorRelatedFunctionalElement(ECInstanceId, ClassName) AS (\n        SELECT\n          COALESCE(peff.TargetECInstanceId, e3ae.ECInstanceId),\n          ec_classname(COALESCE(peff.TargetECClassId, e3ae.ECClassId), 's.c')\n        FROM Element3dAncestorElements e3ae\n        LEFT JOIN Functional.PhysicalElementFulfillsFunction peff ON peff.SourceECInstanceId = e3ae.ECInstanceId\n        WHERE ${recurseUntilRoot ? \"\" : \"e3ae.Depth = 0 OR\"} e3ae.ParentId IS NULL\n      )\n    `,\n  ];\n  const ecsql = [\n    `\n      SELECT DISTINCT ECInstanceId, ec_classname(ECClassId, 's.c') AS ClassName\n        FROM Element2dAncestorElements\n       WHERE ${recurseUntilRoot ? \"\" : \"Depth = 0 OR\"} ParentId IS NULL\n    `,\n    `\n      SELECT DISTINCT ECInstanceId, ClassName FROM Element3dAncestorRelatedFunctionalElement\n    `,\n  ].join(\" UNION \");\n  // cspell:enable\n  yield* executeQuery({\n    queryExecutor,\n    query: { ctes, ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/functional-element/${Guid.createValue()}` },\n  });\n}\n\nfunction formAncestorLevelBinding(ancestorLevel: number, bindings: ECSqlBinding[]) {\n  bindings.push({ type: \"int\", value: ancestorLevel });\n  return \"?\";\n}\n\nasync function* executeQuery(props: {\n  queryExecutor: ECSqlQueryExecutor;\n  query: ECSqlQueryDef;\n  config?: ECSqlQueryReaderOptions;\n}): AsyncIterableIterator<SelectableInstanceKey> {\n  yield* genericExecuteQuery({\n    ...props,\n    parseQueryRow: (row: ECSqlQueryRow) => ({ className: row.ClassName, id: row.ECInstanceId }),\n  });\n}\n"]}
+{"version":3,"file":"SelectionScope.js","sourceRoot":"","sources":["../../../src/unified-selection/SelectionScope.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;AA+DhG,4CAyCC;AAtGD,sDAA8E;AAQ9E,mDAAqF;AACrF,yCAAiE;AAgDjE;;;GAGG;AACI,KAAK,SAAS,CAAC,CAAC,gBAAgB,CAAC,KAA4B;IAClE,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;IACnD,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,mBAAI,CAAC,WAAW,EAAE,CAAC;IAC5D,MAAM,aAAa,GAAG,kBAAkB,CAAC;IACzC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,KAAK,CAAC,CAAC,gBAAgB,CAAC,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;QAC1F,OAAO;IACT,CAAC;IAED,MAAM,EAAE,UAAU,EAAE,aAAa,EAAE,SAAS,EAAE,YAAY,EAAE,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;IAE3F,KAAK,MAAM,EAAE,IAAI,YAAY,EAAE,CAAC;QAC9B,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,2CAA2B,EAAE,CAAC;IACvD,CAAC;IAED,QAAQ,KAAK,CAAC,EAAE,EAAE,CAAC;QACjB,KAAK,SAAS;YACZ,KAAK,CAAC,CAAC,uBAAuB,CAC5B,aAAa,EACb,aAAa,EACZ,KAAoC,CAAC,aAAa,IAAI,CAAC,EACxD,WAAW,EACX,aAAa,CACd,CAAC;YACF,OAAO;QACT,KAAK,UAAU;YACb,KAAK,CAAC,CAAC,wBAAwB,CAAC,aAAa,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;YAC1F,OAAO;QACT,KAAK,OAAO;YACV,KAAK,CAAC,CAAC,qBAAqB,CAAC,aAAa,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;YACvF,OAAO;QACT,KAAK,YAAY;YACf,KAAK,CAAC,CAAC,iCAAiC,CACtC,aAAa,EACb,aAAa,EACZ,KAAoC,CAAC,aAAa,IAAI,CAAC,EACxD,WAAW,EACX,aAAa,CACd,CAAC;YACF,OAAO;IACX,CAAC;AACH,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,uBAAuB,CACrC,aAAiC,EACjC,GAAwB,EACxB,aAAqB,EACrB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,gBAAgB,GAAG,aAAa,GAAG,CAAC,CAAC;IAC3C,MAAM,IAAI,GAAG;QACX;;0CAEsC,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;gBAE3E,IAAA,yBAAc,EAAC,cAAc,EAAE,GAAG,EAAE,QAAQ,CAAC;;;;;gBAK7C,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;KACF,CAAC;IACF,MAAM,KAAK,GAAG;;;YAGJ,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc;GAC/C,CAAC;IACF,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;QAChC,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,YAAY,mBAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KAC1F,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,wBAAwB,CACtC,aAAiC,EACjC,GAAwB,EACxB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,KAAK,GAAG;QACZ;;;;cAIU,IAAA,yBAAc,EAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;KACzD;QACD;;;;cAIU,IAAA,yBAAc,EAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;KACzD;KACF,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACtB,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;QAC1B,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,aAAa,mBAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KAC3F,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,qBAAqB,CACnC,aAAiC,EACjC,GAAwB,EACxB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,KAAK,GAAG;;;;YAIJ,IAAA,yBAAc,EAAC,gBAAgB,EAAE,GAAG,EAAE,QAAQ,CAAC;GACxD,CAAC;IACF,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;QAC1B,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,UAAU,mBAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KACxF,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,iCAAiC,CAC/C,aAAiC,EACjC,GAAwB,EACxB,aAAqB,EACrB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,gBAAgB,GAAG,aAAa,GAAG,CAAC,CAAC;IAC3C,iBAAiB;IACjB,MAAM,IAAI,GAAG;QACX;;;;gBAIY,IAAA,yBAAc,EAAC,cAAc,EAAE,GAAG,EAAE,QAAQ,CAAC;;;;;;;;;;;;;KAaxD;QACD;;;;;;KAMC;QACD;;;;;;;KAOC;QACD;;;;;;KAMC;QACD;;8CAE0C,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;;;;;;gBAO/E,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;QACD;;gDAE4C,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;gBAEjF,IAAA,yBAAc,EAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;;;;;gBAKhD,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;QACD;;;;;;;gBAOY,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,mBAAmB;;KAEtD;KACF,CAAC;IACF,MAAM,KAAK,GAAG;QACZ;;;eAGW,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc;KAChD;QACD;;KAEC;KACF,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAClB,gBAAgB;IAChB,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;QAChC,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,uBAAuB,mBAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KACrG,CAAC,CAAC;AACL,CAAC;AAED,SAAS,wBAAwB,CAAC,aAAqB,EAAE,QAAwB;IAC/E,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;IACrD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,YAAY,CAAC,KAI5B;IACC,KAAK,CAAC,CAAC,IAAA,8BAAmB,EAAC;QACzB,GAAG,KAAK;QACR,aAAa,EAAE,CAAC,GAAkB,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,EAAE,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC;KAC5F,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,GAAY;IACnC,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;QAC5B,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;IACd,CAAC;IAED,oFAAoF;IACpF,IAAI,aAAa,GAAG,IAAI,CAAC;IACzB,IAAI,YAAY,GAAG,IAAI,CAAC;IACxB,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;QACrB,IAAI,mBAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YACzB,aAAa,GAAG,KAAK,CAAC;QACxB,CAAC;aAAM,CAAC;YACN,YAAY,GAAG,KAAK,CAAC;QACvB,CAAC;QAED,IAAI,CAAC,aAAa,IAAI,CAAC,YAAY,EAAE,CAAC;YACpC,MAAM;QACR,CAAC;IACH,CAAC;IAED,mEAAmE;IACnE,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;IAC5C,CAAC;SAAM,IAAI,YAAY,EAAE,CAAC;QACxB,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC;IAC5C,CAAC;IAED,qEAAqE;IACrE,wBAAwB;IACxB,MAAM,oBAAoB,GAAc,EAAE,CAAC;IAC3C,MAAM,mBAAmB,GAAc,EAAE,CAAC;IAC1C,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;QACrB,IAAI,mBAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YACzB,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,UAAU,EAAE,oBAAoB,EAAE,SAAS,EAAE,mBAAmB,EAAE,CAAC;AAC9E,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 { Guid, Id64, Id64Arg, Id64Array, Id64Set } from \"@itwin/core-bentley\";\nimport {\n  ECSqlBinding,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n} from \"@itwin/presentation-shared\";\nimport { SelectableInstanceKey, TRANSIENT_ELEMENT_CLASSNAME } from \"./Selectable.js\";\nimport { formIdBindings, genericExecuteQuery } from \"./Utils.js\";\n\n/**\n * Available selection scopes.\n * @public\n */\ntype SelectionScopeIdentifier = \"element\" | \"model\" | \"category\" | \"functional\";\n\n/**\n * Props for computing element selection.\n * @public\n */\nexport interface ElementSelectionScopeProps {\n  /** Identifies this as either \"element\" or \"functional\" selection scope */\n  id: \"element\" | \"functional\";\n  /**\n   * Specifies how far \"up\" we should walk to find the target element. When not specified or `0`,\n   * the target element matches the request element. When set to `1`, the target element matches the direct parent element.\n   * When set to `2`, the target element is parent of the parent element and so on. In all situations when this is `> 0`,\n   * we're not walking further than the last existing element, for example when `ancestorLevel = 1` (direct parent\n   * element is requested), but the request element doesn't have a parent, the request element is returned as the result.\n   * A negative value would result in the top-most element to be returned.\n   */\n  ancestorLevel?: number;\n}\n\n/**\n * A union of types that can be used to define a selection scope.\n * @public\n * @since 1.4.0\n */\nexport type SelectionScope = ElementSelectionScopeProps | { id: SelectionScopeIdentifier } | SelectionScopeIdentifier;\n\n/**\n * Props for `computeSelection`.\n * @public\n */\nexport interface ComputeSelectionProps {\n  /** iModel query executor. */\n  queryExecutor: ECSqlQueryExecutor;\n  /** IDs of elements to compute selection for. */\n  elementIds: Id64Arg;\n  /** Selection scope to compute selection with. */\n  scope: SelectionScope;\n  /** Optional unique id of the component that consumes this function. It can be any string, as long as it is not shared across different components. */\n  componentId?: string;\n}\n\n/**\n * Computes selection from given element IDs.\n * @public\n */\nexport async function* computeSelection(props: ComputeSelectionProps): AsyncIterableIterator<SelectableInstanceKey> {\n  const { queryExecutor, elementIds, scope } = props;\n  const componentId = props.componentId ?? Guid.createValue();\n  const componentName = \"computeSelection\";\n  if (typeof scope === \"string\") {\n    yield* computeSelection({ queryExecutor, elementIds, scope: { id: scope }, componentId });\n    return;\n  }\n\n  const { persistent: persistentIds, transient: transientIds } = parseElementIds(elementIds);\n\n  for (const id of transientIds) {\n    yield { id, className: TRANSIENT_ELEMENT_CLASSNAME };\n  }\n\n  switch (scope.id) {\n    case \"element\":\n      yield* computeElementSelection(\n        queryExecutor,\n        persistentIds,\n        (scope as ElementSelectionScopeProps).ancestorLevel ?? 0,\n        componentId,\n        componentName,\n      );\n      return;\n    case \"category\":\n      yield* computeCategorySelection(queryExecutor, persistentIds, componentId, componentName);\n      return;\n    case \"model\":\n      yield* computeModelSelection(queryExecutor, persistentIds, componentId, componentName);\n      return;\n    case \"functional\":\n      yield* computeFunctionalElementSelection(\n        queryExecutor,\n        persistentIds,\n        (scope as ElementSelectionScopeProps).ancestorLevel ?? 0,\n        componentId,\n        componentName,\n      );\n      return;\n  }\n}\n\nasync function* computeElementSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: Id64Array | Id64Set,\n  ancestorLevel: number,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const recurseUntilRoot = ancestorLevel < 0;\n  const ctes = [\n    `\n      AncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT ECInstanceId, ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, Parent.Id\n        FROM BisCore.Element\n        WHERE ${formIdBindings(\"ECInstanceId\", ids, bindings)}\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, a.Depth - 1, pe.Parent.Id\n        FROM AncestorElements a\n        JOIN BisCore.Element pe ON pe.ECInstanceId = a.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n  ];\n  const ecsql = `\n    SELECT DISTINCT ECInstanceId, ec_classname(ECClassId, 's.c') AS ClassName\n    FROM AncestorElements\n    WHERE ${recurseUntilRoot ? \"\" : \"Depth = 0 OR\"} ParentId IS NULL\n  `;\n  yield* executeQuery({\n    queryExecutor,\n    query: { ctes, ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/element/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeCategorySelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: Id64Array | Id64Set,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const ecsql = [\n    `\n      SELECT DISTINCT c.ECInstanceId, ec_classname(c.ECClassId, 's.c') AS ClassName\n      FROM BisCore.Category c\n      JOIN BisCore.GeometricElement2d ge ON ge.Category.Id = c.ECInstanceId\n      WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n    `,\n    `\n      SELECT DISTINCT c.ECInstanceId, ec_classname(c.ECClassId, 's.c') AS ClassName\n      FROM BisCore.Category c\n      JOIN BisCore.GeometricElement3d ge ON ge.Category.Id = c.ECInstanceId\n      WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n    `,\n  ].join(\" UNION ALL \");\n  yield* executeQuery({\n    queryExecutor,\n    query: { ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/category/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeModelSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: Id64Array | Id64Set,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const ecsql = `\n    SELECT DISTINCT m.ECInstanceId, ec_classname(m.ECClassId, 's.c') AS ClassName\n    FROM BisCore.Model m\n    JOIN BisCore.Element e ON e.Model.Id = m.ECInstanceId\n    WHERE ${formIdBindings(\"e.ECInstanceId\", ids, bindings)}\n  `;\n  yield* executeQuery({\n    queryExecutor,\n    query: { ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/model/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeFunctionalElementSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: Id64Array | Id64Set,\n  ancestorLevel: number,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const recurseUntilRoot = ancestorLevel < 0;\n  // cspell:disable\n  const ctes = [\n    `\n      Elements2dOrNearestFunctionalElements(OriginalECInstanceId, OriginalECClassId, ECInstanceId, ECClassId, ParentId) AS (\n        SELECT ECInstanceId, ECClassId, ECInstanceId, ECClassId, Parent.Id\n        FROM BisCore.Element\n        WHERE ${formIdBindings(\"ECInstanceId\", ids, bindings)} AND ECClassId IS NOT (BisCore.GeometricElement3d)\n        UNION ALL\n        SELECT\n          e2onfe.OriginalECInstanceId,\n          e2onfe.OriginalECClassId,\n          COALESCE(dgrfe.TargetECInstanceId, pe.ECInstanceId),\n          COALESCE(dgrfe.TargetECClassId, pe.ECClassId),\n          pe.Parent.Id\n        FROM Elements2dOrNearestFunctionalElements e2onfe\n        LEFT JOIN BisCore.Element pe ON pe.ECInstanceId = e2onfe.ParentId\n        LEFT JOIN Functional.DrawingGraphicRepresentsFunctionalElement dgrfe ON dgrfe.SourceECInstanceId = e2onfe.ECInstanceId\n        WHERE e2onfe.ECClassId IS NOT (Functional.FunctionalElement) AND (e2onfe.ParentId IS NOT NULL OR dgrfe.TargetECInstanceId IS NOT NULL)\n      )\n    `,\n    `\n      Element2dNearestFunctionalElements(OriginalECInstanceId, ECInstanceId, ECClassId) AS (\n        SELECT OriginalECInstanceId, ECInstanceId, ECClassId\n        FROM Elements2dOrNearestFunctionalElements\n        WHERE ECClassId IS (Functional.FunctionalElement)\n      )\n    `,\n    `\n      Elements2dWithoutFunctionalElement(ECInstanceId, ECClassId) AS (\n        SELECT e2wfe.OriginalECInstanceId, OriginalECClassId\n        FROM Elements2dOrNearestFunctionalElements e2wfe\n        LEFT JOIN Element2dNearestFunctionalElements e2nfe ON e2nfe.OriginalECInstanceId = e2wfe.OriginalECInstanceId\n        WHERE e2wfe.ParentId IS NULL AND e2nfe.ECInstanceId IS NULL\n      )\n    `,\n    `\n      Elements2d(ECInstanceId, ECClassId) AS (\n        SELECT ECInstanceId, ECClassId FROM Element2dNearestFunctionalElements\n        UNION\n        SELECT ECInstanceId, ECClassId FROM Elements2dWithoutFunctionalElement\n      )\n    `,\n    `\n      Element2dAncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT e.ECInstanceId, e.ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, e.Parent.Id\n        FROM BisCore.Element e\n        JOIN Elements2d e2d ON e2d.ECInstanceId = e.ECInstanceId\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, e2ae.Depth - 1, pe.Parent.Id\n        FROM Element2dAncestorElements e2ae\n        JOIN BisCore.Element pe ON pe.ECInstanceId = e2ae.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n    `\n      Element3dAncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT ge.ECInstanceId, ge.ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, ge.Parent.Id\n        FROM BisCore.GeometricElement3d ge\n        WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, e3ae.Depth - 1, pe.Parent.Id\n        FROM Element3dAncestorElements e3ae\n        JOIN BisCore.Element pe ON pe.ECInstanceId = e3ae.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n    `\n      Element3dAncestorRelatedFunctionalElement(ECInstanceId, ClassName) AS (\n        SELECT\n          COALESCE(peff.TargetECInstanceId, e3ae.ECInstanceId),\n          ec_classname(COALESCE(peff.TargetECClassId, e3ae.ECClassId), 's.c')\n        FROM Element3dAncestorElements e3ae\n        LEFT JOIN Functional.PhysicalElementFulfillsFunction peff ON peff.SourceECInstanceId = e3ae.ECInstanceId\n        WHERE ${recurseUntilRoot ? \"\" : \"e3ae.Depth = 0 OR\"} e3ae.ParentId IS NULL\n      )\n    `,\n  ];\n  const ecsql = [\n    `\n      SELECT DISTINCT ECInstanceId, ec_classname(ECClassId, 's.c') AS ClassName\n        FROM Element2dAncestorElements\n       WHERE ${recurseUntilRoot ? \"\" : \"Depth = 0 OR\"} ParentId IS NULL\n    `,\n    `\n      SELECT DISTINCT ECInstanceId, ClassName FROM Element3dAncestorRelatedFunctionalElement\n    `,\n  ].join(\" UNION \");\n  // cspell:enable\n  yield* executeQuery({\n    queryExecutor,\n    query: { ctes, ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/functional-element/${Guid.createValue()}` },\n  });\n}\n\nfunction formAncestorLevelBinding(ancestorLevel: number, bindings: ECSqlBinding[]) {\n  bindings.push({ type: \"int\", value: ancestorLevel });\n  return \"?\";\n}\n\nasync function* executeQuery(props: {\n  queryExecutor: ECSqlQueryExecutor;\n  query: ECSqlQueryDef;\n  config?: ECSqlQueryReaderOptions;\n}): AsyncIterableIterator<SelectableInstanceKey> {\n  yield* genericExecuteQuery({\n    ...props,\n    parseQueryRow: (row: ECSqlQueryRow) => ({ className: row.ClassName, id: row.ECInstanceId }),\n  });\n}\n\nfunction parseElementIds(ids: Id64Arg): { persistent: Id64Set | Id64Array; transient: Id64Set | Id64Array } {\n  if (typeof ids === \"string\") {\n    ids = [ids];\n  }\n\n  // detect if all ids are either transient or persistent to avoid unnecessary copying\n  let allPersistent = true;\n  let allTransient = true;\n  for (const id of ids) {\n    if (Id64.isTransient(id)) {\n      allPersistent = false;\n    } else {\n      allTransient = false;\n    }\n\n    if (!allPersistent && !allTransient) {\n      break;\n    }\n  }\n\n  // avoid making a copy if ids are only persistent or only transient\n  if (allPersistent) {\n    return { persistent: ids, transient: [] };\n  } else if (allTransient) {\n    return { persistent: [], transient: ids };\n  }\n\n  // if `ids` contain mixed ids, we have to copy.. use Array instead of\n  // a Set for performance\n  const persistentElementIds: Id64Array = [];\n  const transientElementIds: Id64Array = [];\n  for (const id of ids) {\n    if (Id64.isTransient(id)) {\n      transientElementIds.push(id);\n    } else {\n      persistentElementIds.push(id);\n    }\n  }\n\n  return { persistent: persistentElementIds, transient: transientElementIds };\n}\n"]}

package/lib/cjs/unified-selection/Utils.d.ts

@@ -1,11 +1,12 @@
 import "./DisposePolyfill.js";
 import { Observable } from "rxjs";
+import { Id64Array, Id64Set } from "@itwin/core-bentley";
 import { ECSqlBinding, ECSqlQueryDef, ECSqlQueryExecutor, ECSqlQueryReaderOptions, ECSqlQueryRow } from "@itwin/presentation-shared";
 /**
  * Forms ECSql bindings from given ID's.
  * @internal
  */
-export declare function formIdBindings(property: string, ids: string[], bindings: ECSqlBinding[]): string;
+export declare function formIdBindings(property: string, ids: Id64Array | Id64Set, bindings: ECSqlBinding[]): string;
 /**
  * Executes given ECSql query and extracts data from rows. Additionally handles main thread releasing.
  * @internal

package/lib/cjs/unified-selection/Utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Utils.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/Utils.ts"],"names":[],"mappings":"AAKA,OAAO,sBAAsB,CAAC;AAE9B,OAAO,EAA4C,UAAU,EAAM,MAAM,MAAM,CAAC;AAChF,OAAO,EAEL,YAAY,EACZ,aAAa,EACb,kBAAkB,EAClB,uBAAuB,EACvB,aAAa,EACd,MAAM,4BAA4B,CAAC;AAEpC;;;GAGG;AACH,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,MAAM,CAYhG;AAED;;;GAGG;AACH,wBAAuB,mBAAmB,CAAC,CAAC,EAAE,EAC5C,aAAa,EACb,aAAa,EACb,KAAK,EACL,MAAM,GACP,EAAE;IACD,aAAa,EAAE,kBAAkB,CAAC;IAClC,KAAK,EAAE,aAAa,CAAC;IACrB,aAAa,EAAE,CAAC,GAAG,EAAE,aAAa,KAAK,CAAC,CAAC;IACzC,MAAM,CAAC,EAAE,uBAAuB,CAAC;CAClC,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAO3B;AAED;;;GAGG;AACH,wBAAgB,6BAA6B,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,IAC3D,KAAK,UAAU,CAAC,CAAC,CAAC,KAAG,UAAU,CAAC,CAAC,CAAC,CAa3C;AAED;;;;;;;;GAQG;AACH,wBAAgB,WAAW,CAAC,UAAU,EAAE,EAAE,GAAG;IAAE,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,IAAI,CAAA;CAAE,GAAG;IAAE,OAAO,EAAE,MAAM,IAAI,CAAA;CAAE,QAMtG"}
+{"version":3,"file":"Utils.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/Utils.ts"],"names":[],"mappings":"AAKA,OAAO,sBAAsB,CAAC;AAE9B,OAAO,EAA4C,UAAU,EAAM,MAAM,MAAM,CAAC;AAChF,OAAO,EAAQ,SAAS,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,EAEL,YAAY,EACZ,aAAa,EACb,kBAAkB,EAClB,uBAAuB,EACvB,aAAa,EACd,MAAM,4BAA4B,CAAC;AAEpC;;;GAGG;AACH,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,MAAM,CAoB3G;AAED;;;GAGG;AACH,wBAAuB,mBAAmB,CAAC,CAAC,EAAE,EAC5C,aAAa,EACb,aAAa,EACb,KAAK,EACL,MAAM,GACP,EAAE;IACD,aAAa,EAAE,kBAAkB,CAAC;IAClC,KAAK,EAAE,aAAa,CAAC;IACrB,aAAa,EAAE,CAAC,GAAG,EAAE,aAAa,KAAK,CAAC,CAAC;IACzC,MAAM,CAAC,EAAE,uBAAuB,CAAC;CAClC,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAO3B;AAED;;;GAGG;AACH,wBAAgB,6BAA6B,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,IAC3D,KAAK,UAAU,CAAC,CAAC,CAAC,KAAG,UAAU,CAAC,CAAC,CAAC,CAa3C;AAED;;;;;;;;GAQG;AACH,wBAAgB,WAAW,CAAC,UAAU,EAAE,EAAE,GAAG;IAAE,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,IAAI,CAAA;CAAE,GAAG;IAAE,OAAO,EAAE,MAAM,IAAI,CAAA;CAAE,QAMtG"}

package/lib/cjs/unified-selection/Utils.js

@@ -10,21 +10,30 @@ exports.releaseMainThreadOnItemsCount = releaseMainThreadOnItemsCount;
 exports.safeDispose = safeDispose;
 require("./DisposePolyfill.js");
 const rxjs_1 = require("rxjs");
+const core_bentley_1 = require("@itwin/core-bentley");
 const presentation_shared_1 = require("@itwin/presentation-shared");
 /**
  * Forms ECSql bindings from given ID's.
  * @internal
  */
 function formIdBindings(property, ids, bindings) {
-    if (ids.length > 1000) {
-        bindings.push({ type: "idset", value: ids });
+    const count = core_bentley_1.Id64.sizeOf(ids);
+    if (count > 1000) {
+        bindings.push({ type: "idset", value: ids instanceof Set ? [...ids] : ids });
         return `InVirtualSet(?, ${property})`;
     }
-    if (ids.length === 0) {
+    if (count === 0) {
         return `FALSE`;
     }
-    ids.forEach((id) => bindings.push({ type: "id", value: id }));
-    return `${property} IN (${ids.map(() => "?").join(",")})`;
+    let bindingsClause = "";
+    ids.forEach((id) => {
+        bindings.push({ type: "id", value: id });
+        if (bindingsClause.length > 0) {
+            bindingsClause += ",";
+        }
+        bindingsClause += "?";
+    });
+    return `${property} IN (${bindingsClause})`;
 }
 /**
  * Executes given ECSql query and extracts data from rows. Additionally handles main thread releasing.

package/lib/cjs/unified-selection/Utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"Utils.js","sourceRoot":"","sources":["../../../src/unified-selection/Utils.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;AAkBhG,wCAYC;AAMD,kDAiBC;AAMD,sEAcC;AAWD,kCAMC;AAxFD,gCAA8B;AAE9B,+BAAgF;AAChF,oEAOoC;AAEpC;;;GAGG;AACH,SAAgB,cAAc,CAAC,QAAgB,EAAE,GAAa,EAAE,QAAwB;IACtF,IAAI,GAAG,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC;QACtB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QAC7C,OAAO,mBAAmB,QAAQ,GAAG,CAAC;IACxC,CAAC;IAED,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IAC9D,OAAO,GAAG,QAAQ,QAAQ,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;AAC5D,CAAC;AAED;;;GAGG;AACI,KAAK,SAAS,CAAC,CAAC,mBAAmB,CAAI,EAC5C,aAAa,EACb,aAAa,EACb,KAAK,EACL,MAAM,GAMP;IACC,MAAM,iBAAiB,GAAG,IAAA,gEAA0C,GAAE,CAAC;IACvE,MAAM,MAAM,GAAG,aAAa,CAAC,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC9D,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC/B,MAAM,aAAa,CAAC,GAAG,CAAC,CAAC;QACzB,MAAM,iBAAiB,EAAE,CAAC;IAC5B,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAgB,6BAA6B,CAAI,YAAoB;IACnE,OAAO,CAAC,GAAkB,EAAiB,EAAE;QAC3C,OAAO,GAAG,CAAC,IAAI,CACb,IAAA,kBAAW,EAAC,YAAY,CAAC,EACzB,IAAA,gBAAS,EAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;YACpB,MAAM,GAAG,GAAG,IAAA,SAAE,EAAC,IAAI,CAAC,CAAC;YACrB,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,GAAG,YAAY,EAAE,CAAC;gBAC1C,OAAO,GAAG,CAAC;YACb,CAAC;YACD,OAAO,GAAG,CAAC,IAAI,CAAC,IAAA,YAAK,EAAC,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC,CAAC,EACF,IAAA,gBAAS,GAAE,CACZ,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,SAAgB,WAAW,CAAC,UAA2E;IACrG,IAAI,SAAS,IAAI,UAAU,EAAE,CAAC;QAC5B,UAAU,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC;SAAM,IAAI,MAAM,CAAC,OAAO,IAAI,UAAU,EAAE,CAAC;QACxC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC/B,CAAC;AACH,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 \"./DisposePolyfill.js\";\n\nimport { bufferCount, concatAll, concatMap, delay, Observable, of } from \"rxjs\";\nimport {\n  createMainThreadReleaseOnTimePassedHandler,\n  ECSqlBinding,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n} from \"@itwin/presentation-shared\";\n\n/**\n * Forms ECSql bindings from given ID's.\n * @internal\n */\nexport function formIdBindings(property: string, ids: string[], bindings: ECSqlBinding[]): string {\n  if (ids.length > 1000) {\n    bindings.push({ type: \"idset\", value: ids });\n    return `InVirtualSet(?, ${property})`;\n  }\n\n  if (ids.length === 0) {\n    return `FALSE`;\n  }\n\n  ids.forEach((id) => bindings.push({ type: \"id\", value: id }));\n  return `${property} IN (${ids.map(() => \"?\").join(\",\")})`;\n}\n\n/**\n * Executes given ECSql query and extracts data from rows. Additionally handles main thread releasing.\n * @internal\n */\nexport async function* genericExecuteQuery<T>({\n  queryExecutor,\n  parseQueryRow,\n  query,\n  config,\n}: {\n  queryExecutor: ECSqlQueryExecutor;\n  query: ECSqlQueryDef;\n  parseQueryRow: (row: ECSqlQueryRow) => T;\n  config?: ECSqlQueryReaderOptions;\n}): AsyncIterableIterator<T> {\n  const releaseMainThread = createMainThreadReleaseOnTimePassedHandler();\n  const reader = queryExecutor.createQueryReader(query, config);\n  for await (const row of reader) {\n    yield parseQueryRow(row);\n    await releaseMainThread();\n  }\n}\n\n/**\n * Emits a certain amount of values, then releases the main thread for other timers to use.\n * @internal\n */\nexport function releaseMainThreadOnItemsCount<T>(elementCount: number) {\n  return (obs: Observable<T>): Observable<T> => {\n    return obs.pipe(\n      bufferCount(elementCount),\n      concatMap((buff, i) => {\n        const out = of(buff);\n        if (i === 0 && buff.length < elementCount) {\n          return out;\n        }\n        return out.pipe(delay(0));\n      }),\n      concatAll(),\n    );\n  };\n}\n\n/**\n * A helper that disposes the given object, if it's disposable.\n *\n * The first option is to dispose using the deprecated `dispose` method if it exists on the object.\n * If not, we use the new `Symbol.dispose` method. If that doesn't exist either, the object is\n * considered as non-disposable and nothing is done with it.\n *\n * @internal\n */\nexport function safeDispose(disposable: {} | { [Symbol.dispose]: () => void } | { dispose: () => void }) {\n  if (\"dispose\" in disposable) {\n    disposable.dispose();\n  } else if (Symbol.dispose in disposable) {\n    disposable[Symbol.dispose]();\n  }\n}\n"]}
+{"version":3,"file":"Utils.js","sourceRoot":"","sources":["../../../src/unified-selection/Utils.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;AAmBhG,wCAoBC;AAMD,kDAiBC;AAMD,sEAcC;AAWD,kCAMC;AAjGD,gCAA8B;AAE9B,+BAAgF;AAChF,sDAA+D;AAC/D,oEAOoC;AAEpC;;;GAGG;AACH,SAAgB,cAAc,CAAC,QAAgB,EAAE,GAAwB,EAAE,QAAwB;IACjG,MAAM,KAAK,GAAG,mBAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC/B,IAAI,KAAK,GAAG,IAAI,EAAE,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;QAC7E,OAAO,mBAAmB,QAAQ,GAAG,CAAC;IACxC,CAAC;IAED,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;QAChB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,cAAc,GAAG,EAAE,CAAC;IACxB,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;QACjB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACzC,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,cAAc,IAAI,GAAG,CAAC;QACxB,CAAC;QACD,cAAc,IAAI,GAAG,CAAC;IACxB,CAAC,CAAC,CAAC;IACH,OAAO,GAAG,QAAQ,QAAQ,cAAc,GAAG,CAAC;AAC9C,CAAC;AAED;;;GAGG;AACI,KAAK,SAAS,CAAC,CAAC,mBAAmB,CAAI,EAC5C,aAAa,EACb,aAAa,EACb,KAAK,EACL,MAAM,GAMP;IACC,MAAM,iBAAiB,GAAG,IAAA,gEAA0C,GAAE,CAAC;IACvE,MAAM,MAAM,GAAG,aAAa,CAAC,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC9D,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC/B,MAAM,aAAa,CAAC,GAAG,CAAC,CAAC;QACzB,MAAM,iBAAiB,EAAE,CAAC;IAC5B,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAgB,6BAA6B,CAAI,YAAoB;IACnE,OAAO,CAAC,GAAkB,EAAiB,EAAE;QAC3C,OAAO,GAAG,CAAC,IAAI,CACb,IAAA,kBAAW,EAAC,YAAY,CAAC,EACzB,IAAA,gBAAS,EAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;YACpB,MAAM,GAAG,GAAG,IAAA,SAAE,EAAC,IAAI,CAAC,CAAC;YACrB,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,GAAG,YAAY,EAAE,CAAC;gBAC1C,OAAO,GAAG,CAAC;YACb,CAAC;YACD,OAAO,GAAG,CAAC,IAAI,CAAC,IAAA,YAAK,EAAC,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC,CAAC,EACF,IAAA,gBAAS,GAAE,CACZ,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,SAAgB,WAAW,CAAC,UAA2E;IACrG,IAAI,SAAS,IAAI,UAAU,EAAE,CAAC;QAC5B,UAAU,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC;SAAM,IAAI,MAAM,CAAC,OAAO,IAAI,UAAU,EAAE,CAAC;QACxC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC/B,CAAC;AACH,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 \"./DisposePolyfill.js\";\n\nimport { bufferCount, concatAll, concatMap, delay, Observable, of } from \"rxjs\";\nimport { Id64, Id64Array, Id64Set } from \"@itwin/core-bentley\";\nimport {\n  createMainThreadReleaseOnTimePassedHandler,\n  ECSqlBinding,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n} from \"@itwin/presentation-shared\";\n\n/**\n * Forms ECSql bindings from given ID's.\n * @internal\n */\nexport function formIdBindings(property: string, ids: Id64Array | Id64Set, bindings: ECSqlBinding[]): string {\n  const count = Id64.sizeOf(ids);\n  if (count > 1000) {\n    bindings.push({ type: \"idset\", value: ids instanceof Set ? [...ids] : ids });\n    return `InVirtualSet(?, ${property})`;\n  }\n\n  if (count === 0) {\n    return `FALSE`;\n  }\n\n  let bindingsClause = \"\";\n  ids.forEach((id) => {\n    bindings.push({ type: \"id\", value: id });\n    if (bindingsClause.length > 0) {\n      bindingsClause += \",\";\n    }\n    bindingsClause += \"?\";\n  });\n  return `${property} IN (${bindingsClause})`;\n}\n\n/**\n * Executes given ECSql query and extracts data from rows. Additionally handles main thread releasing.\n * @internal\n */\nexport async function* genericExecuteQuery<T>({\n  queryExecutor,\n  parseQueryRow,\n  query,\n  config,\n}: {\n  queryExecutor: ECSqlQueryExecutor;\n  query: ECSqlQueryDef;\n  parseQueryRow: (row: ECSqlQueryRow) => T;\n  config?: ECSqlQueryReaderOptions;\n}): AsyncIterableIterator<T> {\n  const releaseMainThread = createMainThreadReleaseOnTimePassedHandler();\n  const reader = queryExecutor.createQueryReader(query, config);\n  for await (const row of reader) {\n    yield parseQueryRow(row);\n    await releaseMainThread();\n  }\n}\n\n/**\n * Emits a certain amount of values, then releases the main thread for other timers to use.\n * @internal\n */\nexport function releaseMainThreadOnItemsCount<T>(elementCount: number) {\n  return (obs: Observable<T>): Observable<T> => {\n    return obs.pipe(\n      bufferCount(elementCount),\n      concatMap((buff, i) => {\n        const out = of(buff);\n        if (i === 0 && buff.length < elementCount) {\n          return out;\n        }\n        return out.pipe(delay(0));\n      }),\n      concatAll(),\n    );\n  };\n}\n\n/**\n * A helper that disposes the given object, if it's disposable.\n *\n * The first option is to dispose using the deprecated `dispose` method if it exists on the object.\n * If not, we use the new `Symbol.dispose` method. If that doesn't exist either, the object is\n * considered as non-disposable and nothing is done with it.\n *\n * @internal\n */\nexport function safeDispose(disposable: {} | { [Symbol.dispose]: () => void } | { dispose: () => void }) {\n  if (\"dispose\" in disposable) {\n    disposable.dispose();\n  } else if (Symbol.dispose in disposable) {\n    disposable[Symbol.dispose]();\n  }\n}\n"]}

package/lib/esm/unified-selection/SelectionScope.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionScope.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/SelectionScope.ts"],"names":[],"mappings":"AAKA,OAAO,EAAc,OAAO,EAAa,MAAM,qBAAqB,CAAC;AACrE,OAAO,EAGL,kBAAkB,EAGnB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAGxD;;;GAGG;AACH,KAAK,wBAAwB,GAAG,SAAS,GAAG,OAAO,GAAG,UAAU,GAAG,YAAY,CAAC;AAEhF;;;GAGG;AACH,MAAM,WAAW,0BAA0B;IACzC,0EAA0E;IAC1E,EAAE,EAAE,SAAS,GAAG,YAAY,CAAC;IAC7B;;;;;;;OAOG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,0BAA0B,GAAG;IAAE,EAAE,EAAE,wBAAwB,CAAA;CAAE,GAAG,wBAAwB,CAAC;AAEtH;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,6BAA6B;IAC7B,aAAa,EAAE,kBAAkB,CAAC;IAClC,gDAAgD;IAChD,UAAU,EAAE,OAAO,CAAC;IACpB,iDAAiD;IACjD,KAAK,EAAE,cAAc,CAAC;IACtB,sJAAsJ;IACtJ,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;;GAGG;AACH,wBAAuB,gBAAgB,CAAC,KAAK,EAAE,qBAAqB,GAAG,qBAAqB,CAAC,qBAAqB,CAAC,CA0ClH"}
+{"version":3,"file":"SelectionScope.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/SelectionScope.ts"],"names":[],"mappings":"AAKA,OAAO,EAAc,OAAO,EAAsB,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EAGL,kBAAkB,EAGnB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,qBAAqB,EAA+B,MAAM,iBAAiB,CAAC;AAGrF;;;GAGG;AACH,KAAK,wBAAwB,GAAG,SAAS,GAAG,OAAO,GAAG,UAAU,GAAG,YAAY,CAAC;AAEhF;;;GAGG;AACH,MAAM,WAAW,0BAA0B;IACzC,0EAA0E;IAC1E,EAAE,EAAE,SAAS,GAAG,YAAY,CAAC;IAC7B;;;;;;;OAOG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,0BAA0B,GAAG;IAAE,EAAE,EAAE,wBAAwB,CAAA;CAAE,GAAG,wBAAwB,CAAC;AAEtH;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,6BAA6B;IAC7B,aAAa,EAAE,kBAAkB,CAAC;IAClC,gDAAgD;IAChD,UAAU,EAAE,OAAO,CAAC;IACpB,iDAAiD;IACjD,KAAK,EAAE,cAAc,CAAC;IACtB,sJAAsJ;IACtJ,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;;GAGG;AACH,wBAAuB,gBAAgB,CAAC,KAAK,EAAE,qBAAqB,GAAG,qBAAqB,CAAC,qBAAqB,CAAC,CAyClH"}

package/lib/esm/unified-selection/SelectionScope.js

@@ -3,6 +3,7 @@
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
 import { Guid, Id64 } from "@itwin/core-bentley";
+import { TRANSIENT_ELEMENT_CLASSNAME } from "./Selectable.js";
 import { formIdBindings, genericExecuteQuery } from "./Utils.js";
 /**
  * Computes selection from given element IDs.
@@ -16,28 +17,26 @@ export async function* computeSelection(props) {
         yield* computeSelection({ queryExecutor, elementIds, scope: { id: scope }, componentId });
         return;
     }
-    const nonTransientIds = [];
-    for (const id of Id64.iterable(elementIds)) {
-        if (!Id64.isTransient(id)) {
-            nonTransientIds.push(id);
-        }
+    const { persistent: persistentIds, transient: transientIds } = parseElementIds(elementIds);
+    for (const id of transientIds) {
+        yield { id, className: TRANSIENT_ELEMENT_CLASSNAME };
     }
     switch (scope.id) {
         case "element":
-            yield* computeElementSelection(queryExecutor, nonTransientIds, scope.ancestorLevel ?? 0, componentId, componentName);
+            yield* computeElementSelection(queryExecutor, persistentIds, scope.ancestorLevel ?? 0, componentId, componentName);
             return;
         case "category":
-            yield* computeCategorySelection(queryExecutor, nonTransientIds, componentId, componentName);
+            yield* computeCategorySelection(queryExecutor, persistentIds, componentId, componentName);
             return;
         case "model":
-            yield* computeModelSelection(queryExecutor, nonTransientIds, componentId, componentName);
+            yield* computeModelSelection(queryExecutor, persistentIds, componentId, componentName);
             return;
         case "functional":
-            yield* computeFunctionalElementSelection(queryExecutor, nonTransientIds, scope.ancestorLevel ?? 0, componentId, componentName);
+            yield* computeFunctionalElementSelection(queryExecutor, persistentIds, scope.ancestorLevel ?? 0, componentId, componentName);
             return;
     }
 }
-async function* computeElementSelection(queryExecutor, elementIds, ancestorLevel, componentId, componentName) {
+async function* computeElementSelection(queryExecutor, ids, ancestorLevel, componentId, componentName) {
     const bindings = [];
     const recurseUntilRoot = ancestorLevel < 0;
     const ctes = [
@@ -45,7 +44,7 @@ async function* computeElementSelection(queryExecutor, elementIds, ancestorLevel
       AncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (
         SELECT ECInstanceId, ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, Parent.Id
         FROM BisCore.Element
-        WHERE ${formIdBindings("ECInstanceId", elementIds, bindings)}
+        WHERE ${formIdBindings("ECInstanceId", ids, bindings)}
         UNION ALL
         SELECT pe.ECInstanceId, pe.ECClassId, a.Depth - 1, pe.Parent.Id
         FROM AncestorElements a
@@ -208,4 +207,43 @@ async function* executeQuery(props) {
         parseQueryRow: (row) => ({ className: row.ClassName, id: row.ECInstanceId }),
     });
 }
+function parseElementIds(ids) {
+    if (typeof ids === "string") {
+        ids = [ids];
+    }
+    // detect if all ids are either transient or persistent to avoid unnecessary copying
+    let allPersistent = true;
+    let allTransient = true;
+    for (const id of ids) {
+        if (Id64.isTransient(id)) {
+            allPersistent = false;
+        }
+        else {
+            allTransient = false;
+        }
+        if (!allPersistent && !allTransient) {
+            break;
+        }
+    }
+    // avoid making a copy if ids are only persistent or only transient
+    if (allPersistent) {
+        return { persistent: ids, transient: [] };
+    }
+    else if (allTransient) {
+        return { persistent: [], transient: ids };
+    }
+    // if `ids` contain mixed ids, we have to copy.. use Array instead of
+    // a Set for performance
+    const persistentElementIds = [];
+    const transientElementIds = [];
+    for (const id of ids) {
+        if (Id64.isTransient(id)) {
+            transientElementIds.push(id);
+        }
+        else {
+            persistentElementIds.push(id);
+        }
+    }
+    return { persistent: persistentElementIds, transient: transientElementIds };
+}
 //# sourceMappingURL=SelectionScope.js.map

package/lib/esm/unified-selection/SelectionScope.js.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionScope.js","sourceRoot":"","sources":["../../../src/unified-selection/SelectionScope.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,IAAI,EAAE,IAAI,EAAsB,MAAM,qBAAqB,CAAC;AASrE,OAAO,EAAE,cAAc,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAgDjE;;;GAGG;AACH,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,gBAAgB,CAAC,KAA4B;IAClE,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;IACnD,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;IAC5D,MAAM,aAAa,GAAG,kBAAkB,CAAC;IACzC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,KAAK,CAAC,CAAC,gBAAgB,CAAC,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;QAC1F,OAAO;IACT,CAAC;IAED,MAAM,eAAe,GAAc,EAAE,CAAC;IACtC,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YAC1B,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,QAAQ,KAAK,CAAC,EAAE,EAAE,CAAC;QACjB,KAAK,SAAS;YACZ,KAAK,CAAC,CAAC,uBAAuB,CAC5B,aAAa,EACb,eAAe,EACd,KAAoC,CAAC,aAAa,IAAI,CAAC,EACxD,WAAW,EACX,aAAa,CACd,CAAC;YACF,OAAO;QACT,KAAK,UAAU;YACb,KAAK,CAAC,CAAC,wBAAwB,CAAC,aAAa,EAAE,eAAe,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;YAC5F,OAAO;QACT,KAAK,OAAO;YACV,KAAK,CAAC,CAAC,qBAAqB,CAAC,aAAa,EAAE,eAAe,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;YACzF,OAAO;QACT,KAAK,YAAY;YACf,KAAK,CAAC,CAAC,iCAAiC,CACtC,aAAa,EACb,eAAe,EACd,KAAoC,CAAC,aAAa,IAAI,CAAC,EACxD,WAAW,EACX,aAAa,CACd,CAAC;YACF,OAAO;IACX,CAAC;AACH,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,uBAAuB,CACrC,aAAiC,EACjC,UAAoB,EACpB,aAAqB,EACrB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,gBAAgB,GAAG,aAAa,GAAG,CAAC,CAAC;IAC3C,MAAM,IAAI,GAAG;QACX;;0CAEsC,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;gBAE3E,cAAc,CAAC,cAAc,EAAE,UAAU,EAAE,QAAQ,CAAC;;;;;gBAKpD,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;KACF,CAAC;IACF,MAAM,KAAK,GAAG;;;YAGJ,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc;GAC/C,CAAC;IACF,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;QAChC,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,YAAY,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KAC1F,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,wBAAwB,CACtC,aAAiC,EACjC,GAAa,EACb,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,KAAK,GAAG;QACZ;;;;cAIU,cAAc,CAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;KACzD;QACD;;;;cAIU,cAAc,CAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;KACzD;KACF,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACtB,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;QAC1B,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,aAAa,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KAC3F,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,qBAAqB,CACnC,aAAiC,EACjC,GAAa,EACb,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,KAAK,GAAG;;;;YAIJ,cAAc,CAAC,gBAAgB,EAAE,GAAG,EAAE,QAAQ,CAAC;GACxD,CAAC;IACF,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;QAC1B,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,UAAU,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KACxF,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,iCAAiC,CAC/C,aAAiC,EACjC,GAAa,EACb,aAAqB,EACrB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,gBAAgB,GAAG,aAAa,GAAG,CAAC,CAAC;IAC3C,iBAAiB;IACjB,MAAM,IAAI,GAAG;QACX;;;;gBAIY,cAAc,CAAC,cAAc,EAAE,GAAG,EAAE,QAAQ,CAAC;;;;;;;;;;;;;KAaxD;QACD;;;;;;KAMC;QACD;;;;;;;KAOC;QACD;;;;;;KAMC;QACD;;8CAE0C,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;;;;;;gBAO/E,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;QACD;;gDAE4C,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;gBAEjF,cAAc,CAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;;;;;gBAKhD,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;QACD;;;;;;;gBAOY,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,mBAAmB;;KAEtD;KACF,CAAC;IACF,MAAM,KAAK,GAAG;QACZ;;;eAGW,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc;KAChD;QACD;;KAEC;KACF,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAClB,gBAAgB;IAChB,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;QAChC,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,uBAAuB,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KACrG,CAAC,CAAC;AACL,CAAC;AAED,SAAS,wBAAwB,CAAC,aAAqB,EAAE,QAAwB;IAC/E,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;IACrD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,YAAY,CAAC,KAI5B;IACC,KAAK,CAAC,CAAC,mBAAmB,CAAC;QACzB,GAAG,KAAK;QACR,aAAa,EAAE,CAAC,GAAkB,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,EAAE,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC;KAC5F,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { Guid, Id64, Id64Arg, Id64Array } from \"@itwin/core-bentley\";\nimport {\n  ECSqlBinding,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n} from \"@itwin/presentation-shared\";\nimport { SelectableInstanceKey } from \"./Selectable.js\";\nimport { formIdBindings, genericExecuteQuery } from \"./Utils.js\";\n\n/**\n * Available selection scopes.\n * @public\n */\ntype SelectionScopeIdentifier = \"element\" | \"model\" | \"category\" | \"functional\";\n\n/**\n * Props for computing element selection.\n * @public\n */\nexport interface ElementSelectionScopeProps {\n  /** Identifies this as either \"element\" or \"functional\" selection scope */\n  id: \"element\" | \"functional\";\n  /**\n   * Specifies how far \"up\" we should walk to find the target element. When not specified or `0`,\n   * the target element matches the request element. When set to `1`, the target element matches the direct parent element.\n   * When set to `2`, the target element is parent of the parent element and so on. In all situations when this is `> 0`,\n   * we're not walking further than the last existing element, for example when `ancestorLevel = 1` (direct parent\n   * element is requested), but the request element doesn't have a parent, the request element is returned as the result.\n   * A negative value would result in the top-most element to be returned.\n   */\n  ancestorLevel?: number;\n}\n\n/**\n * A union of types that can be used to define a selection scope.\n * @public\n * @since 1.4.0\n */\nexport type SelectionScope = ElementSelectionScopeProps | { id: SelectionScopeIdentifier } | SelectionScopeIdentifier;\n\n/**\n * Props for `computeSelection`.\n * @public\n */\nexport interface ComputeSelectionProps {\n  /** iModel query executor. */\n  queryExecutor: ECSqlQueryExecutor;\n  /** IDs of elements to compute selection for. */\n  elementIds: Id64Arg;\n  /** Selection scope to compute selection with. */\n  scope: SelectionScope;\n  /** Optional unique id of the component that consumes this function. It can be any string, as long as it is not shared across different components. */\n  componentId?: string;\n}\n\n/**\n * Computes selection from given element IDs.\n * @public\n */\nexport async function* computeSelection(props: ComputeSelectionProps): AsyncIterableIterator<SelectableInstanceKey> {\n  const { queryExecutor, elementIds, scope } = props;\n  const componentId = props.componentId ?? Guid.createValue();\n  const componentName = \"computeSelection\";\n  if (typeof scope === \"string\") {\n    yield* computeSelection({ queryExecutor, elementIds, scope: { id: scope }, componentId });\n    return;\n  }\n\n  const nonTransientIds: Id64Array = [];\n  for (const id of Id64.iterable(elementIds)) {\n    if (!Id64.isTransient(id)) {\n      nonTransientIds.push(id);\n    }\n  }\n\n  switch (scope.id) {\n    case \"element\":\n      yield* computeElementSelection(\n        queryExecutor,\n        nonTransientIds,\n        (scope as ElementSelectionScopeProps).ancestorLevel ?? 0,\n        componentId,\n        componentName,\n      );\n      return;\n    case \"category\":\n      yield* computeCategorySelection(queryExecutor, nonTransientIds, componentId, componentName);\n      return;\n    case \"model\":\n      yield* computeModelSelection(queryExecutor, nonTransientIds, componentId, componentName);\n      return;\n    case \"functional\":\n      yield* computeFunctionalElementSelection(\n        queryExecutor,\n        nonTransientIds,\n        (scope as ElementSelectionScopeProps).ancestorLevel ?? 0,\n        componentId,\n        componentName,\n      );\n      return;\n  }\n}\n\nasync function* computeElementSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  elementIds: string[],\n  ancestorLevel: number,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const recurseUntilRoot = ancestorLevel < 0;\n  const ctes = [\n    `\n      AncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT ECInstanceId, ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, Parent.Id\n        FROM BisCore.Element\n        WHERE ${formIdBindings(\"ECInstanceId\", elementIds, bindings)}\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, a.Depth - 1, pe.Parent.Id\n        FROM AncestorElements a\n        JOIN BisCore.Element pe ON pe.ECInstanceId = a.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n  ];\n  const ecsql = `\n    SELECT DISTINCT ECInstanceId, ec_classname(ECClassId, 's.c') AS ClassName\n    FROM AncestorElements\n    WHERE ${recurseUntilRoot ? \"\" : \"Depth = 0 OR\"} ParentId IS NULL\n  `;\n  yield* executeQuery({\n    queryExecutor,\n    query: { ctes, ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/element/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeCategorySelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: string[],\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const ecsql = [\n    `\n      SELECT DISTINCT c.ECInstanceId, ec_classname(c.ECClassId, 's.c') AS ClassName\n      FROM BisCore.Category c\n      JOIN BisCore.GeometricElement2d ge ON ge.Category.Id = c.ECInstanceId\n      WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n    `,\n    `\n      SELECT DISTINCT c.ECInstanceId, ec_classname(c.ECClassId, 's.c') AS ClassName\n      FROM BisCore.Category c\n      JOIN BisCore.GeometricElement3d ge ON ge.Category.Id = c.ECInstanceId\n      WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n    `,\n  ].join(\" UNION ALL \");\n  yield* executeQuery({\n    queryExecutor,\n    query: { ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/category/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeModelSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: string[],\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const ecsql = `\n    SELECT DISTINCT m.ECInstanceId, ec_classname(m.ECClassId, 's.c') AS ClassName\n    FROM BisCore.Model m\n    JOIN BisCore.Element e ON e.Model.Id = m.ECInstanceId\n    WHERE ${formIdBindings(\"e.ECInstanceId\", ids, bindings)}\n  `;\n  yield* executeQuery({\n    queryExecutor,\n    query: { ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/model/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeFunctionalElementSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: string[],\n  ancestorLevel: number,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const recurseUntilRoot = ancestorLevel < 0;\n  // cspell:disable\n  const ctes = [\n    `\n      Elements2dOrNearestFunctionalElements(OriginalECInstanceId, OriginalECClassId, ECInstanceId, ECClassId, ParentId) AS (\n        SELECT ECInstanceId, ECClassId, ECInstanceId, ECClassId, Parent.Id\n        FROM BisCore.Element\n        WHERE ${formIdBindings(\"ECInstanceId\", ids, bindings)} AND ECClassId IS NOT (BisCore.GeometricElement3d)\n        UNION ALL\n        SELECT\n          e2onfe.OriginalECInstanceId,\n          e2onfe.OriginalECClassId,\n          COALESCE(dgrfe.TargetECInstanceId, pe.ECInstanceId),\n          COALESCE(dgrfe.TargetECClassId, pe.ECClassId),\n          pe.Parent.Id\n        FROM Elements2dOrNearestFunctionalElements e2onfe\n        LEFT JOIN BisCore.Element pe ON pe.ECInstanceId = e2onfe.ParentId\n        LEFT JOIN Functional.DrawingGraphicRepresentsFunctionalElement dgrfe ON dgrfe.SourceECInstanceId = e2onfe.ECInstanceId\n        WHERE e2onfe.ECClassId IS NOT (Functional.FunctionalElement) AND (e2onfe.ParentId IS NOT NULL OR dgrfe.TargetECInstanceId IS NOT NULL)\n      )\n    `,\n    `\n      Element2dNearestFunctionalElements(OriginalECInstanceId, ECInstanceId, ECClassId) AS (\n        SELECT OriginalECInstanceId, ECInstanceId, ECClassId\n        FROM Elements2dOrNearestFunctionalElements\n        WHERE ECClassId IS (Functional.FunctionalElement)\n      )\n    `,\n    `\n      Elements2dWithoutFunctionalElement(ECInstanceId, ECClassId) AS (\n        SELECT e2wfe.OriginalECInstanceId, OriginalECClassId\n        FROM Elements2dOrNearestFunctionalElements e2wfe\n        LEFT JOIN Element2dNearestFunctionalElements e2nfe ON e2nfe.OriginalECInstanceId = e2wfe.OriginalECInstanceId\n        WHERE e2wfe.ParentId IS NULL AND e2nfe.ECInstanceId IS NULL\n      )\n    `,\n    `\n      Elements2d(ECInstanceId, ECClassId) AS (\n        SELECT ECInstanceId, ECClassId FROM Element2dNearestFunctionalElements\n        UNION\n        SELECT ECInstanceId, ECClassId FROM Elements2dWithoutFunctionalElement\n      )\n    `,\n    `\n      Element2dAncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT e.ECInstanceId, e.ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, e.Parent.Id\n        FROM BisCore.Element e\n        JOIN Elements2d e2d ON e2d.ECInstanceId = e.ECInstanceId\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, e2ae.Depth - 1, pe.Parent.Id\n        FROM Element2dAncestorElements e2ae\n        JOIN BisCore.Element pe ON pe.ECInstanceId = e2ae.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n    `\n      Element3dAncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT ge.ECInstanceId, ge.ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, ge.Parent.Id\n        FROM BisCore.GeometricElement3d ge\n        WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, e3ae.Depth - 1, pe.Parent.Id\n        FROM Element3dAncestorElements e3ae\n        JOIN BisCore.Element pe ON pe.ECInstanceId = e3ae.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n    `\n      Element3dAncestorRelatedFunctionalElement(ECInstanceId, ClassName) AS (\n        SELECT\n          COALESCE(peff.TargetECInstanceId, e3ae.ECInstanceId),\n          ec_classname(COALESCE(peff.TargetECClassId, e3ae.ECClassId), 's.c')\n        FROM Element3dAncestorElements e3ae\n        LEFT JOIN Functional.PhysicalElementFulfillsFunction peff ON peff.SourceECInstanceId = e3ae.ECInstanceId\n        WHERE ${recurseUntilRoot ? \"\" : \"e3ae.Depth = 0 OR\"} e3ae.ParentId IS NULL\n      )\n    `,\n  ];\n  const ecsql = [\n    `\n      SELECT DISTINCT ECInstanceId, ec_classname(ECClassId, 's.c') AS ClassName\n        FROM Element2dAncestorElements\n       WHERE ${recurseUntilRoot ? \"\" : \"Depth = 0 OR\"} ParentId IS NULL\n    `,\n    `\n      SELECT DISTINCT ECInstanceId, ClassName FROM Element3dAncestorRelatedFunctionalElement\n    `,\n  ].join(\" UNION \");\n  // cspell:enable\n  yield* executeQuery({\n    queryExecutor,\n    query: { ctes, ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/functional-element/${Guid.createValue()}` },\n  });\n}\n\nfunction formAncestorLevelBinding(ancestorLevel: number, bindings: ECSqlBinding[]) {\n  bindings.push({ type: \"int\", value: ancestorLevel });\n  return \"?\";\n}\n\nasync function* executeQuery(props: {\n  queryExecutor: ECSqlQueryExecutor;\n  query: ECSqlQueryDef;\n  config?: ECSqlQueryReaderOptions;\n}): AsyncIterableIterator<SelectableInstanceKey> {\n  yield* genericExecuteQuery({\n    ...props,\n    parseQueryRow: (row: ECSqlQueryRow) => ({ className: row.ClassName, id: row.ECInstanceId }),\n  });\n}\n"]}
+{"version":3,"file":"SelectionScope.js","sourceRoot":"","sources":["../../../src/unified-selection/SelectionScope.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,IAAI,EAAE,IAAI,EAA+B,MAAM,qBAAqB,CAAC;AAQ9E,OAAO,EAAyB,2BAA2B,EAAE,MAAM,iBAAiB,CAAC;AACrF,OAAO,EAAE,cAAc,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAgDjE;;;GAGG;AACH,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,gBAAgB,CAAC,KAA4B;IAClE,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;IACnD,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;IAC5D,MAAM,aAAa,GAAG,kBAAkB,CAAC;IACzC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,KAAK,CAAC,CAAC,gBAAgB,CAAC,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;QAC1F,OAAO;IACT,CAAC;IAED,MAAM,EAAE,UAAU,EAAE,aAAa,EAAE,SAAS,EAAE,YAAY,EAAE,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;IAE3F,KAAK,MAAM,EAAE,IAAI,YAAY,EAAE,CAAC;QAC9B,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,2BAA2B,EAAE,CAAC;IACvD,CAAC;IAED,QAAQ,KAAK,CAAC,EAAE,EAAE,CAAC;QACjB,KAAK,SAAS;YACZ,KAAK,CAAC,CAAC,uBAAuB,CAC5B,aAAa,EACb,aAAa,EACZ,KAAoC,CAAC,aAAa,IAAI,CAAC,EACxD,WAAW,EACX,aAAa,CACd,CAAC;YACF,OAAO;QACT,KAAK,UAAU;YACb,KAAK,CAAC,CAAC,wBAAwB,CAAC,aAAa,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;YAC1F,OAAO;QACT,KAAK,OAAO;YACV,KAAK,CAAC,CAAC,qBAAqB,CAAC,aAAa,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;YACvF,OAAO;QACT,KAAK,YAAY;YACf,KAAK,CAAC,CAAC,iCAAiC,CACtC,aAAa,EACb,aAAa,EACZ,KAAoC,CAAC,aAAa,IAAI,CAAC,EACxD,WAAW,EACX,aAAa,CACd,CAAC;YACF,OAAO;IACX,CAAC;AACH,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,uBAAuB,CACrC,aAAiC,EACjC,GAAwB,EACxB,aAAqB,EACrB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,gBAAgB,GAAG,aAAa,GAAG,CAAC,CAAC;IAC3C,MAAM,IAAI,GAAG;QACX;;0CAEsC,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;gBAE3E,cAAc,CAAC,cAAc,EAAE,GAAG,EAAE,QAAQ,CAAC;;;;;gBAK7C,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;KACF,CAAC;IACF,MAAM,KAAK,GAAG;;;YAGJ,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc;GAC/C,CAAC;IACF,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;QAChC,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,YAAY,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KAC1F,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,wBAAwB,CACtC,aAAiC,EACjC,GAAwB,EACxB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,KAAK,GAAG;QACZ;;;;cAIU,cAAc,CAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;KACzD;QACD;;;;cAIU,cAAc,CAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;KACzD;KACF,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACtB,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;QAC1B,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,aAAa,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KAC3F,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,qBAAqB,CACnC,aAAiC,EACjC,GAAwB,EACxB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,KAAK,GAAG;;;;YAIJ,cAAc,CAAC,gBAAgB,EAAE,GAAG,EAAE,QAAQ,CAAC;GACxD,CAAC;IACF,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;QAC1B,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,UAAU,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KACxF,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,iCAAiC,CAC/C,aAAiC,EACjC,GAAwB,EACxB,aAAqB,EACrB,WAAmB,EACnB,aAAqB;IAErB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,gBAAgB,GAAG,aAAa,GAAG,CAAC,CAAC;IAC3C,iBAAiB;IACjB,MAAM,IAAI,GAAG;QACX;;;;gBAIY,cAAc,CAAC,cAAc,EAAE,GAAG,EAAE,QAAQ,CAAC;;;;;;;;;;;;;KAaxD;QACD;;;;;;KAMC;QACD;;;;;;;KAOC;QACD;;;;;;KAMC;QACD;;8CAE0C,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;;;;;;gBAO/E,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;QACD;;gDAE4C,wBAAwB,CAAC,aAAa,EAAE,QAAQ,CAAC;;gBAEjF,cAAc,CAAC,iBAAiB,EAAE,GAAG,EAAE,QAAQ,CAAC;;;;;gBAKhD,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe;;KAElD;QACD;;;;;;;gBAOY,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,mBAAmB;;KAEtD;KACF,CAAC;IACF,MAAM,KAAK,GAAG;QACZ;;;eAGW,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc;KAChD;QACD;;KAEC;KACF,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAClB,gBAAgB;IAChB,KAAK,CAAC,CAAC,YAAY,CAAC;QAClB,aAAa;QACb,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;QAChC,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,aAAa,IAAI,WAAW,uBAAuB,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE;KACrG,CAAC,CAAC;AACL,CAAC;AAED,SAAS,wBAAwB,CAAC,aAAqB,EAAE,QAAwB;IAC/E,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;IACrD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,YAAY,CAAC,KAI5B;IACC,KAAK,CAAC,CAAC,mBAAmB,CAAC;QACzB,GAAG,KAAK;QACR,aAAa,EAAE,CAAC,GAAkB,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,EAAE,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC;KAC5F,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,GAAY;IACnC,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;QAC5B,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;IACd,CAAC;IAED,oFAAoF;IACpF,IAAI,aAAa,GAAG,IAAI,CAAC;IACzB,IAAI,YAAY,GAAG,IAAI,CAAC;IACxB,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;QACrB,IAAI,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YACzB,aAAa,GAAG,KAAK,CAAC;QACxB,CAAC;aAAM,CAAC;YACN,YAAY,GAAG,KAAK,CAAC;QACvB,CAAC;QAED,IAAI,CAAC,aAAa,IAAI,CAAC,YAAY,EAAE,CAAC;YACpC,MAAM;QACR,CAAC;IACH,CAAC;IAED,mEAAmE;IACnE,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;IAC5C,CAAC;SAAM,IAAI,YAAY,EAAE,CAAC;QACxB,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC;IAC5C,CAAC;IAED,qEAAqE;IACrE,wBAAwB;IACxB,MAAM,oBAAoB,GAAc,EAAE,CAAC;IAC3C,MAAM,mBAAmB,GAAc,EAAE,CAAC;IAC1C,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;QACrB,IAAI,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YACzB,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,UAAU,EAAE,oBAAoB,EAAE,SAAS,EAAE,mBAAmB,EAAE,CAAC;AAC9E,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 { Guid, Id64, Id64Arg, Id64Array, Id64Set } from \"@itwin/core-bentley\";\nimport {\n  ECSqlBinding,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n} from \"@itwin/presentation-shared\";\nimport { SelectableInstanceKey, TRANSIENT_ELEMENT_CLASSNAME } from \"./Selectable.js\";\nimport { formIdBindings, genericExecuteQuery } from \"./Utils.js\";\n\n/**\n * Available selection scopes.\n * @public\n */\ntype SelectionScopeIdentifier = \"element\" | \"model\" | \"category\" | \"functional\";\n\n/**\n * Props for computing element selection.\n * @public\n */\nexport interface ElementSelectionScopeProps {\n  /** Identifies this as either \"element\" or \"functional\" selection scope */\n  id: \"element\" | \"functional\";\n  /**\n   * Specifies how far \"up\" we should walk to find the target element. When not specified or `0`,\n   * the target element matches the request element. When set to `1`, the target element matches the direct parent element.\n   * When set to `2`, the target element is parent of the parent element and so on. In all situations when this is `> 0`,\n   * we're not walking further than the last existing element, for example when `ancestorLevel = 1` (direct parent\n   * element is requested), but the request element doesn't have a parent, the request element is returned as the result.\n   * A negative value would result in the top-most element to be returned.\n   */\n  ancestorLevel?: number;\n}\n\n/**\n * A union of types that can be used to define a selection scope.\n * @public\n * @since 1.4.0\n */\nexport type SelectionScope = ElementSelectionScopeProps | { id: SelectionScopeIdentifier } | SelectionScopeIdentifier;\n\n/**\n * Props for `computeSelection`.\n * @public\n */\nexport interface ComputeSelectionProps {\n  /** iModel query executor. */\n  queryExecutor: ECSqlQueryExecutor;\n  /** IDs of elements to compute selection for. */\n  elementIds: Id64Arg;\n  /** Selection scope to compute selection with. */\n  scope: SelectionScope;\n  /** Optional unique id of the component that consumes this function. It can be any string, as long as it is not shared across different components. */\n  componentId?: string;\n}\n\n/**\n * Computes selection from given element IDs.\n * @public\n */\nexport async function* computeSelection(props: ComputeSelectionProps): AsyncIterableIterator<SelectableInstanceKey> {\n  const { queryExecutor, elementIds, scope } = props;\n  const componentId = props.componentId ?? Guid.createValue();\n  const componentName = \"computeSelection\";\n  if (typeof scope === \"string\") {\n    yield* computeSelection({ queryExecutor, elementIds, scope: { id: scope }, componentId });\n    return;\n  }\n\n  const { persistent: persistentIds, transient: transientIds } = parseElementIds(elementIds);\n\n  for (const id of transientIds) {\n    yield { id, className: TRANSIENT_ELEMENT_CLASSNAME };\n  }\n\n  switch (scope.id) {\n    case \"element\":\n      yield* computeElementSelection(\n        queryExecutor,\n        persistentIds,\n        (scope as ElementSelectionScopeProps).ancestorLevel ?? 0,\n        componentId,\n        componentName,\n      );\n      return;\n    case \"category\":\n      yield* computeCategorySelection(queryExecutor, persistentIds, componentId, componentName);\n      return;\n    case \"model\":\n      yield* computeModelSelection(queryExecutor, persistentIds, componentId, componentName);\n      return;\n    case \"functional\":\n      yield* computeFunctionalElementSelection(\n        queryExecutor,\n        persistentIds,\n        (scope as ElementSelectionScopeProps).ancestorLevel ?? 0,\n        componentId,\n        componentName,\n      );\n      return;\n  }\n}\n\nasync function* computeElementSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: Id64Array | Id64Set,\n  ancestorLevel: number,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const recurseUntilRoot = ancestorLevel < 0;\n  const ctes = [\n    `\n      AncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT ECInstanceId, ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, Parent.Id\n        FROM BisCore.Element\n        WHERE ${formIdBindings(\"ECInstanceId\", ids, bindings)}\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, a.Depth - 1, pe.Parent.Id\n        FROM AncestorElements a\n        JOIN BisCore.Element pe ON pe.ECInstanceId = a.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n  ];\n  const ecsql = `\n    SELECT DISTINCT ECInstanceId, ec_classname(ECClassId, 's.c') AS ClassName\n    FROM AncestorElements\n    WHERE ${recurseUntilRoot ? \"\" : \"Depth = 0 OR\"} ParentId IS NULL\n  `;\n  yield* executeQuery({\n    queryExecutor,\n    query: { ctes, ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/element/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeCategorySelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: Id64Array | Id64Set,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const ecsql = [\n    `\n      SELECT DISTINCT c.ECInstanceId, ec_classname(c.ECClassId, 's.c') AS ClassName\n      FROM BisCore.Category c\n      JOIN BisCore.GeometricElement2d ge ON ge.Category.Id = c.ECInstanceId\n      WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n    `,\n    `\n      SELECT DISTINCT c.ECInstanceId, ec_classname(c.ECClassId, 's.c') AS ClassName\n      FROM BisCore.Category c\n      JOIN BisCore.GeometricElement3d ge ON ge.Category.Id = c.ECInstanceId\n      WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n    `,\n  ].join(\" UNION ALL \");\n  yield* executeQuery({\n    queryExecutor,\n    query: { ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/category/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeModelSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: Id64Array | Id64Set,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const ecsql = `\n    SELECT DISTINCT m.ECInstanceId, ec_classname(m.ECClassId, 's.c') AS ClassName\n    FROM BisCore.Model m\n    JOIN BisCore.Element e ON e.Model.Id = m.ECInstanceId\n    WHERE ${formIdBindings(\"e.ECInstanceId\", ids, bindings)}\n  `;\n  yield* executeQuery({\n    queryExecutor,\n    query: { ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/model/${Guid.createValue()}` },\n  });\n}\n\nasync function* computeFunctionalElementSelection(\n  queryExecutor: ECSqlQueryExecutor,\n  ids: Id64Array | Id64Set,\n  ancestorLevel: number,\n  componentId: string,\n  componentName: string,\n): AsyncIterableIterator<SelectableInstanceKey> {\n  const bindings: ECSqlBinding[] = [];\n  const recurseUntilRoot = ancestorLevel < 0;\n  // cspell:disable\n  const ctes = [\n    `\n      Elements2dOrNearestFunctionalElements(OriginalECInstanceId, OriginalECClassId, ECInstanceId, ECClassId, ParentId) AS (\n        SELECT ECInstanceId, ECClassId, ECInstanceId, ECClassId, Parent.Id\n        FROM BisCore.Element\n        WHERE ${formIdBindings(\"ECInstanceId\", ids, bindings)} AND ECClassId IS NOT (BisCore.GeometricElement3d)\n        UNION ALL\n        SELECT\n          e2onfe.OriginalECInstanceId,\n          e2onfe.OriginalECClassId,\n          COALESCE(dgrfe.TargetECInstanceId, pe.ECInstanceId),\n          COALESCE(dgrfe.TargetECClassId, pe.ECClassId),\n          pe.Parent.Id\n        FROM Elements2dOrNearestFunctionalElements e2onfe\n        LEFT JOIN BisCore.Element pe ON pe.ECInstanceId = e2onfe.ParentId\n        LEFT JOIN Functional.DrawingGraphicRepresentsFunctionalElement dgrfe ON dgrfe.SourceECInstanceId = e2onfe.ECInstanceId\n        WHERE e2onfe.ECClassId IS NOT (Functional.FunctionalElement) AND (e2onfe.ParentId IS NOT NULL OR dgrfe.TargetECInstanceId IS NOT NULL)\n      )\n    `,\n    `\n      Element2dNearestFunctionalElements(OriginalECInstanceId, ECInstanceId, ECClassId) AS (\n        SELECT OriginalECInstanceId, ECInstanceId, ECClassId\n        FROM Elements2dOrNearestFunctionalElements\n        WHERE ECClassId IS (Functional.FunctionalElement)\n      )\n    `,\n    `\n      Elements2dWithoutFunctionalElement(ECInstanceId, ECClassId) AS (\n        SELECT e2wfe.OriginalECInstanceId, OriginalECClassId\n        FROM Elements2dOrNearestFunctionalElements e2wfe\n        LEFT JOIN Element2dNearestFunctionalElements e2nfe ON e2nfe.OriginalECInstanceId = e2wfe.OriginalECInstanceId\n        WHERE e2wfe.ParentId IS NULL AND e2nfe.ECInstanceId IS NULL\n      )\n    `,\n    `\n      Elements2d(ECInstanceId, ECClassId) AS (\n        SELECT ECInstanceId, ECClassId FROM Element2dNearestFunctionalElements\n        UNION\n        SELECT ECInstanceId, ECClassId FROM Elements2dWithoutFunctionalElement\n      )\n    `,\n    `\n      Element2dAncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT e.ECInstanceId, e.ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, e.Parent.Id\n        FROM BisCore.Element e\n        JOIN Elements2d e2d ON e2d.ECInstanceId = e.ECInstanceId\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, e2ae.Depth - 1, pe.Parent.Id\n        FROM Element2dAncestorElements e2ae\n        JOIN BisCore.Element pe ON pe.ECInstanceId = e2ae.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n    `\n      Element3dAncestorElements(ECInstanceId, ECClassId, Depth, ParentId) AS (\n        SELECT ge.ECInstanceId, ge.ECClassId, ${formAncestorLevelBinding(ancestorLevel, bindings)}, ge.Parent.Id\n        FROM BisCore.GeometricElement3d ge\n        WHERE ${formIdBindings(\"ge.ECInstanceId\", ids, bindings)}\n        UNION ALL\n        SELECT pe.ECInstanceId, pe.ECClassId, e3ae.Depth - 1, pe.Parent.Id\n        FROM Element3dAncestorElements e3ae\n        JOIN BisCore.Element pe ON pe.ECInstanceId = e3ae.ParentId\n        WHERE ${recurseUntilRoot ? \"\" : \"Depth > 0 AND\"} ParentId IS NOT NULL\n      )\n    `,\n    `\n      Element3dAncestorRelatedFunctionalElement(ECInstanceId, ClassName) AS (\n        SELECT\n          COALESCE(peff.TargetECInstanceId, e3ae.ECInstanceId),\n          ec_classname(COALESCE(peff.TargetECClassId, e3ae.ECClassId), 's.c')\n        FROM Element3dAncestorElements e3ae\n        LEFT JOIN Functional.PhysicalElementFulfillsFunction peff ON peff.SourceECInstanceId = e3ae.ECInstanceId\n        WHERE ${recurseUntilRoot ? \"\" : \"e3ae.Depth = 0 OR\"} e3ae.ParentId IS NULL\n      )\n    `,\n  ];\n  const ecsql = [\n    `\n      SELECT DISTINCT ECInstanceId, ec_classname(ECClassId, 's.c') AS ClassName\n        FROM Element2dAncestorElements\n       WHERE ${recurseUntilRoot ? \"\" : \"Depth = 0 OR\"} ParentId IS NULL\n    `,\n    `\n      SELECT DISTINCT ECInstanceId, ClassName FROM Element3dAncestorRelatedFunctionalElement\n    `,\n  ].join(\" UNION \");\n  // cspell:enable\n  yield* executeQuery({\n    queryExecutor,\n    query: { ctes, ecsql, bindings },\n    config: { restartToken: `${componentName}/${componentId}/functional-element/${Guid.createValue()}` },\n  });\n}\n\nfunction formAncestorLevelBinding(ancestorLevel: number, bindings: ECSqlBinding[]) {\n  bindings.push({ type: \"int\", value: ancestorLevel });\n  return \"?\";\n}\n\nasync function* executeQuery(props: {\n  queryExecutor: ECSqlQueryExecutor;\n  query: ECSqlQueryDef;\n  config?: ECSqlQueryReaderOptions;\n}): AsyncIterableIterator<SelectableInstanceKey> {\n  yield* genericExecuteQuery({\n    ...props,\n    parseQueryRow: (row: ECSqlQueryRow) => ({ className: row.ClassName, id: row.ECInstanceId }),\n  });\n}\n\nfunction parseElementIds(ids: Id64Arg): { persistent: Id64Set | Id64Array; transient: Id64Set | Id64Array } {\n  if (typeof ids === \"string\") {\n    ids = [ids];\n  }\n\n  // detect if all ids are either transient or persistent to avoid unnecessary copying\n  let allPersistent = true;\n  let allTransient = true;\n  for (const id of ids) {\n    if (Id64.isTransient(id)) {\n      allPersistent = false;\n    } else {\n      allTransient = false;\n    }\n\n    if (!allPersistent && !allTransient) {\n      break;\n    }\n  }\n\n  // avoid making a copy if ids are only persistent or only transient\n  if (allPersistent) {\n    return { persistent: ids, transient: [] };\n  } else if (allTransient) {\n    return { persistent: [], transient: ids };\n  }\n\n  // if `ids` contain mixed ids, we have to copy.. use Array instead of\n  // a Set for performance\n  const persistentElementIds: Id64Array = [];\n  const transientElementIds: Id64Array = [];\n  for (const id of ids) {\n    if (Id64.isTransient(id)) {\n      transientElementIds.push(id);\n    } else {\n      persistentElementIds.push(id);\n    }\n  }\n\n  return { persistent: persistentElementIds, transient: transientElementIds };\n}\n"]}

package/lib/esm/unified-selection/Utils.d.ts

@@ -1,11 +1,12 @@
 import "./DisposePolyfill.js";
 import { Observable } from "rxjs";
+import { Id64Array, Id64Set } from "@itwin/core-bentley";
 import { ECSqlBinding, ECSqlQueryDef, ECSqlQueryExecutor, ECSqlQueryReaderOptions, ECSqlQueryRow } from "@itwin/presentation-shared";
 /**
  * Forms ECSql bindings from given ID's.
  * @internal
  */
-export declare function formIdBindings(property: string, ids: string[], bindings: ECSqlBinding[]): string;
+export declare function formIdBindings(property: string, ids: Id64Array | Id64Set, bindings: ECSqlBinding[]): string;
 /**
  * Executes given ECSql query and extracts data from rows. Additionally handles main thread releasing.
  * @internal

package/lib/esm/unified-selection/Utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Utils.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/Utils.ts"],"names":[],"mappings":"AAKA,OAAO,sBAAsB,CAAC;AAE9B,OAAO,EAA4C,UAAU,EAAM,MAAM,MAAM,CAAC;AAChF,OAAO,EAEL,YAAY,EACZ,aAAa,EACb,kBAAkB,EAClB,uBAAuB,EACvB,aAAa,EACd,MAAM,4BAA4B,CAAC;AAEpC;;;GAGG;AACH,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,MAAM,CAYhG;AAED;;;GAGG;AACH,wBAAuB,mBAAmB,CAAC,CAAC,EAAE,EAC5C,aAAa,EACb,aAAa,EACb,KAAK,EACL,MAAM,GACP,EAAE;IACD,aAAa,EAAE,kBAAkB,CAAC;IAClC,KAAK,EAAE,aAAa,CAAC;IACrB,aAAa,EAAE,CAAC,GAAG,EAAE,aAAa,KAAK,CAAC,CAAC;IACzC,MAAM,CAAC,EAAE,uBAAuB,CAAC;CAClC,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAO3B;AAED;;;GAGG;AACH,wBAAgB,6BAA6B,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,IAC3D,KAAK,UAAU,CAAC,CAAC,CAAC,KAAG,UAAU,CAAC,CAAC,CAAC,CAa3C;AAED;;;;;;;;GAQG;AACH,wBAAgB,WAAW,CAAC,UAAU,EAAE,EAAE,GAAG;IAAE,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,IAAI,CAAA;CAAE,GAAG;IAAE,OAAO,EAAE,MAAM,IAAI,CAAA;CAAE,QAMtG"}
+{"version":3,"file":"Utils.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/Utils.ts"],"names":[],"mappings":"AAKA,OAAO,sBAAsB,CAAC;AAE9B,OAAO,EAA4C,UAAU,EAAM,MAAM,MAAM,CAAC;AAChF,OAAO,EAAQ,SAAS,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,EAEL,YAAY,EACZ,aAAa,EACb,kBAAkB,EAClB,uBAAuB,EACvB,aAAa,EACd,MAAM,4BAA4B,CAAC;AAEpC;;;GAGG;AACH,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,MAAM,CAoB3G;AAED;;;GAGG;AACH,wBAAuB,mBAAmB,CAAC,CAAC,EAAE,EAC5C,aAAa,EACb,aAAa,EACb,KAAK,EACL,MAAM,GACP,EAAE;IACD,aAAa,EAAE,kBAAkB,CAAC;IAClC,KAAK,EAAE,aAAa,CAAC;IACrB,aAAa,EAAE,CAAC,GAAG,EAAE,aAAa,KAAK,CAAC,CAAC;IACzC,MAAM,CAAC,EAAE,uBAAuB,CAAC;CAClC,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAO3B;AAED;;;GAGG;AACH,wBAAgB,6BAA6B,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,IAC3D,KAAK,UAAU,CAAC,CAAC,CAAC,KAAG,UAAU,CAAC,CAAC,CAAC,CAa3C;AAED;;;;;;;;GAQG;AACH,wBAAgB,WAAW,CAAC,UAAU,EAAE,EAAE,GAAG;IAAE,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,IAAI,CAAA;CAAE,GAAG;IAAE,OAAO,EAAE,MAAM,IAAI,CAAA;CAAE,QAMtG"}

package/lib/esm/unified-selection/Utils.js

@@ -4,21 +4,30 @@
  *--------------------------------------------------------------------------------------------*/
 import "./DisposePolyfill.js";
 import { bufferCount, concatAll, concatMap, delay, of } from "rxjs";
+import { Id64 } from "@itwin/core-bentley";
 import { createMainThreadReleaseOnTimePassedHandler, } from "@itwin/presentation-shared";
 /**
  * Forms ECSql bindings from given ID's.
  * @internal
  */
 export function formIdBindings(property, ids, bindings) {
-    if (ids.length > 1000) {
-        bindings.push({ type: "idset", value: ids });
+    const count = Id64.sizeOf(ids);
+    if (count > 1000) {
+        bindings.push({ type: "idset", value: ids instanceof Set ? [...ids] : ids });
         return `InVirtualSet(?, ${property})`;
     }
-    if (ids.length === 0) {
+    if (count === 0) {
         return `FALSE`;
     }
-    ids.forEach((id) => bindings.push({ type: "id", value: id }));
-    return `${property} IN (${ids.map(() => "?").join(",")})`;
+    let bindingsClause = "";
+    ids.forEach((id) => {
+        bindings.push({ type: "id", value: id });
+        if (bindingsClause.length > 0) {
+            bindingsClause += ",";
+        }
+        bindingsClause += "?";
+    });
+    return `${property} IN (${bindingsClause})`;
 }
 /**
  * Executes given ECSql query and extracts data from rows. Additionally handles main thread releasing.

package/lib/esm/unified-selection/Utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"Utils.js","sourceRoot":"","sources":["../../../src/unified-selection/Utils.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,sBAAsB,CAAC;AAE9B,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,SAAS,EAAE,KAAK,EAAc,EAAE,EAAE,MAAM,MAAM,CAAC;AAChF,OAAO,EACL,0CAA0C,GAM3C,MAAM,4BAA4B,CAAC;AAEpC;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,QAAgB,EAAE,GAAa,EAAE,QAAwB;IACtF,IAAI,GAAG,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC;QACtB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QAC7C,OAAO,mBAAmB,QAAQ,GAAG,CAAC;IACxC,CAAC;IAED,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IAC9D,OAAO,GAAG,QAAQ,QAAQ,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;AAC5D,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,mBAAmB,CAAI,EAC5C,aAAa,EACb,aAAa,EACb,KAAK,EACL,MAAM,GAMP;IACC,MAAM,iBAAiB,GAAG,0CAA0C,EAAE,CAAC;IACvE,MAAM,MAAM,GAAG,aAAa,CAAC,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC9D,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC/B,MAAM,aAAa,CAAC,GAAG,CAAC,CAAC;QACzB,MAAM,iBAAiB,EAAE,CAAC;IAC5B,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,6BAA6B,CAAI,YAAoB;IACnE,OAAO,CAAC,GAAkB,EAAiB,EAAE;QAC3C,OAAO,GAAG,CAAC,IAAI,CACb,WAAW,CAAC,YAAY,CAAC,EACzB,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;YACpB,MAAM,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;YACrB,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,GAAG,YAAY,EAAE,CAAC;gBAC1C,OAAO,GAAG,CAAC;YACb,CAAC;YACD,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC,CAAC,EACF,SAAS,EAAE,CACZ,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,WAAW,CAAC,UAA2E;IACrG,IAAI,SAAS,IAAI,UAAU,EAAE,CAAC;QAC5B,UAAU,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC;SAAM,IAAI,MAAM,CAAC,OAAO,IAAI,UAAU,EAAE,CAAC;QACxC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC/B,CAAC;AACH,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 \"./DisposePolyfill.js\";\n\nimport { bufferCount, concatAll, concatMap, delay, Observable, of } from \"rxjs\";\nimport {\n  createMainThreadReleaseOnTimePassedHandler,\n  ECSqlBinding,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n} from \"@itwin/presentation-shared\";\n\n/**\n * Forms ECSql bindings from given ID's.\n * @internal\n */\nexport function formIdBindings(property: string, ids: string[], bindings: ECSqlBinding[]): string {\n  if (ids.length > 1000) {\n    bindings.push({ type: \"idset\", value: ids });\n    return `InVirtualSet(?, ${property})`;\n  }\n\n  if (ids.length === 0) {\n    return `FALSE`;\n  }\n\n  ids.forEach((id) => bindings.push({ type: \"id\", value: id }));\n  return `${property} IN (${ids.map(() => \"?\").join(\",\")})`;\n}\n\n/**\n * Executes given ECSql query and extracts data from rows. Additionally handles main thread releasing.\n * @internal\n */\nexport async function* genericExecuteQuery<T>({\n  queryExecutor,\n  parseQueryRow,\n  query,\n  config,\n}: {\n  queryExecutor: ECSqlQueryExecutor;\n  query: ECSqlQueryDef;\n  parseQueryRow: (row: ECSqlQueryRow) => T;\n  config?: ECSqlQueryReaderOptions;\n}): AsyncIterableIterator<T> {\n  const releaseMainThread = createMainThreadReleaseOnTimePassedHandler();\n  const reader = queryExecutor.createQueryReader(query, config);\n  for await (const row of reader) {\n    yield parseQueryRow(row);\n    await releaseMainThread();\n  }\n}\n\n/**\n * Emits a certain amount of values, then releases the main thread for other timers to use.\n * @internal\n */\nexport function releaseMainThreadOnItemsCount<T>(elementCount: number) {\n  return (obs: Observable<T>): Observable<T> => {\n    return obs.pipe(\n      bufferCount(elementCount),\n      concatMap((buff, i) => {\n        const out = of(buff);\n        if (i === 0 && buff.length < elementCount) {\n          return out;\n        }\n        return out.pipe(delay(0));\n      }),\n      concatAll(),\n    );\n  };\n}\n\n/**\n * A helper that disposes the given object, if it's disposable.\n *\n * The first option is to dispose using the deprecated `dispose` method if it exists on the object.\n * If not, we use the new `Symbol.dispose` method. If that doesn't exist either, the object is\n * considered as non-disposable and nothing is done with it.\n *\n * @internal\n */\nexport function safeDispose(disposable: {} | { [Symbol.dispose]: () => void } | { dispose: () => void }) {\n  if (\"dispose\" in disposable) {\n    disposable.dispose();\n  } else if (Symbol.dispose in disposable) {\n    disposable[Symbol.dispose]();\n  }\n}\n"]}
+{"version":3,"file":"Utils.js","sourceRoot":"","sources":["../../../src/unified-selection/Utils.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,sBAAsB,CAAC;AAE9B,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,SAAS,EAAE,KAAK,EAAc,EAAE,EAAE,MAAM,MAAM,CAAC;AAChF,OAAO,EAAE,IAAI,EAAsB,MAAM,qBAAqB,CAAC;AAC/D,OAAO,EACL,0CAA0C,GAM3C,MAAM,4BAA4B,CAAC;AAEpC;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,QAAgB,EAAE,GAAwB,EAAE,QAAwB;IACjG,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC/B,IAAI,KAAK,GAAG,IAAI,EAAE,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;QAC7E,OAAO,mBAAmB,QAAQ,GAAG,CAAC;IACxC,CAAC;IAED,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;QAChB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,cAAc,GAAG,EAAE,CAAC;IACxB,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;QACjB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACzC,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,cAAc,IAAI,GAAG,CAAC;QACxB,CAAC;QACD,cAAc,IAAI,GAAG,CAAC;IACxB,CAAC,CAAC,CAAC;IACH,OAAO,GAAG,QAAQ,QAAQ,cAAc,GAAG,CAAC;AAC9C,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,mBAAmB,CAAI,EAC5C,aAAa,EACb,aAAa,EACb,KAAK,EACL,MAAM,GAMP;IACC,MAAM,iBAAiB,GAAG,0CAA0C,EAAE,CAAC;IACvE,MAAM,MAAM,GAAG,aAAa,CAAC,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC9D,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC/B,MAAM,aAAa,CAAC,GAAG,CAAC,CAAC;QACzB,MAAM,iBAAiB,EAAE,CAAC;IAC5B,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,6BAA6B,CAAI,YAAoB;IACnE,OAAO,CAAC,GAAkB,EAAiB,EAAE;QAC3C,OAAO,GAAG,CAAC,IAAI,CACb,WAAW,CAAC,YAAY,CAAC,EACzB,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;YACpB,MAAM,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;YACrB,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,GAAG,YAAY,EAAE,CAAC;gBAC1C,OAAO,GAAG,CAAC;YACb,CAAC;YACD,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC,CAAC,EACF,SAAS,EAAE,CACZ,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,WAAW,CAAC,UAA2E;IACrG,IAAI,SAAS,IAAI,UAAU,EAAE,CAAC;QAC5B,UAAU,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC;SAAM,IAAI,MAAM,CAAC,OAAO,IAAI,UAAU,EAAE,CAAC;QACxC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC/B,CAAC;AACH,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 \"./DisposePolyfill.js\";\n\nimport { bufferCount, concatAll, concatMap, delay, Observable, of } from \"rxjs\";\nimport { Id64, Id64Array, Id64Set } from \"@itwin/core-bentley\";\nimport {\n  createMainThreadReleaseOnTimePassedHandler,\n  ECSqlBinding,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n} from \"@itwin/presentation-shared\";\n\n/**\n * Forms ECSql bindings from given ID's.\n * @internal\n */\nexport function formIdBindings(property: string, ids: Id64Array | Id64Set, bindings: ECSqlBinding[]): string {\n  const count = Id64.sizeOf(ids);\n  if (count > 1000) {\n    bindings.push({ type: \"idset\", value: ids instanceof Set ? [...ids] : ids });\n    return `InVirtualSet(?, ${property})`;\n  }\n\n  if (count === 0) {\n    return `FALSE`;\n  }\n\n  let bindingsClause = \"\";\n  ids.forEach((id) => {\n    bindings.push({ type: \"id\", value: id });\n    if (bindingsClause.length > 0) {\n      bindingsClause += \",\";\n    }\n    bindingsClause += \"?\";\n  });\n  return `${property} IN (${bindingsClause})`;\n}\n\n/**\n * Executes given ECSql query and extracts data from rows. Additionally handles main thread releasing.\n * @internal\n */\nexport async function* genericExecuteQuery<T>({\n  queryExecutor,\n  parseQueryRow,\n  query,\n  config,\n}: {\n  queryExecutor: ECSqlQueryExecutor;\n  query: ECSqlQueryDef;\n  parseQueryRow: (row: ECSqlQueryRow) => T;\n  config?: ECSqlQueryReaderOptions;\n}): AsyncIterableIterator<T> {\n  const releaseMainThread = createMainThreadReleaseOnTimePassedHandler();\n  const reader = queryExecutor.createQueryReader(query, config);\n  for await (const row of reader) {\n    yield parseQueryRow(row);\n    await releaseMainThread();\n  }\n}\n\n/**\n * Emits a certain amount of values, then releases the main thread for other timers to use.\n * @internal\n */\nexport function releaseMainThreadOnItemsCount<T>(elementCount: number) {\n  return (obs: Observable<T>): Observable<T> => {\n    return obs.pipe(\n      bufferCount(elementCount),\n      concatMap((buff, i) => {\n        const out = of(buff);\n        if (i === 0 && buff.length < elementCount) {\n          return out;\n        }\n        return out.pipe(delay(0));\n      }),\n      concatAll(),\n    );\n  };\n}\n\n/**\n * A helper that disposes the given object, if it's disposable.\n *\n * The first option is to dispose using the deprecated `dispose` method if it exists on the object.\n * If not, we use the new `Symbol.dispose` method. If that doesn't exist either, the object is\n * considered as non-disposable and nothing is done with it.\n *\n * @internal\n */\nexport function safeDispose(disposable: {} | { [Symbol.dispose]: () => void } | { dispose: () => void }) {\n  if (\"dispose\" in disposable) {\n    disposable.dispose();\n  } else if (Symbol.dispose in disposable) {\n    disposable[Symbol.dispose]();\n  }\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@itwin/unified-selection",
-  "version": "1.7.1",
+  "version": "1.7.3",
   "description": "Package for managing unified selection in iTwin.js applications.",
   "license": "MIT",
   "repository": {
@@ -36,13 +36,13 @@
     "./package.json": "./package.json"
   },
   "dependencies": {
-    "@itwin/core-bentley": "^5.8.4",
+    "@itwin/core-bentley": "^5.9.1",
     "rxjs": "^7.8.2",
     "rxjs-for-await": "^1.0.0",
-    "@itwin/presentation-shared": "^1.2.12"
+    "@itwin/presentation-shared": "^1.2.13"
   },
   "devDependencies": {
-    "@itwin/build-tools": "^5.10.0-dev.1",
+    "@itwin/build-tools": "^5.10.0-dev.6",
     "@itwin/eslint-plugin": "^6.1.0",
     "@types/node": "^24.12.2",
     "@vitest/coverage-v8": "^4.1.5",