@itwin/presentation-backend 4.0.0-dev.8 → 4.0.0-dev.80

package/lib/cjs/presentation-backend/SelectionScopesHelper.js

@@ -1,83 +1,82 @@
-"use strict";
-/*---------------------------------------------------------------------------------------------
-* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
-* See LICENSE.md in the project root for license terms and full copyright notice.
-*--------------------------------------------------------------------------------------------*/
-/** @packageDocumentation
- * @module Core
- */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.SelectionScopesHelper = void 0;
-const core_backend_1 = require("@itwin/core-backend");
-const core_bentley_1 = require("@itwin/core-bentley");
-const presentation_common_1 = require("@itwin/presentation-common");
-const Utils_1 = require("./Utils");
-/**
- * Contains helper methods for computing selection scopes. Will get removed
- * once rules-driven scopes are implemented.
- *
- * @internal
- */
-class SelectionScopesHelper {
-    // istanbul ignore next
-    constructor() { }
-    static getSelectionScopes() {
-        const createSelectionScope = (scopeId, label, description) => ({
-            id: scopeId,
-            label,
-            description,
-        });
-        return [
-            createSelectionScope("element", "Element", "Select the picked element"),
-            createSelectionScope("assembly", "Assembly", "Select parent of the picked element"),
-            createSelectionScope("top-assembly", "Top Assembly", "Select the topmost parent of the picked element"),
-            // WIP: temporarily comment-out "category" and "model" scopes since we can't hilite contents of them fast enough
-            // createSelectionScope("category", "Category", "Select all elements in the picked element's category"),
-            // createSelectionScope("model", "Model", "Select all elements in the picked element's model"),
-        ];
-    }
-    static getElementKey(iModel, elementId, ancestorLevel) {
-        var _a, _b, _c, _d;
-        let currId = elementId;
-        let parentId = (_b = (_a = iModel.elements.tryGetElementProps(currId)) === null || _a === void 0 ? void 0 : _a.parent) === null || _b === void 0 ? void 0 : _b.id;
-        while (parentId && ancestorLevel !== 0) {
-            currId = parentId;
-            parentId = (_d = (_c = iModel.elements.tryGetElementProps(currId)) === null || _c === void 0 ? void 0 : _c.parent) === null || _d === void 0 ? void 0 : _d.id;
-            --ancestorLevel;
-        }
-        return (0, Utils_1.getElementKey)(iModel, currId);
-    }
-    static computeElementSelection(iModel, elementIds, ancestorLevel) {
-        const parentKeys = new presentation_common_1.KeySet();
-        elementIds.forEach(skipTransients((id) => {
-            const key = this.getElementKey(iModel, id, ancestorLevel);
-            key && parentKeys.add(key);
-        }));
-        return parentKeys;
-    }
-    static computeCategorySelection(iModel, ids) {
-        const categoryKeys = new presentation_common_1.KeySet();
-        ids.forEach(skipTransients((id) => {
-            const el = iModel.elements.tryGetElement(id);
-            if (el instanceof core_backend_1.GeometricElement) {
-                const category = iModel.elements.tryGetElementProps(el.category);
-                if (category)
-                    categoryKeys.add({ className: category.classFullName, id: category.id });
-            }
-        }));
-        return categoryKeys;
-    }
-    static computeModelSelection(iModel, ids) {
-        const modelKeys = new presentation_common_1.KeySet();
-        ids.forEach(skipTransients((id) => {
-            const el = iModel.elements.tryGetElementProps(id);
-            const model = el ? iModel.models.tryGetModelProps(el.model) : undefined;
-            if (model)
-                modelKeys.add({ className: model.classFullName, id: model.id });
-        }));
-        return modelKeys;
-    }
-    static getRelatedFunctionalElementKey(imodel, graphicalElementId) {
+"use strict";
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+/** @packageDocumentation
+ * @module Core
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SelectionScopesHelper = void 0;
+const core_backend_1 = require("@itwin/core-backend");
+const core_bentley_1 = require("@itwin/core-bentley");
+const presentation_common_1 = require("@itwin/presentation-common");
+const Utils_1 = require("./Utils");
+/**
+ * Contains helper methods for computing selection scopes. Will get removed
+ * once rules-driven scopes are implemented.
+ *
+ * @internal
+ */
+class SelectionScopesHelper {
+    // istanbul ignore next
+    constructor() { }
+    static getSelectionScopes() {
+        const createSelectionScope = (scopeId, label, description) => ({
+            id: scopeId,
+            label,
+            description,
+        });
+        return [
+            createSelectionScope("element", "Element", "Select the picked element"),
+            createSelectionScope("assembly", "Assembly", "Select parent of the picked element"),
+            createSelectionScope("top-assembly", "Top Assembly", "Select the topmost parent of the picked element"),
+            // WIP: temporarily comment-out "category" and "model" scopes since we can't hilite contents of them fast enough
+            // createSelectionScope("category", "Category", "Select all elements in the picked element's category"),
+            // createSelectionScope("model", "Model", "Select all elements in the picked element's model"),
+        ];
+    }
+    static getElementKey(iModel, elementId, ancestorLevel) {
+        let currId = elementId;
+        let parentId = iModel.elements.tryGetElementProps(currId)?.parent?.id;
+        while (parentId && ancestorLevel !== 0) {
+            currId = parentId;
+            parentId = iModel.elements.tryGetElementProps(currId)?.parent?.id;
+            --ancestorLevel;
+        }
+        return (0, Utils_1.getElementKey)(iModel, currId);
+    }
+    static computeElementSelection(iModel, elementIds, ancestorLevel) {
+        const parentKeys = new presentation_common_1.KeySet();
+        elementIds.forEach(skipTransients((id) => {
+            const key = this.getElementKey(iModel, id, ancestorLevel);
+            key && parentKeys.add(key);
+        }));
+        return parentKeys;
+    }
+    static computeCategorySelection(iModel, ids) {
+        const categoryKeys = new presentation_common_1.KeySet();
+        ids.forEach(skipTransients((id) => {
+            const el = iModel.elements.tryGetElement(id);
+            if (el instanceof core_backend_1.GeometricElement) {
+                const category = iModel.elements.tryGetElementProps(el.category);
+                if (category)
+                    categoryKeys.add({ className: category.classFullName, id: category.id });
+            }
+        }));
+        return categoryKeys;
+    }
+    static computeModelSelection(iModel, ids) {
+        const modelKeys = new presentation_common_1.KeySet();
+        ids.forEach(skipTransients((id) => {
+            const el = iModel.elements.tryGetElementProps(id);
+            const model = el ? iModel.models.tryGetModelProps(el.model) : undefined;
+            if (model)
+                modelKeys.add({ className: model.classFullName, id: model.id });
+        }));
+        return modelKeys;
+    }
+    static getRelatedFunctionalElementKey(imodel, graphicalElementId) {
         const query = `
       SELECT funcSchemaDef.Name || '.' || funcClassDef.Name funcElClassName, fe.ECInstanceId funcElId
         FROM bis.Element e
@@ -87,30 +86,29 @@ class SelectionScopesHelper {
         INNER JOIN meta.ECClassDef funcClassDef ON funcClassDef.ECInstanceId = fe.ECClassId
         INNER JOIN meta.ECSchemaDef funcSchemaDef ON funcSchemaDef.ECInstanceId = funcClassDef.Schema.Id
        WHERE e.ECInstanceId = ?
-      `;
-        return imodel.withPreparedStatement(query, (stmt) => {
-            stmt.bindId(1, graphicalElementId);
-            // istanbul ignore else
-            if (core_bentley_1.DbResult.BE_SQLITE_ROW === stmt.step()) {
-                const row = stmt.getRow();
-                if (row.funcElClassName && row.funcElId)
-                    return { className: row.funcElClassName.replace(".", ":"), id: row.funcElId };
-            }
-            return undefined;
-        });
-    }
-    static findFirstRelatedFunctionalElementKey(imodel, graphicalElementId) {
-        var _a, _b;
-        let currId = graphicalElementId;
-        while (currId) {
-            const relatedFunctionalKey = this.getRelatedFunctionalElementKey(imodel, currId);
-            if (relatedFunctionalKey)
-                return relatedFunctionalKey;
-            currId = (_b = (_a = imodel.elements.tryGetElementProps(currId)) === null || _a === void 0 ? void 0 : _a.parent) === null || _b === void 0 ? void 0 : _b.id;
-        }
-        return undefined;
-    }
-    static elementClassDerivesFrom(imodel, elementId, baseClassFullName) {
+      `;
+        return imodel.withPreparedStatement(query, (stmt) => {
+            stmt.bindId(1, graphicalElementId);
+            // istanbul ignore else
+            if (core_bentley_1.DbResult.BE_SQLITE_ROW === stmt.step()) {
+                const row = stmt.getRow();
+                if (row.funcElClassName && row.funcElId)
+                    return { className: row.funcElClassName.replace(".", ":"), id: row.funcElId };
+            }
+            return undefined;
+        });
+    }
+    static findFirstRelatedFunctionalElementKey(imodel, graphicalElementId) {
+        let currId = graphicalElementId;
+        while (currId) {
+            const relatedFunctionalKey = this.getRelatedFunctionalElementKey(imodel, currId);
+            if (relatedFunctionalKey)
+                return relatedFunctionalKey;
+            currId = imodel.elements.tryGetElementProps(currId)?.parent?.id;
+        }
+        return undefined;
+    }
+    static elementClassDerivesFrom(imodel, elementId, baseClassFullName) {
         const query = `
       SELECT 1
       FROM bis.Element e
@@ -118,113 +116,112 @@ class SelectionScopesHelper {
       INNER JOIN meta.ECClassDef baseClass ON baseClass.ECInstanceId = baseClassRels.TargetECInstanceId
       INNER JOIN meta.ECSchemaDef baseSchema ON baseSchema.ECInstanceId = baseClass.Schema.Id
       WHERE e.ECInstanceId = ? AND (baseSchema.Name || ':' || baseClass.Name) = ?
-      `;
-        return imodel.withPreparedStatement(query, (stmt) => {
-            stmt.bindId(1, elementId);
-            stmt.bindString(2, baseClassFullName);
-            return (core_bentley_1.DbResult.BE_SQLITE_ROW === stmt.step());
-        });
-    }
-    static computeFunctionalElementSelection(iModel, ids) {
-        const keys = new presentation_common_1.KeySet();
-        ids.forEach(skipTransients((id) => {
-            const is3d = this.elementClassDerivesFrom(iModel, id, core_backend_1.GeometricElement3d.classFullName);
-            if (!is3d) {
-                // if the input is not a 3d element, we try to find the first related functional element
-                const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);
-                if (firstFunctionalKey) {
-                    keys.add(firstFunctionalKey);
-                    return;
-                }
-            }
-            let keyToAdd;
-            if (is3d) {
-                // if we're computing scope for a 3d element, try to switch to its related functional element
-                keyToAdd = this.getRelatedFunctionalElementKey(iModel, id);
-            }
-            if (!keyToAdd)
-                keyToAdd = (0, Utils_1.getElementKey)(iModel, id);
-            keyToAdd && keys.add(keyToAdd);
-        }));
-        return keys;
-    }
-    static computeFunctionalAssemblySelection(iModel, ids) {
-        const keys = new presentation_common_1.KeySet();
-        ids.forEach(skipTransients((id) => {
-            let idToGetAssemblyFor = id;
-            const is3d = this.elementClassDerivesFrom(iModel, id, core_backend_1.GeometricElement3d.classFullName);
-            if (!is3d) {
-                // if the input is not a 3d element, we try to find the first related functional element
-                const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);
-                if (firstFunctionalKey)
-                    idToGetAssemblyFor = firstFunctionalKey.id;
-            }
-            // find the assembly of either the given element or the functional element
-            const assemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, 1);
-            let keyToAdd = assemblyKey;
-            if (is3d && keyToAdd) {
-                // if we're computing scope for a 3d element, try to switch to its related functional element
-                const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);
-                if (relatedFunctionalKey)
-                    keyToAdd = relatedFunctionalKey;
-            }
-            keyToAdd && keys.add(keyToAdd);
-        }));
-        return keys;
-    }
-    static computeFunctionalTopAssemblySelection(iModel, ids) {
-        const keys = new presentation_common_1.KeySet();
-        ids.forEach(skipTransients((id) => {
-            let idToGetAssemblyFor = id;
-            const is3d = this.elementClassDerivesFrom(iModel, id, core_backend_1.GeometricElement3d.classFullName);
-            if (!is3d) {
-                // if the input is not a 3d element, we try to find the first related functional element
-                const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);
-                if (firstFunctionalKey)
-                    idToGetAssemblyFor = firstFunctionalKey.id;
-            }
-            // find the top assembly of either the given element or the functional element
-            const topAssemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, Number.MAX_SAFE_INTEGER);
-            let keyToAdd = topAssemblyKey;
-            if (is3d && keyToAdd) {
-                // if we're computing scope for a 3d element, try to switch to its related functional element
-                const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);
-                if (relatedFunctionalKey)
-                    keyToAdd = relatedFunctionalKey;
-            }
-            keyToAdd && keys.add(keyToAdd);
-        }));
-        return keys;
-    }
-    static async computeSelection(requestOptions, elementIds, scopeId) {
-        var _a;
-        if (!(0, presentation_common_1.isComputeSelectionRequestOptions)(requestOptions)) {
-            return this.computeSelection({
-                ...requestOptions,
-                elementIds: elementIds,
-                scope: { id: scopeId },
-            });
-        }
-        switch (requestOptions.scope.id) {
-            case "element": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, (_a = requestOptions.scope.ancestorLevel) !== null && _a !== void 0 ? _a : 0);
-            case "assembly": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, 1);
-            case "top-assembly": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, Number.MAX_SAFE_INTEGER);
-            case "category": return this.computeCategorySelection(requestOptions.imodel, requestOptions.elementIds);
-            case "model": return this.computeModelSelection(requestOptions.imodel, requestOptions.elementIds);
-            case "functional":
-            case "functional-element":
-                return this.computeFunctionalElementSelection(requestOptions.imodel, requestOptions.elementIds);
-            case "functional-assembly": return this.computeFunctionalAssemblySelection(requestOptions.imodel, requestOptions.elementIds);
-            case "functional-top-assembly": return this.computeFunctionalTopAssemblySelection(requestOptions.imodel, requestOptions.elementIds);
-        }
-        throw new presentation_common_1.PresentationError(presentation_common_1.PresentationStatus.InvalidArgument, "scopeId");
-    }
-}
-exports.SelectionScopesHelper = SelectionScopesHelper;
-const skipTransients = (callback) => {
-    return (id) => {
-        if (!core_bentley_1.Id64.isTransient(id))
-            callback(id);
-    };
-};
+      `;
+        return imodel.withPreparedStatement(query, (stmt) => {
+            stmt.bindId(1, elementId);
+            stmt.bindString(2, baseClassFullName);
+            return (core_bentley_1.DbResult.BE_SQLITE_ROW === stmt.step());
+        });
+    }
+    static computeFunctionalElementSelection(iModel, ids) {
+        const keys = new presentation_common_1.KeySet();
+        ids.forEach(skipTransients((id) => {
+            const is3d = this.elementClassDerivesFrom(iModel, id, core_backend_1.GeometricElement3d.classFullName);
+            if (!is3d) {
+                // if the input is not a 3d element, we try to find the first related functional element
+                const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);
+                if (firstFunctionalKey) {
+                    keys.add(firstFunctionalKey);
+                    return;
+                }
+            }
+            let keyToAdd;
+            if (is3d) {
+                // if we're computing scope for a 3d element, try to switch to its related functional element
+                keyToAdd = this.getRelatedFunctionalElementKey(iModel, id);
+            }
+            if (!keyToAdd)
+                keyToAdd = (0, Utils_1.getElementKey)(iModel, id);
+            keyToAdd && keys.add(keyToAdd);
+        }));
+        return keys;
+    }
+    static computeFunctionalAssemblySelection(iModel, ids) {
+        const keys = new presentation_common_1.KeySet();
+        ids.forEach(skipTransients((id) => {
+            let idToGetAssemblyFor = id;
+            const is3d = this.elementClassDerivesFrom(iModel, id, core_backend_1.GeometricElement3d.classFullName);
+            if (!is3d) {
+                // if the input is not a 3d element, we try to find the first related functional element
+                const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);
+                if (firstFunctionalKey)
+                    idToGetAssemblyFor = firstFunctionalKey.id;
+            }
+            // find the assembly of either the given element or the functional element
+            const assemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, 1);
+            let keyToAdd = assemblyKey;
+            if (is3d && keyToAdd) {
+                // if we're computing scope for a 3d element, try to switch to its related functional element
+                const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);
+                if (relatedFunctionalKey)
+                    keyToAdd = relatedFunctionalKey;
+            }
+            keyToAdd && keys.add(keyToAdd);
+        }));
+        return keys;
+    }
+    static computeFunctionalTopAssemblySelection(iModel, ids) {
+        const keys = new presentation_common_1.KeySet();
+        ids.forEach(skipTransients((id) => {
+            let idToGetAssemblyFor = id;
+            const is3d = this.elementClassDerivesFrom(iModel, id, core_backend_1.GeometricElement3d.classFullName);
+            if (!is3d) {
+                // if the input is not a 3d element, we try to find the first related functional element
+                const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);
+                if (firstFunctionalKey)
+                    idToGetAssemblyFor = firstFunctionalKey.id;
+            }
+            // find the top assembly of either the given element or the functional element
+            const topAssemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, Number.MAX_SAFE_INTEGER);
+            let keyToAdd = topAssemblyKey;
+            if (is3d && keyToAdd) {
+                // if we're computing scope for a 3d element, try to switch to its related functional element
+                const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);
+                if (relatedFunctionalKey)
+                    keyToAdd = relatedFunctionalKey;
+            }
+            keyToAdd && keys.add(keyToAdd);
+        }));
+        return keys;
+    }
+    static async computeSelection(requestOptions, elementIds, scopeId) {
+        if (!(0, presentation_common_1.isComputeSelectionRequestOptions)(requestOptions)) {
+            return this.computeSelection({
+                ...requestOptions,
+                elementIds: elementIds,
+                scope: { id: scopeId },
+            });
+        }
+        switch (requestOptions.scope.id) {
+            case "element": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, requestOptions.scope.ancestorLevel ?? 0);
+            case "assembly": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, 1);
+            case "top-assembly": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, Number.MAX_SAFE_INTEGER);
+            case "category": return this.computeCategorySelection(requestOptions.imodel, requestOptions.elementIds);
+            case "model": return this.computeModelSelection(requestOptions.imodel, requestOptions.elementIds);
+            case "functional":
+            case "functional-element":
+                return this.computeFunctionalElementSelection(requestOptions.imodel, requestOptions.elementIds);
+            case "functional-assembly": return this.computeFunctionalAssemblySelection(requestOptions.imodel, requestOptions.elementIds);
+            case "functional-top-assembly": return this.computeFunctionalTopAssemblySelection(requestOptions.imodel, requestOptions.elementIds);
+        }
+        throw new presentation_common_1.PresentationError(presentation_common_1.PresentationStatus.InvalidArgument, "scopeId");
+    }
+}
+exports.SelectionScopesHelper = SelectionScopesHelper;
+const skipTransients = (callback) => {
+    return (id) => {
+        if (!core_bentley_1.Id64.isTransient(id))
+            callback(id);
+    };
+};
 //# sourceMappingURL=SelectionScopesHelper.js.map

package/lib/cjs/presentation-backend/SelectionScopesHelper.js.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionScopesHelper.js","sourceRoot":"","sources":["../../../src/presentation-backend/SelectionScopesHelper.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;AAC/F;;GAEG;;;AAEH,sDAAqF;AACrF,sDAAiE;AACjE,oEAGoC;AACpC,mCAAwC;AAExC;;;;;GAKG;AACH,MAAa,qBAAqB;IAEhC,uBAAuB;IACvB,gBAAwB,CAAC;IAElB,MAAM,CAAC,kBAAkB;QAC9B,MAAM,oBAAoB,GAAG,CAAC,OAAe,EAAE,KAAa,EAAE,WAAmB,EAAkB,EAAE,CAAC,CAAC;YACrG,EAAE,EAAE,OAAO;YACX,KAAK;YACL,WAAW;SACZ,CAAC,CAAC;QACH,OAAO;YACL,oBAAoB,CAAC,SAAS,EAAE,SAAS,EAAE,2BAA2B,CAAC;YACvE,oBAAoB,CAAC,UAAU,EAAE,UAAU,EAAE,qCAAqC,CAAC;YACnF,oBAAoB,CAAC,cAAc,EAAE,cAAc,EAAE,iDAAiD,CAAC;YACvG,gHAAgH;YAChH,wGAAwG;YACxG,+FAA+F;SAChG,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,MAAgB,EAAE,SAAqB,EAAE,aAAqB;;QACzF,IAAI,MAAM,GAAG,SAAS,CAAC;QACvB,IAAI,QAAQ,GAAG,MAAA,MAAA,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,0CAAE,MAAM,0CAAE,EAAE,CAAC;QACtE,OAAO,QAAQ,IAAI,aAAa,KAAK,CAAC,EAAE;YACtC,MAAM,GAAG,QAAQ,CAAC;YAClB,QAAQ,GAAG,MAAA,MAAA,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,0CAAE,MAAM,0CAAE,EAAE,CAAC;YAClE,EAAE,aAAa,CAAC;SACjB;QACD,OAAO,IAAA,qBAAa,EAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,MAAgB,EAAE,UAAwB,EAAE,aAAqB;QACtG,MAAM,UAAU,GAAG,IAAI,4BAAM,EAAE,CAAC;QAChC,UAAU,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE;YACvC,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,EAAE,aAAa,CAAC,CAAC;YAC1D,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,MAAM,CAAC,wBAAwB,CAAC,MAAgB,EAAE,GAAiB;QACzE,MAAM,YAAY,GAAG,IAAI,4BAAM,EAAE,CAAC;QAClC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE;YAChC,MAAM,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAC7C,IAAI,EAAE,YAAY,+BAAgB,EAAE;gBAClC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;gBACjE,IAAI,QAAQ;oBACV,YAAY,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,aAAa,EAAE,EAAE,EAAE,QAAQ,CAAC,EAAG,EAAE,CAAC,CAAC;aAC7E;QACH,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,MAAM,CAAC,qBAAqB,CAAC,MAAgB,EAAE,GAAiB;QACtE,MAAM,SAAS,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC/B,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE;YAChC,MAAM,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACxE,IAAI,KAAK;gBACP,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,aAAa,EAAE,EAAE,EAAE,KAAK,CAAC,EAAG,EAAE,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,8BAA8B,CAAC,MAAgB,EAAE,kBAA8B;QAC5F,MAAM,KAAK,GAAG;;;;;;;;;OASX,CAAC;QACJ,OAAO,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC,IAAI,EAA2B,EAAE;YAC3E,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,kBAAkB,CAAC,CAAC;YACnC,uBAAuB;YACvB,IAAI,uBAAQ,CAAC,aAAa,KAAK,IAAI,CAAC,IAAI,EAAE,EAAE;gBAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC1B,IAAI,GAAG,CAAC,eAAe,IAAI,GAAG,CAAC,QAAQ;oBACrC,OAAO,EAAE,SAAS,EAAE,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;aACjF;YACD,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,oCAAoC,CAAC,MAAgB,EAAE,kBAA8B;;QAClG,IAAI,MAAM,GAA2B,kBAAkB,CAAC;QACxD,OAAO,MAAM,EAAE;YACb,MAAM,oBAAoB,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACjF,IAAI,oBAAoB;gBACtB,OAAO,oBAAoB,CAAC;YAC9B,MAAM,GAAG,MAAA,MAAA,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,0CAAE,MAAM,0CAAE,EAAE,CAAC;SACjE;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,MAAgB,EAAE,SAAqB,EAAE,iBAAyB;QACvG,MAAM,KAAK,GAAG;;;;;;;OAOX,CAAC;QACJ,OAAO,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC,IAAI,EAAW,EAAE;YAC3D,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAC1B,IAAI,CAAC,UAAU,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;YACtC,OAAO,CAAC,uBAAQ,CAAC,aAAa,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,iCAAiC,CAAC,MAAgB,EAAE,GAAiB;QAClF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAQ,EAAE;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,iCAAkB,CAAC,aAAa,CAAC,CAAC;YACxF,IAAI,CAAC,IAAI,EAAE;gBACT,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACjF,IAAI,kBAAkB,EAAE;oBACtB,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;oBAC7B,OAAO;iBACR;aACF;YACD,IAAI,QAAiC,CAAC;YACtC,IAAI,IAAI,EAAE;gBACR,6FAA6F;gBAC7F,QAAQ,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;aAC5D;YACD,IAAI,CAAC,QAAQ;gBACX,QAAQ,GAAG,IAAA,qBAAa,EAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACvC,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,kCAAkC,CAAC,MAAgB,EAAE,GAAiB;QACnF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAQ,EAAE;YACtC,IAAI,kBAAkB,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,iCAAkB,CAAC,aAAa,CAAC,CAAC;YACxF,IAAI,CAAC,IAAI,EAAE;gBACT,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACjF,IAAI,kBAAkB;oBACpB,kBAAkB,GAAG,kBAAkB,CAAC,EAAE,CAAC;aAC9C;YACD,0EAA0E;YAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YACtE,IAAI,QAAQ,GAAG,WAAW,CAAC;YAC3B,IAAI,IAAI,IAAI,QAAQ,EAAE;gBACpB,6FAA6F;gBAC7F,MAAM,oBAAoB,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;gBACtF,IAAI,oBAAoB;oBACtB,QAAQ,GAAG,oBAAoB,CAAC;aACnC;YACD,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,qCAAqC,CAAC,MAAgB,EAAE,GAAiB;QACtF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAQ,EAAE;YACtC,IAAI,kBAAkB,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,iCAAkB,CAAC,aAAa,CAAC,CAAC;YACxF,IAAI,CAAC,IAAI,EAAE;gBACT,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACjF,IAAI,kBAAkB;oBACpB,kBAAkB,GAAG,kBAAkB,CAAC,EAAE,CAAC;aAC9C;YACD,8EAA8E;YAC9E,MAAM,cAAc,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,kBAAkB,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAC/F,IAAI,QAAQ,GAAG,cAAc,CAAC;YAC9B,IAAI,IAAI,IAAI,QAAQ,EAAE;gBACpB,6FAA6F;gBAC7F,MAAM,oBAAoB,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;gBACtF,IAAI,oBAAoB;oBACtB,QAAQ,GAAG,oBAAoB,CAAC;aACnC;YACD,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,IAAI,CAAC;IACd,CAAC;IAIM,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,cAAiG,EAAE,UAAyB,EAAE,OAAgB;;QACjL,IAAI,CAAC,IAAA,sDAAgC,EAAC,cAAc,CAAC,EAAE;YACrD,OAAO,IAAI,CAAC,gBAAgB,CAAC;gBAC3B,GAAG,cAAc;gBACjB,UAAU,EAAE,UAAW;gBACvB,KAAK,EAAE,EAAE,EAAE,EAAE,OAAQ,EAAE;aACxB,CAAC,CAAC;SACJ;QAED,QAAQ,cAAc,CAAC,KAAK,CAAC,EAAE,EAAE;YAC/B,KAAK,SAAS,CAAC,CAAC,OAAO,IAAI,CAAC,uBAAuB,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,EAAE,MAAC,cAAc,CAAC,KAAoC,CAAC,aAAa,mCAAI,CAAC,CAAC,CAAC;YAC/K,KAAK,UAAU,CAAC,CAAC,OAAO,IAAI,CAAC,uBAAuB,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAC1G,KAAK,cAAc,CAAC,CAAC,OAAO,IAAI,CAAC,uBAAuB,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACpI,KAAK,UAAU,CAAC,CAAC,OAAO,IAAI,CAAC,wBAAwB,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;YACxG,KAAK,OAAO,CAAC,CAAC,OAAO,IAAI,CAAC,qBAAqB,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;YAClG,KAAK,YAAY,CAAC;YAClB,KAAK,oBAAoB;gBACvB,OAAO,IAAI,CAAC,iCAAiC,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;YAClG,KAAK,qBAAqB,CAAC,CAAC,OAAO,IAAI,CAAC,kCAAkC,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;YAC7H,KAAK,yBAAyB,CAAC,CAAC,OAAO,IAAI,CAAC,qCAAqC,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;SACrI;QACD,MAAM,IAAI,uCAAiB,CAAC,wCAAkB,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;IAC7E,CAAC;CAEF;AAvND,sDAuNC;AAED,MAAM,cAAc,GAAG,CAAC,QAAkC,EAAE,EAAE;IAC5D,OAAO,CAAC,EAAc,EAAE,EAAE;QACxB,IAAI,CAAC,mBAAI,CAAC,WAAW,CAAC,EAAE,CAAC;YACvB,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC,CAAC;AACJ,CAAC,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n/** @packageDocumentation\r\n * @module Core\r\n */\r\n\r\nimport { GeometricElement, GeometricElement3d, IModelDb } from \"@itwin/core-backend\";\r\nimport { DbResult, Id64, Id64String } from \"@itwin/core-bentley\";\r\nimport {\r\n  ComputeSelectionRequestOptions, ElementSelectionScopeProps, InstanceKey, isComputeSelectionRequestOptions, KeySet, PresentationError,\r\n  PresentationStatus, SelectionScope, SelectionScopeRequestOptions,\r\n} from \"@itwin/presentation-common\";\r\nimport { getElementKey } from \"./Utils\";\r\n\r\n/**\r\n * Contains helper methods for computing selection scopes. Will get removed\r\n * once rules-driven scopes are implemented.\r\n *\r\n * @internal\r\n */\r\nexport class SelectionScopesHelper {\r\n\r\n  // istanbul ignore next\r\n  private constructor() { }\r\n\r\n  public static getSelectionScopes(): SelectionScope[] {\r\n    const createSelectionScope = (scopeId: string, label: string, description: string): SelectionScope => ({\r\n      id: scopeId,\r\n      label,\r\n      description,\r\n    });\r\n    return [\r\n      createSelectionScope(\"element\", \"Element\", \"Select the picked element\"),\r\n      createSelectionScope(\"assembly\", \"Assembly\", \"Select parent of the picked element\"),\r\n      createSelectionScope(\"top-assembly\", \"Top Assembly\", \"Select the topmost parent of the picked element\"),\r\n      // WIP: temporarily comment-out \"category\" and \"model\" scopes since we can't hilite contents of them fast enough\r\n      // createSelectionScope(\"category\", \"Category\", \"Select all elements in the picked element's category\"),\r\n      // createSelectionScope(\"model\", \"Model\", \"Select all elements in the picked element's model\"),\r\n    ];\r\n  }\r\n\r\n  private static getElementKey(iModel: IModelDb, elementId: Id64String, ancestorLevel: number) {\r\n    let currId = elementId;\r\n    let parentId = iModel.elements.tryGetElementProps(currId)?.parent?.id;\r\n    while (parentId && ancestorLevel !== 0) {\r\n      currId = parentId;\r\n      parentId = iModel.elements.tryGetElementProps(currId)?.parent?.id;\r\n      --ancestorLevel;\r\n    }\r\n    return getElementKey(iModel, currId);\r\n  }\r\n\r\n  private static computeElementSelection(iModel: IModelDb, elementIds: Id64String[], ancestorLevel: number) {\r\n    const parentKeys = new KeySet();\r\n    elementIds.forEach(skipTransients((id) => {\r\n      const key = this.getElementKey(iModel, id, ancestorLevel);\r\n      key && parentKeys.add(key);\r\n    }));\r\n    return parentKeys;\r\n  }\r\n\r\n  private static computeCategorySelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const categoryKeys = new KeySet();\r\n    ids.forEach(skipTransients((id) => {\r\n      const el = iModel.elements.tryGetElement(id);\r\n      if (el instanceof GeometricElement) {\r\n        const category = iModel.elements.tryGetElementProps(el.category);\r\n        if (category)\r\n          categoryKeys.add({ className: category.classFullName, id: category.id! });\r\n      }\r\n    }));\r\n    return categoryKeys;\r\n  }\r\n\r\n  private static computeModelSelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const modelKeys = new KeySet();\r\n    ids.forEach(skipTransients((id) => {\r\n      const el = iModel.elements.tryGetElementProps(id);\r\n      const model = el ? iModel.models.tryGetModelProps(el.model) : undefined;\r\n      if (model)\r\n        modelKeys.add({ className: model.classFullName, id: model.id! });\r\n    }));\r\n    return modelKeys;\r\n  }\r\n\r\n  private static getRelatedFunctionalElementKey(imodel: IModelDb, graphicalElementId: Id64String): InstanceKey | undefined {\r\n    const query = `\r\n      SELECT funcSchemaDef.Name || '.' || funcClassDef.Name funcElClassName, fe.ECInstanceId funcElId\r\n        FROM bis.Element e\r\n        LEFT JOIN func.PhysicalElementFulfillsFunction rel1 ON rel1.SourceECInstanceId = e.ECInstanceId\r\n        LEFT JOIN func.DrawingGraphicRepresentsFunctionalElement rel2 ON rel2.SourceECInstanceId = e.ECInstanceId\r\n        LEFT JOIN func.FunctionalElement fe ON fe.ECInstanceId IN (rel1.TargetECInstanceId, rel2.TargetECInstanceId)\r\n        INNER JOIN meta.ECClassDef funcClassDef ON funcClassDef.ECInstanceId = fe.ECClassId\r\n        INNER JOIN meta.ECSchemaDef funcSchemaDef ON funcSchemaDef.ECInstanceId = funcClassDef.Schema.Id\r\n       WHERE e.ECInstanceId = ?\r\n      `;\r\n    return imodel.withPreparedStatement(query, (stmt): InstanceKey | undefined => {\r\n      stmt.bindId(1, graphicalElementId);\r\n      // istanbul ignore else\r\n      if (DbResult.BE_SQLITE_ROW === stmt.step()) {\r\n        const row = stmt.getRow();\r\n        if (row.funcElClassName && row.funcElId)\r\n          return { className: row.funcElClassName.replace(\".\", \":\"), id: row.funcElId };\r\n      }\r\n      return undefined;\r\n    });\r\n  }\r\n\r\n  private static findFirstRelatedFunctionalElementKey(imodel: IModelDb, graphicalElementId: Id64String): InstanceKey | undefined {\r\n    let currId: Id64String | undefined = graphicalElementId;\r\n    while (currId) {\r\n      const relatedFunctionalKey = this.getRelatedFunctionalElementKey(imodel, currId);\r\n      if (relatedFunctionalKey)\r\n        return relatedFunctionalKey;\r\n      currId = imodel.elements.tryGetElementProps(currId)?.parent?.id;\r\n    }\r\n    return undefined;\r\n  }\r\n\r\n  private static elementClassDerivesFrom(imodel: IModelDb, elementId: Id64String, baseClassFullName: string): boolean {\r\n    const query = `\r\n      SELECT 1\r\n      FROM bis.Element e\r\n      INNER JOIN meta.ClassHasAllBaseClasses baseClassRels ON baseClassRels.SourceECInstanceId = e.ECClassId\r\n      INNER JOIN meta.ECClassDef baseClass ON baseClass.ECInstanceId = baseClassRels.TargetECInstanceId\r\n      INNER JOIN meta.ECSchemaDef baseSchema ON baseSchema.ECInstanceId = baseClass.Schema.Id\r\n      WHERE e.ECInstanceId = ? AND (baseSchema.Name || ':' || baseClass.Name) = ?\r\n      `;\r\n    return imodel.withPreparedStatement(query, (stmt): boolean => {\r\n      stmt.bindId(1, elementId);\r\n      stmt.bindString(2, baseClassFullName);\r\n      return (DbResult.BE_SQLITE_ROW === stmt.step());\r\n    });\r\n  }\r\n\r\n  private static computeFunctionalElementSelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const keys = new KeySet();\r\n    ids.forEach(skipTransients((id): void => {\r\n      const is3d = this.elementClassDerivesFrom(iModel, id, GeometricElement3d.classFullName);\r\n      if (!is3d) {\r\n        // if the input is not a 3d element, we try to find the first related functional element\r\n        const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);\r\n        if (firstFunctionalKey) {\r\n          keys.add(firstFunctionalKey);\r\n          return;\r\n        }\r\n      }\r\n      let keyToAdd: InstanceKey | undefined;\r\n      if (is3d) {\r\n        // if we're computing scope for a 3d element, try to switch to its related functional element\r\n        keyToAdd = this.getRelatedFunctionalElementKey(iModel, id);\r\n      }\r\n      if (!keyToAdd)\r\n        keyToAdd = getElementKey(iModel, id);\r\n      keyToAdd && keys.add(keyToAdd);\r\n    }));\r\n    return keys;\r\n  }\r\n\r\n  private static computeFunctionalAssemblySelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const keys = new KeySet();\r\n    ids.forEach(skipTransients((id): void => {\r\n      let idToGetAssemblyFor = id;\r\n      const is3d = this.elementClassDerivesFrom(iModel, id, GeometricElement3d.classFullName);\r\n      if (!is3d) {\r\n        // if the input is not a 3d element, we try to find the first related functional element\r\n        const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);\r\n        if (firstFunctionalKey)\r\n          idToGetAssemblyFor = firstFunctionalKey.id;\r\n      }\r\n      // find the assembly of either the given element or the functional element\r\n      const assemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, 1);\r\n      let keyToAdd = assemblyKey;\r\n      if (is3d && keyToAdd) {\r\n        // if we're computing scope for a 3d element, try to switch to its related functional element\r\n        const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);\r\n        if (relatedFunctionalKey)\r\n          keyToAdd = relatedFunctionalKey;\r\n      }\r\n      keyToAdd && keys.add(keyToAdd);\r\n    }));\r\n    return keys;\r\n  }\r\n\r\n  private static computeFunctionalTopAssemblySelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const keys = new KeySet();\r\n    ids.forEach(skipTransients((id): void => {\r\n      let idToGetAssemblyFor = id;\r\n      const is3d = this.elementClassDerivesFrom(iModel, id, GeometricElement3d.classFullName);\r\n      if (!is3d) {\r\n        // if the input is not a 3d element, we try to find the first related functional element\r\n        const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);\r\n        if (firstFunctionalKey)\r\n          idToGetAssemblyFor = firstFunctionalKey.id;\r\n      }\r\n      // find the top assembly of either the given element or the functional element\r\n      const topAssemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, Number.MAX_SAFE_INTEGER);\r\n      let keyToAdd = topAssemblyKey;\r\n      if (is3d && keyToAdd) {\r\n        // if we're computing scope for a 3d element, try to switch to its related functional element\r\n        const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);\r\n        if (relatedFunctionalKey)\r\n          keyToAdd = relatedFunctionalKey;\r\n      }\r\n      keyToAdd && keys.add(keyToAdd);\r\n    }));\r\n    return keys;\r\n  }\r\n\r\n  public static async computeSelection(requestOptions: SelectionScopeRequestOptions<IModelDb>, ids: Id64String[], scopeId: string): Promise<KeySet>;\r\n  public static async computeSelection(requestOptions: ComputeSelectionRequestOptions<IModelDb>): Promise<KeySet>;\r\n  public static async computeSelection(requestOptions: ComputeSelectionRequestOptions<IModelDb> | SelectionScopeRequestOptions<IModelDb>, elementIds?: Id64String[], scopeId?: string): Promise<KeySet> {\r\n    if (!isComputeSelectionRequestOptions(requestOptions)) {\r\n      return this.computeSelection({\r\n        ...requestOptions,\r\n        elementIds: elementIds!,\r\n        scope: { id: scopeId! },\r\n      });\r\n    }\r\n\r\n    switch (requestOptions.scope.id) {\r\n      case \"element\": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, (requestOptions.scope as ElementSelectionScopeProps).ancestorLevel ?? 0);\r\n      case \"assembly\": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, 1);\r\n      case \"top-assembly\": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, Number.MAX_SAFE_INTEGER);\r\n      case \"category\": return this.computeCategorySelection(requestOptions.imodel, requestOptions.elementIds);\r\n      case \"model\": return this.computeModelSelection(requestOptions.imodel, requestOptions.elementIds);\r\n      case \"functional\":\r\n      case \"functional-element\":\r\n        return this.computeFunctionalElementSelection(requestOptions.imodel, requestOptions.elementIds);\r\n      case \"functional-assembly\": return this.computeFunctionalAssemblySelection(requestOptions.imodel, requestOptions.elementIds);\r\n      case \"functional-top-assembly\": return this.computeFunctionalTopAssemblySelection(requestOptions.imodel, requestOptions.elementIds);\r\n    }\r\n    throw new PresentationError(PresentationStatus.InvalidArgument, \"scopeId\");\r\n  }\r\n\r\n}\r\n\r\nconst skipTransients = (callback: (id: Id64String) => void) => {\r\n  return (id: Id64String) => {\r\n    if (!Id64.isTransient(id))\r\n      callback(id);\r\n  };\r\n};\r\n"]}
+{"version":3,"file":"SelectionScopesHelper.js","sourceRoot":"","sources":["../../../src/presentation-backend/SelectionScopesHelper.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;AAC/F;;GAEG;;;AAEH,sDAAqF;AACrF,sDAAiE;AACjE,oEAGoC;AACpC,mCAAwC;AAExC;;;;;GAKG;AACH,MAAa,qBAAqB;IAEhC,uBAAuB;IACvB,gBAAwB,CAAC;IAElB,MAAM,CAAC,kBAAkB;QAC9B,MAAM,oBAAoB,GAAG,CAAC,OAAe,EAAE,KAAa,EAAE,WAAmB,EAAkB,EAAE,CAAC,CAAC;YACrG,EAAE,EAAE,OAAO;YACX,KAAK;YACL,WAAW;SACZ,CAAC,CAAC;QACH,OAAO;YACL,oBAAoB,CAAC,SAAS,EAAE,SAAS,EAAE,2BAA2B,CAAC;YACvE,oBAAoB,CAAC,UAAU,EAAE,UAAU,EAAE,qCAAqC,CAAC;YACnF,oBAAoB,CAAC,cAAc,EAAE,cAAc,EAAE,iDAAiD,CAAC;YACvG,gHAAgH;YAChH,wGAAwG;YACxG,+FAA+F;SAChG,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,MAAgB,EAAE,SAAqB,EAAE,aAAqB;QACzF,IAAI,MAAM,GAAG,SAAS,CAAC;QACvB,IAAI,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;QACtE,OAAO,QAAQ,IAAI,aAAa,KAAK,CAAC,EAAE;YACtC,MAAM,GAAG,QAAQ,CAAC;YAClB,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;YAClE,EAAE,aAAa,CAAC;SACjB;QACD,OAAO,IAAA,qBAAa,EAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,MAAgB,EAAE,UAAwB,EAAE,aAAqB;QACtG,MAAM,UAAU,GAAG,IAAI,4BAAM,EAAE,CAAC;QAChC,UAAU,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE;YACvC,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,EAAE,aAAa,CAAC,CAAC;YAC1D,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,MAAM,CAAC,wBAAwB,CAAC,MAAgB,EAAE,GAAiB;QACzE,MAAM,YAAY,GAAG,IAAI,4BAAM,EAAE,CAAC;QAClC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE;YAChC,MAAM,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAC7C,IAAI,EAAE,YAAY,+BAAgB,EAAE;gBAClC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;gBACjE,IAAI,QAAQ;oBACV,YAAY,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,aAAa,EAAE,EAAE,EAAE,QAAQ,CAAC,EAAG,EAAE,CAAC,CAAC;aAC7E;QACH,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,MAAM,CAAC,qBAAqB,CAAC,MAAgB,EAAE,GAAiB;QACtE,MAAM,SAAS,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC/B,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE;YAChC,MAAM,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACxE,IAAI,KAAK;gBACP,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,aAAa,EAAE,EAAE,EAAE,KAAK,CAAC,EAAG,EAAE,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,8BAA8B,CAAC,MAAgB,EAAE,kBAA8B;QAC5F,MAAM,KAAK,GAAG;;;;;;;;;OASX,CAAC;QACJ,OAAO,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC,IAAI,EAA2B,EAAE;YAC3E,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,kBAAkB,CAAC,CAAC;YACnC,uBAAuB;YACvB,IAAI,uBAAQ,CAAC,aAAa,KAAK,IAAI,CAAC,IAAI,EAAE,EAAE;gBAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC1B,IAAI,GAAG,CAAC,eAAe,IAAI,GAAG,CAAC,QAAQ;oBACrC,OAAO,EAAE,SAAS,EAAE,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;aACjF;YACD,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,oCAAoC,CAAC,MAAgB,EAAE,kBAA8B;QAClG,IAAI,MAAM,GAA2B,kBAAkB,CAAC;QACxD,OAAO,MAAM,EAAE;YACb,MAAM,oBAAoB,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACjF,IAAI,oBAAoB;gBACtB,OAAO,oBAAoB,CAAC;YAC9B,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;SACjE;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,MAAgB,EAAE,SAAqB,EAAE,iBAAyB;QACvG,MAAM,KAAK,GAAG;;;;;;;OAOX,CAAC;QACJ,OAAO,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC,IAAI,EAAW,EAAE;YAC3D,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAC1B,IAAI,CAAC,UAAU,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;YACtC,OAAO,CAAC,uBAAQ,CAAC,aAAa,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,iCAAiC,CAAC,MAAgB,EAAE,GAAiB;QAClF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAQ,EAAE;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,iCAAkB,CAAC,aAAa,CAAC,CAAC;YACxF,IAAI,CAAC,IAAI,EAAE;gBACT,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACjF,IAAI,kBAAkB,EAAE;oBACtB,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;oBAC7B,OAAO;iBACR;aACF;YACD,IAAI,QAAiC,CAAC;YACtC,IAAI,IAAI,EAAE;gBACR,6FAA6F;gBAC7F,QAAQ,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;aAC5D;YACD,IAAI,CAAC,QAAQ;gBACX,QAAQ,GAAG,IAAA,qBAAa,EAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACvC,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,kCAAkC,CAAC,MAAgB,EAAE,GAAiB;QACnF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAQ,EAAE;YACtC,IAAI,kBAAkB,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,iCAAkB,CAAC,aAAa,CAAC,CAAC;YACxF,IAAI,CAAC,IAAI,EAAE;gBACT,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACjF,IAAI,kBAAkB;oBACpB,kBAAkB,GAAG,kBAAkB,CAAC,EAAE,CAAC;aAC9C;YACD,0EAA0E;YAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YACtE,IAAI,QAAQ,GAAG,WAAW,CAAC;YAC3B,IAAI,IAAI,IAAI,QAAQ,EAAE;gBACpB,6FAA6F;gBAC7F,MAAM,oBAAoB,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;gBACtF,IAAI,oBAAoB;oBACtB,QAAQ,GAAG,oBAAoB,CAAC;aACnC;YACD,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,qCAAqC,CAAC,MAAgB,EAAE,GAAiB;QACtF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAQ,EAAE;YACtC,IAAI,kBAAkB,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,iCAAkB,CAAC,aAAa,CAAC,CAAC;YACxF,IAAI,CAAC,IAAI,EAAE;gBACT,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACjF,IAAI,kBAAkB;oBACpB,kBAAkB,GAAG,kBAAkB,CAAC,EAAE,CAAC;aAC9C;YACD,8EAA8E;YAC9E,MAAM,cAAc,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,kBAAkB,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAC/F,IAAI,QAAQ,GAAG,cAAc,CAAC;YAC9B,IAAI,IAAI,IAAI,QAAQ,EAAE;gBACpB,6FAA6F;gBAC7F,MAAM,oBAAoB,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;gBACtF,IAAI,oBAAoB;oBACtB,QAAQ,GAAG,oBAAoB,CAAC;aACnC;YACD,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,IAAI,CAAC;IACd,CAAC;IAIM,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,cAAiG,EAAE,UAAyB,EAAE,OAAgB;QACjL,IAAI,CAAC,IAAA,sDAAgC,EAAC,cAAc,CAAC,EAAE;YACrD,OAAO,IAAI,CAAC,gBAAgB,CAAC;gBAC3B,GAAG,cAAc;gBACjB,UAAU,EAAE,UAAW;gBACvB,KAAK,EAAE,EAAE,EAAE,EAAE,OAAQ,EAAE;aACxB,CAAC,CAAC;SACJ;QAED,QAAQ,cAAc,CAAC,KAAK,CAAC,EAAE,EAAE;YAC/B,KAAK,SAAS,CAAC,CAAC,OAAO,IAAI,CAAC,uBAAuB,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,EAAG,cAAc,CAAC,KAAoC,CAAC,aAAa,IAAI,CAAC,CAAC,CAAC;YAC/K,KAAK,UAAU,CAAC,CAAC,OAAO,IAAI,CAAC,uBAAuB,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAC1G,KAAK,cAAc,CAAC,CAAC,OAAO,IAAI,CAAC,uBAAuB,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACpI,KAAK,UAAU,CAAC,CAAC,OAAO,IAAI,CAAC,wBAAwB,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;YACxG,KAAK,OAAO,CAAC,CAAC,OAAO,IAAI,CAAC,qBAAqB,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;YAClG,KAAK,YAAY,CAAC;YAClB,KAAK,oBAAoB;gBACvB,OAAO,IAAI,CAAC,iCAAiC,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;YAClG,KAAK,qBAAqB,CAAC,CAAC,OAAO,IAAI,CAAC,kCAAkC,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;YAC7H,KAAK,yBAAyB,CAAC,CAAC,OAAO,IAAI,CAAC,qCAAqC,CAAC,cAAc,CAAC,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;SACrI;QACD,MAAM,IAAI,uCAAiB,CAAC,wCAAkB,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;IAC7E,CAAC;CAEF;AAvND,sDAuNC;AAED,MAAM,cAAc,GAAG,CAAC,QAAkC,EAAE,EAAE;IAC5D,OAAO,CAAC,EAAc,EAAE,EAAE;QACxB,IAAI,CAAC,mBAAI,CAAC,WAAW,CAAC,EAAE,CAAC;YACvB,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC,CAAC;AACJ,CAAC,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n/** @packageDocumentation\r\n * @module Core\r\n */\r\n\r\nimport { GeometricElement, GeometricElement3d, IModelDb } from \"@itwin/core-backend\";\r\nimport { DbResult, Id64, Id64String } from \"@itwin/core-bentley\";\r\nimport {\r\n  ComputeSelectionRequestOptions, ElementSelectionScopeProps, InstanceKey, isComputeSelectionRequestOptions, KeySet, PresentationError,\r\n  PresentationStatus, SelectionScope, SelectionScopeRequestOptions,\r\n} from \"@itwin/presentation-common\";\r\nimport { getElementKey } from \"./Utils\";\r\n\r\n/**\r\n * Contains helper methods for computing selection scopes. Will get removed\r\n * once rules-driven scopes are implemented.\r\n *\r\n * @internal\r\n */\r\nexport class SelectionScopesHelper {\r\n\r\n  // istanbul ignore next\r\n  private constructor() { }\r\n\r\n  public static getSelectionScopes(): SelectionScope[] {\r\n    const createSelectionScope = (scopeId: string, label: string, description: string): SelectionScope => ({\r\n      id: scopeId,\r\n      label,\r\n      description,\r\n    });\r\n    return [\r\n      createSelectionScope(\"element\", \"Element\", \"Select the picked element\"),\r\n      createSelectionScope(\"assembly\", \"Assembly\", \"Select parent of the picked element\"),\r\n      createSelectionScope(\"top-assembly\", \"Top Assembly\", \"Select the topmost parent of the picked element\"),\r\n      // WIP: temporarily comment-out \"category\" and \"model\" scopes since we can't hilite contents of them fast enough\r\n      // createSelectionScope(\"category\", \"Category\", \"Select all elements in the picked element's category\"),\r\n      // createSelectionScope(\"model\", \"Model\", \"Select all elements in the picked element's model\"),\r\n    ];\r\n  }\r\n\r\n  private static getElementKey(iModel: IModelDb, elementId: Id64String, ancestorLevel: number) {\r\n    let currId = elementId;\r\n    let parentId = iModel.elements.tryGetElementProps(currId)?.parent?.id;\r\n    while (parentId && ancestorLevel !== 0) {\r\n      currId = parentId;\r\n      parentId = iModel.elements.tryGetElementProps(currId)?.parent?.id;\r\n      --ancestorLevel;\r\n    }\r\n    return getElementKey(iModel, currId);\r\n  }\r\n\r\n  private static computeElementSelection(iModel: IModelDb, elementIds: Id64String[], ancestorLevel: number) {\r\n    const parentKeys = new KeySet();\r\n    elementIds.forEach(skipTransients((id) => {\r\n      const key = this.getElementKey(iModel, id, ancestorLevel);\r\n      key && parentKeys.add(key);\r\n    }));\r\n    return parentKeys;\r\n  }\r\n\r\n  private static computeCategorySelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const categoryKeys = new KeySet();\r\n    ids.forEach(skipTransients((id) => {\r\n      const el = iModel.elements.tryGetElement(id);\r\n      if (el instanceof GeometricElement) {\r\n        const category = iModel.elements.tryGetElementProps(el.category);\r\n        if (category)\r\n          categoryKeys.add({ className: category.classFullName, id: category.id! });\r\n      }\r\n    }));\r\n    return categoryKeys;\r\n  }\r\n\r\n  private static computeModelSelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const modelKeys = new KeySet();\r\n    ids.forEach(skipTransients((id) => {\r\n      const el = iModel.elements.tryGetElementProps(id);\r\n      const model = el ? iModel.models.tryGetModelProps(el.model) : undefined;\r\n      if (model)\r\n        modelKeys.add({ className: model.classFullName, id: model.id! });\r\n    }));\r\n    return modelKeys;\r\n  }\r\n\r\n  private static getRelatedFunctionalElementKey(imodel: IModelDb, graphicalElementId: Id64String): InstanceKey | undefined {\r\n    const query = `\r\n      SELECT funcSchemaDef.Name || '.' || funcClassDef.Name funcElClassName, fe.ECInstanceId funcElId\r\n        FROM bis.Element e\r\n        LEFT JOIN func.PhysicalElementFulfillsFunction rel1 ON rel1.SourceECInstanceId = e.ECInstanceId\r\n        LEFT JOIN func.DrawingGraphicRepresentsFunctionalElement rel2 ON rel2.SourceECInstanceId = e.ECInstanceId\r\n        LEFT JOIN func.FunctionalElement fe ON fe.ECInstanceId IN (rel1.TargetECInstanceId, rel2.TargetECInstanceId)\r\n        INNER JOIN meta.ECClassDef funcClassDef ON funcClassDef.ECInstanceId = fe.ECClassId\r\n        INNER JOIN meta.ECSchemaDef funcSchemaDef ON funcSchemaDef.ECInstanceId = funcClassDef.Schema.Id\r\n       WHERE e.ECInstanceId = ?\r\n      `;\r\n    return imodel.withPreparedStatement(query, (stmt): InstanceKey | undefined => {\r\n      stmt.bindId(1, graphicalElementId);\r\n      // istanbul ignore else\r\n      if (DbResult.BE_SQLITE_ROW === stmt.step()) {\r\n        const row = stmt.getRow();\r\n        if (row.funcElClassName && row.funcElId)\r\n          return { className: row.funcElClassName.replace(\".\", \":\"), id: row.funcElId };\r\n      }\r\n      return undefined;\r\n    });\r\n  }\r\n\r\n  private static findFirstRelatedFunctionalElementKey(imodel: IModelDb, graphicalElementId: Id64String): InstanceKey | undefined {\r\n    let currId: Id64String | undefined = graphicalElementId;\r\n    while (currId) {\r\n      const relatedFunctionalKey = this.getRelatedFunctionalElementKey(imodel, currId);\r\n      if (relatedFunctionalKey)\r\n        return relatedFunctionalKey;\r\n      currId = imodel.elements.tryGetElementProps(currId)?.parent?.id;\r\n    }\r\n    return undefined;\r\n  }\r\n\r\n  private static elementClassDerivesFrom(imodel: IModelDb, elementId: Id64String, baseClassFullName: string): boolean {\r\n    const query = `\r\n      SELECT 1\r\n      FROM bis.Element e\r\n      INNER JOIN meta.ClassHasAllBaseClasses baseClassRels ON baseClassRels.SourceECInstanceId = e.ECClassId\r\n      INNER JOIN meta.ECClassDef baseClass ON baseClass.ECInstanceId = baseClassRels.TargetECInstanceId\r\n      INNER JOIN meta.ECSchemaDef baseSchema ON baseSchema.ECInstanceId = baseClass.Schema.Id\r\n      WHERE e.ECInstanceId = ? AND (baseSchema.Name || ':' || baseClass.Name) = ?\r\n      `;\r\n    return imodel.withPreparedStatement(query, (stmt): boolean => {\r\n      stmt.bindId(1, elementId);\r\n      stmt.bindString(2, baseClassFullName);\r\n      return (DbResult.BE_SQLITE_ROW === stmt.step());\r\n    });\r\n  }\r\n\r\n  private static computeFunctionalElementSelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const keys = new KeySet();\r\n    ids.forEach(skipTransients((id): void => {\r\n      const is3d = this.elementClassDerivesFrom(iModel, id, GeometricElement3d.classFullName);\r\n      if (!is3d) {\r\n        // if the input is not a 3d element, we try to find the first related functional element\r\n        const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);\r\n        if (firstFunctionalKey) {\r\n          keys.add(firstFunctionalKey);\r\n          return;\r\n        }\r\n      }\r\n      let keyToAdd: InstanceKey | undefined;\r\n      if (is3d) {\r\n        // if we're computing scope for a 3d element, try to switch to its related functional element\r\n        keyToAdd = this.getRelatedFunctionalElementKey(iModel, id);\r\n      }\r\n      if (!keyToAdd)\r\n        keyToAdd = getElementKey(iModel, id);\r\n      keyToAdd && keys.add(keyToAdd);\r\n    }));\r\n    return keys;\r\n  }\r\n\r\n  private static computeFunctionalAssemblySelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const keys = new KeySet();\r\n    ids.forEach(skipTransients((id): void => {\r\n      let idToGetAssemblyFor = id;\r\n      const is3d = this.elementClassDerivesFrom(iModel, id, GeometricElement3d.classFullName);\r\n      if (!is3d) {\r\n        // if the input is not a 3d element, we try to find the first related functional element\r\n        const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);\r\n        if (firstFunctionalKey)\r\n          idToGetAssemblyFor = firstFunctionalKey.id;\r\n      }\r\n      // find the assembly of either the given element or the functional element\r\n      const assemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, 1);\r\n      let keyToAdd = assemblyKey;\r\n      if (is3d && keyToAdd) {\r\n        // if we're computing scope for a 3d element, try to switch to its related functional element\r\n        const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);\r\n        if (relatedFunctionalKey)\r\n          keyToAdd = relatedFunctionalKey;\r\n      }\r\n      keyToAdd && keys.add(keyToAdd);\r\n    }));\r\n    return keys;\r\n  }\r\n\r\n  private static computeFunctionalTopAssemblySelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const keys = new KeySet();\r\n    ids.forEach(skipTransients((id): void => {\r\n      let idToGetAssemblyFor = id;\r\n      const is3d = this.elementClassDerivesFrom(iModel, id, GeometricElement3d.classFullName);\r\n      if (!is3d) {\r\n        // if the input is not a 3d element, we try to find the first related functional element\r\n        const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);\r\n        if (firstFunctionalKey)\r\n          idToGetAssemblyFor = firstFunctionalKey.id;\r\n      }\r\n      // find the top assembly of either the given element or the functional element\r\n      const topAssemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, Number.MAX_SAFE_INTEGER);\r\n      let keyToAdd = topAssemblyKey;\r\n      if (is3d && keyToAdd) {\r\n        // if we're computing scope for a 3d element, try to switch to its related functional element\r\n        const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);\r\n        if (relatedFunctionalKey)\r\n          keyToAdd = relatedFunctionalKey;\r\n      }\r\n      keyToAdd && keys.add(keyToAdd);\r\n    }));\r\n    return keys;\r\n  }\r\n\r\n  public static async computeSelection(requestOptions: SelectionScopeRequestOptions<IModelDb>, ids: Id64String[], scopeId: string): Promise<KeySet>;\r\n  public static async computeSelection(requestOptions: ComputeSelectionRequestOptions<IModelDb>): Promise<KeySet>;\r\n  public static async computeSelection(requestOptions: ComputeSelectionRequestOptions<IModelDb> | SelectionScopeRequestOptions<IModelDb>, elementIds?: Id64String[], scopeId?: string): Promise<KeySet> {\r\n    if (!isComputeSelectionRequestOptions(requestOptions)) {\r\n      return this.computeSelection({\r\n        ...requestOptions,\r\n        elementIds: elementIds!,\r\n        scope: { id: scopeId! },\r\n      });\r\n    }\r\n\r\n    switch (requestOptions.scope.id) {\r\n      case \"element\": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, (requestOptions.scope as ElementSelectionScopeProps).ancestorLevel ?? 0);\r\n      case \"assembly\": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, 1);\r\n      case \"top-assembly\": return this.computeElementSelection(requestOptions.imodel, requestOptions.elementIds, Number.MAX_SAFE_INTEGER);\r\n      case \"category\": return this.computeCategorySelection(requestOptions.imodel, requestOptions.elementIds);\r\n      case \"model\": return this.computeModelSelection(requestOptions.imodel, requestOptions.elementIds);\r\n      case \"functional\":\r\n      case \"functional-element\":\r\n        return this.computeFunctionalElementSelection(requestOptions.imodel, requestOptions.elementIds);\r\n      case \"functional-assembly\": return this.computeFunctionalAssemblySelection(requestOptions.imodel, requestOptions.elementIds);\r\n      case \"functional-top-assembly\": return this.computeFunctionalTopAssemblySelection(requestOptions.imodel, requestOptions.elementIds);\r\n    }\r\n    throw new PresentationError(PresentationStatus.InvalidArgument, \"scopeId\");\r\n  }\r\n\r\n}\r\n\r\nconst skipTransients = (callback: (id: Id64String) => void) => {\r\n  return (id: Id64String) => {\r\n    if (!Id64.isTransient(id))\r\n      callback(id);\r\n  };\r\n};\r\n"]}