@itwin/unified-selection 0.1.0 → 0.3.0

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

@@ -0,0 +1,205 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IModelSelectionHandler = exports.enableUnifiedSelectionSyncWithIModel = void 0;
+const rxjs_1 = require("rxjs");
+const core_bentley_1 = require("@itwin/core-bentley");
+const CachingHiliteSetProvider_1 = require("./CachingHiliteSetProvider");
+const HiliteSetProvider_1 = require("./HiliteSetProvider");
+const SelectionScope_1 = require("./SelectionScope");
+const IModel_1 = require("./types/IModel");
+/**
+ * Enables synchronization between iModel selection and unified selection.
+ * @returns function for disposing the synchronization.
+ * @beta
+ */
+function enableUnifiedSelectionSyncWithIModel(props) {
+    const selectionHandler = new IModelSelectionHandler(props);
+    return () => selectionHandler.dispose();
+}
+exports.enableUnifiedSelectionSyncWithIModel = enableUnifiedSelectionSyncWithIModel;
+/**
+ * A handler that syncs selection between unified selection storage
+ * (`SelectionStorage`) and an iModel (`iModel.selectionSet`, `iModel.hilited`).
+ * @internal
+ */
+class IModelSelectionHandler {
+    _selectionSourceName = "Tool";
+    _imodelAccess;
+    _selectionStorage;
+    _hiliteSetProvider;
+    _cachingHiliteSetProvider;
+    _activeScopeProvider;
+    _isSuspended;
+    _cancelOngoingChanges = new rxjs_1.Subject();
+    _unifiedSelectionListenerDisposeFunc;
+    _imodelListenerDisposeFunc;
+    _hasCustomCachingHiliteSetProvider;
+    constructor(props) {
+        this._imodelAccess = props.imodelAccess;
+        this._selectionStorage = props.selectionStorage;
+        this._activeScopeProvider = props.activeScopeProvider;
+        this._isSuspended = false;
+        this._hasCustomCachingHiliteSetProvider = !!props.cachingHiliteSetProvider;
+        this._cachingHiliteSetProvider =
+            props.cachingHiliteSetProvider ??
+                (0, CachingHiliteSetProvider_1.createCachingHiliteSetProvider)({
+                    selectionStorage: this._selectionStorage,
+                    imodelProvider: () => this._imodelAccess,
+                });
+        this._hiliteSetProvider = (0, HiliteSetProvider_1.createHiliteSetProvider)({ imodelAccess: this._imodelAccess });
+        this._imodelListenerDisposeFunc = this._imodelAccess.selectionSet.onChanged.addListener(this.onIModelSelectionChanged);
+        this._unifiedSelectionListenerDisposeFunc = this._selectionStorage.selectionChangeEvent.addListener(this.onUnifiedSelectionChanged);
+        // stop imodel from syncing tool selection with hilited list - we want to override that behavior
+        this._imodelAccess.hiliteSet.wantSyncWithSelectionSet = false;
+        this.applyCurrentHiliteSet({ activeSelectionAction: "clearAll" });
+    }
+    dispose() {
+        this._cancelOngoingChanges.next();
+        this._imodelListenerDisposeFunc();
+        this._unifiedSelectionListenerDisposeFunc();
+        if (!this._hasCustomCachingHiliteSetProvider) {
+            this._cachingHiliteSetProvider.dispose();
+        }
+    }
+    /** Temporarily suspends tool selection synchronization until the returned `IDisposable` is disposed. */
+    suspendIModelToolSelectionSync() {
+        const wasSuspended = this._isSuspended;
+        this._isSuspended = true;
+        return {
+            dispose: () => (this._isSuspended = wasSuspended),
+        };
+    }
+    handleUnifiedSelectionChange(changeType, selectables, source) {
+        if (changeType === "clear" || changeType === "replace") {
+            this.applyCurrentHiliteSet({ activeSelectionAction: changeType === "replace" && source === "Tool" ? "clearHilited" : "clearAll" });
+            return;
+        }
+        if (changeType === "add") {
+            (0, rxjs_1.from)(this._hiliteSetProvider.getHiliteSet({ selectables }))
+                .pipe((0, rxjs_1.takeUntil)(this._cancelOngoingChanges))
+                .subscribe({
+                next: (set) => {
+                    this.addHiliteSet(set);
+                },
+            });
+            return;
+        }
+        (0, rxjs_1.from)(this._hiliteSetProvider.getHiliteSet({ selectables }))
+            .pipe((0, rxjs_1.takeUntil)(this._cancelOngoingChanges))
+            .subscribe({
+            next: (set) => {
+                this.removeHiliteSet(set);
+            },
+            complete: () => {
+                this.applyCurrentHiliteSet({ activeSelectionAction: "keep" });
+            },
+        });
+    }
+    onUnifiedSelectionChanged = (args) => {
+        // iModels are only interested in top-level selection changes
+        if (args.imodelKey !== this._imodelAccess.key || args.level !== 0) {
+            return;
+        }
+        if (args.changeType === "replace" || args.changeType === "clear") {
+            this._cancelOngoingChanges.next();
+        }
+        this.handleUnifiedSelectionChange(args.changeType, args.selectables, args.source);
+    };
+    applyCurrentHiliteSet({ activeSelectionAction }) {
+        if (activeSelectionAction !== "keep") {
+            (0, core_bentley_1.using)(this.suspendIModelToolSelectionSync(), (_) => {
+                this._imodelAccess.hiliteSet.clear();
+                if (activeSelectionAction === "clearAll") {
+                    this._imodelAccess.selectionSet.emptyAll();
+                }
+            });
+        }
+        (0, rxjs_1.from)(this._cachingHiliteSetProvider.getHiliteSet({ imodelKey: this._imodelAccess.key }))
+            .pipe((0, rxjs_1.takeUntil)(this._cancelOngoingChanges))
+            .subscribe({
+            next: (ids) => {
+                this.addHiliteSet(ids);
+            },
+        });
+    }
+    addHiliteSet(set) {
+        (0, core_bentley_1.using)(this.suspendIModelToolSelectionSync(), (_) => {
+            if (set.models && set.models.length) {
+                this._imodelAccess.hiliteSet.models.addIds(set.models);
+            }
+            if (set.subCategories && set.subCategories.length) {
+                this._imodelAccess.hiliteSet.subcategories.addIds(set.subCategories);
+            }
+            if (set.elements && set.elements.length) {
+                this._imodelAccess.hiliteSet.elements.addIds(set.elements);
+                this._imodelAccess.selectionSet.add(set.elements);
+            }
+        });
+    }
+    removeHiliteSet(set) {
+        (0, core_bentley_1.using)(this.suspendIModelToolSelectionSync(), (_) => {
+            if (set.models.length) {
+                this._imodelAccess.hiliteSet.models.deleteIds(set.models);
+            }
+            if (set.subCategories.length) {
+                this._imodelAccess.hiliteSet.subcategories.deleteIds(set.subCategories);
+            }
+            if (set.elements.length) {
+                this._imodelAccess.hiliteSet.elements.deleteIds(set.elements);
+                this._imodelAccess.selectionSet.remove(set.elements);
+            }
+        });
+    }
+    onIModelSelectionChanged = async (event) => {
+        if (this._isSuspended) {
+            return;
+        }
+        if (IModel_1.CoreSelectionSetEventType.Clear === event.type) {
+            this._selectionStorage.clearSelection({ imodelKey: this._imodelAccess.key, source: this._selectionSourceName });
+            return;
+        }
+        const elementIds = this.getSelectionSetChangeIds(event);
+        const scopedSelection = (0, SelectionScope_1.computeSelection)({ queryExecutor: this._imodelAccess, elementIds, scope: this._activeScopeProvider() });
+        await this.handleIModelSelectionChange(event.type, scopedSelection);
+    };
+    async handleIModelSelectionChange(type, iterator) {
+        if (type === IModel_1.CoreSelectionSetEventType.Remove) {
+            for await (const selectable of iterator) {
+                this._selectionStorage.removeFromSelection({ imodelKey: this._imodelAccess.key, source: this._selectionSourceName, selectables: [selectable] });
+            }
+            return;
+        }
+        if (type === IModel_1.CoreSelectionSetEventType.Replace) {
+            this._selectionStorage.clearSelection({ imodelKey: this._imodelAccess.key, source: this._selectionSourceName });
+        }
+        for await (const selectable of iterator) {
+            this._selectionStorage.addToSelection({ imodelKey: this._imodelAccess.key, source: this._selectionSourceName, selectables: [selectable] });
+        }
+    }
+    getSelectionSetChangeIds(event) {
+        switch (event.type) {
+            case IModel_1.CoreSelectionSetEventType.Add:
+            case IModel_1.CoreSelectionSetEventType.Replace:
+                (0, core_bentley_1.assert)(!!event.added);
+                return this.idArgToIds(event.added);
+            default:
+                (0, core_bentley_1.assert)(!!event.removed);
+                return this.idArgToIds(event.removed);
+        }
+    }
+    idArgToIds(ids) {
+        if (typeof ids === "string") {
+            return [ids];
+        }
+        if (ids instanceof Set) {
+            return [...ids];
+        }
+        return ids;
+    }
+}
+exports.IModelSelectionHandler = IModelSelectionHandler;
+//# sourceMappingURL=EnableUnifiedSelectionSyncWithIModel.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"EnableUnifiedSelectionSyncWithIModel.js","sourceRoot":"","sources":["../../../src/unified-selection/EnableUnifiedSelectionSyncWithIModel.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,+BAAgD;AAChD,sDAA6D;AAE7D,yEAAsG;AACtG,2DAA4F;AAG5F,qDAA2E;AAE3E,2CAAqI;AAoDrI;;;;GAIG;AACH,SAAgB,oCAAoC,CAAC,KAAgD;IACnG,MAAM,gBAAgB,GAAG,IAAI,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAC3D,OAAO,GAAG,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;AAC1C,CAAC;AAHD,oFAGC;AAED;;;;GAIG;AACH,MAAa,sBAAsB;IACzB,oBAAoB,GAAG,MAAM,CAAC;IAE9B,aAAa,CAA4D;IACzE,iBAAiB,CAAmB;IACpC,kBAAkB,CAAoB;IACtC,yBAAyB,CAA2B;IACpD,oBAAoB,CAAuC;IAE3D,YAAY,CAAU;IACtB,qBAAqB,GAAG,IAAI,cAAO,EAAQ,CAAC;IAC5C,oCAAoC,CAAa;IACjD,0BAA0B,CAAa;IACvC,kCAAkC,CAAU;IAEpD,YAAmB,KAAgD;QACjE,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC,mBAAmB,CAAC;QACtD,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,kCAAkC,GAAG,CAAC,CAAC,KAAK,CAAC,wBAAwB,CAAC;QAE3E,IAAI,CAAC,yBAAyB;YAC5B,KAAK,CAAC,wBAAwB;gBAC9B,IAAA,yDAA8B,EAAC;oBAC7B,gBAAgB,EAAE,IAAI,CAAC,iBAAiB;oBACxC,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,aAAa;iBACzC,CAAC,CAAC;QAEL,IAAI,CAAC,kBAAkB,GAAG,IAAA,2CAAuB,EAAC,EAAE,YAAY,EAAE,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;QACxF,IAAI,CAAC,0BAA0B,GAAG,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACvH,IAAI,CAAC,oCAAoC,GAAG,IAAI,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,WAAW,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QAEpI,gGAAgG;QAChG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,wBAAwB,GAAG,KAAK,CAAC;QAC9D,IAAI,CAAC,qBAAqB,CAAC,EAAE,qBAAqB,EAAE,UAAU,EAAE,CAAC,CAAC;IACpE,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,qBAAqB,CAAC,IAAI,EAAE,CAAC;QAClC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QAClC,IAAI,CAAC,oCAAoC,EAAE,CAAC;QAC5C,IAAI,CAAC,IAAI,CAAC,kCAAkC,EAAE,CAAC;YAC7C,IAAI,CAAC,yBAAyB,CAAC,OAAO,EAAE,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,wGAAwG;IACjG,8BAA8B;QACnC,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;QACvC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,OAAO;YACL,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;SAClD,CAAC;IACJ,CAAC;IAEO,4BAA4B,CAAC,UAAsC,EAAE,WAAwB,EAAE,MAAc;QACnH,IAAI,UAAU,KAAK,OAAO,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;YACvD,IAAI,CAAC,qBAAqB,CAAC,EAAE,qBAAqB,EAAE,UAAU,KAAK,SAAS,IAAI,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC;YACnI,OAAO;QACT,CAAC;QAED,IAAI,UAAU,KAAK,KAAK,EAAE,CAAC;YACzB,IAAA,WAAI,EAAC,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC;iBACxD,IAAI,CAAC,IAAA,gBAAS,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;iBAC3C,SAAS,CAAC;gBACT,IAAI,EAAE,CAAC,GAAG,EAAE,EAAE;oBACZ,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;gBACzB,CAAC;aACF,CAAC,CAAC;YACL,OAAO;QACT,CAAC;QAED,IAAA,WAAI,EAAC,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC;aACxD,IAAI,CAAC,IAAA,gBAAS,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;aAC3C,SAAS,CAAC;YACT,IAAI,EAAE,CAAC,GAAG,EAAE,EAAE;gBACZ,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAC5B,CAAC;YACD,QAAQ,EAAE,GAAG,EAAE;gBACb,IAAI,CAAC,qBAAqB,CAAC,EAAE,qBAAqB,EAAE,MAAM,EAAE,CAAC,CAAC;YAChE,CAAC;SACF,CAAC,CAAC;IACP,CAAC;IAEO,yBAAyB,GAAG,CAAC,IAAqC,EAAE,EAAE;QAC5E,6DAA6D;QAC7D,IAAI,IAAI,CAAC,SAAS,KAAK,IAAI,CAAC,aAAa,CAAC,GAAG,IAAI,IAAI,CAAC,KAAK,KAAK,CAAC,EAAE,CAAC;YAClE,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,IAAI,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;YACjE,IAAI,CAAC,qBAAqB,CAAC,IAAI,EAAE,CAAC;QACpC,CAAC;QACD,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACpF,CAAC,CAAC;IAEM,qBAAqB,CAAC,EAAE,qBAAqB,EAAmE;QACtH,IAAI,qBAAqB,KAAK,MAAM,EAAE,CAAC;YACrC,IAAA,oBAAK,EAAC,IAAI,CAAC,8BAA8B,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE;gBACjD,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;gBACrC,IAAI,qBAAqB,KAAK,UAAU,EAAE,CAAC;oBACzC,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;gBAC7C,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAA,WAAI,EAAC,IAAI,CAAC,yBAAyB,CAAC,YAAY,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC;aACrF,IAAI,CAAC,IAAA,gBAAS,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;aAC3C,SAAS,CAAC;YACT,IAAI,EAAE,CAAC,GAAG,EAAE,EAAE;gBACZ,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;SACF,CAAC,CAAC;IACP,CAAC;IAEO,YAAY,CAAC,GAAc;QACjC,IAAA,oBAAK,EAAC,IAAI,CAAC,8BAA8B,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE;YACjD,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBACpC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACzD,CAAC;YACD,IAAI,GAAG,CAAC,aAAa,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;gBAClD,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;YACvE,CAAC;YACD,IAAI,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACxC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBAC3D,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACpD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,eAAe,CAAC,GAAc;QACpC,IAAA,oBAAK,EAAC,IAAI,CAAC,8BAA8B,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE;YACjD,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBACtB,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC5D,CAAC;YACD,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;gBAC7B,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,aAAa,CAAC,SAAS,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;YAC1E,CAAC;YACD,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACxB,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBAC9D,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACvD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,wBAAwB,GAAG,KAAK,EAAE,KAAmC,EAAiB,EAAE;QAC9F,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO;QACT,CAAC;QAED,IAAI,kCAAyB,CAAC,KAAK,KAAK,KAAM,CAAC,IAAI,EAAE,CAAC;YACpD,IAAI,CAAC,iBAAiB,CAAC,cAAc,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,CAAC,CAAC;YAChH,OAAO;QACT,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,wBAAwB,CAAC,KAAM,CAAC,CAAC;QACzD,MAAM,eAAe,GAAG,IAAA,iCAAgB,EAAC,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;QAChI,MAAM,IAAI,CAAC,2BAA2B,CAAC,KAAM,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;IACvE,CAAC,CAAC;IAEM,KAAK,CAAC,2BAA2B,CAAC,IAA+B,EAAE,QAAsD;QAC/H,IAAI,IAAI,KAAK,kCAAyB,CAAC,MAAM,EAAE,CAAC;YAC9C,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,QAAQ,EAAE,CAAC;gBACxC,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAClJ,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,IAAI,KAAK,kCAAyB,CAAC,OAAO,EAAE,CAAC;YAC/C,IAAI,CAAC,iBAAiB,CAAC,cAAc,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,CAAC,CAAC;QAClH,CAAC;QAED,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,QAAQ,EAAE,CAAC;YACxC,IAAI,CAAC,iBAAiB,CAAC,cAAc,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,WAAW,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAC7I,CAAC;IACH,CAAC;IAEO,wBAAwB,CAAC,KAAkC;QACjE,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,kCAAyB,CAAC,GAAG,CAAC;YACnC,KAAK,kCAAyB,CAAC,OAAO;gBACpC,IAAA,qBAAM,EAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACtB,OAAO,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACtC;gBACE,IAAA,qBAAM,EAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBACxB,OAAO,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,GAAY;QAC7B,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,CAAC;QACf,CAAC;QACD,IAAI,GAAG,YAAY,GAAG,EAAE,CAAC;YACvB,OAAO,CAAC,GAAG,GAAG,CAAC,CAAC;QAClB,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;CACF;AAvMD,wDAuMC","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 { from, Subject, takeUntil } from \"rxjs\";\nimport { assert, Id64Arg, using } from \"@itwin/core-bentley\";\nimport { ECSchemaProvider, ECSqlQueryExecutor } from \"@itwin/presentation-shared\";\nimport { CachingHiliteSetProvider, createCachingHiliteSetProvider } from \"./CachingHiliteSetProvider\";\nimport { createHiliteSetProvider, HiliteSet, HiliteSetProvider } from \"./HiliteSetProvider\";\nimport { SelectableInstanceKey, Selectables } from \"./Selectable\";\nimport { StorageSelectionChangeEventArgs, StorageSelectionChangeType } from \"./SelectionChangeEvent\";\nimport { computeSelection, ComputeSelectionProps } from \"./SelectionScope\";\nimport { SelectionStorage } from \"./SelectionStorage\";\nimport { CoreIModelHiliteSet, CoreIModelSelectionSet, CoreSelectionSetEventType, CoreSelectionSetEventUnsafe } from \"./types/IModel\";\n\n/**\n * Props for `enableUnifiedSelectionSyncWithIModel`.\n * @internal Not exported through barrel, but used in public API as an argument. May be supplemented with optional attributes any time.\n */\nexport interface EnableUnifiedSelectionSyncWithIModelProps {\n  /**\n   * Provides access to different iModel's features: query executing, metadata access, selection and hilite sets.\n   * It's recommended to use `@itwin/presentation-core-interop` to create `ECSqlQueryExecutor` and `ECSchemaProvider` from\n   * [IModelConnection](https://www.itwinjs.org/reference/core-frontend/imodelconnection/imodelconnection/) and map its `key`,\n   * `hilited` and `selectionSet` attributes like this:\n   *\n   * ```ts\n   * import { createECSqlQueryExecutor, createECSchemaProvider } from \"@itwin/presentation-core-interop\";\n   * import { IModelConnection } from \"@itwin/core-frontend\";\n   *\n   * const imodel: IModelConnection = ...\n   * const imodelAccess = {\n   *   ...createECSqlQueryExecutor(imodel),\n   *   ...createECSchemaProvider(MyAppFrontend.getSchemaContext(imodel)),\n   *   key: imodel.key,\n   *   hiliteSet: imodel.hilited,\n   *   selectionSet: imodel.selectionSet,\n   * };\n   * ```.\n   */\n  imodelAccess: ECSqlQueryExecutor &\n    ECSchemaProvider & {\n      /** Key of the iModel. Generally taken from `IModelConnection.key`. */\n      readonly key: string;\n      /** The set of currently hilited elements taken from `IModelConnection.hilited`. */\n      readonly hiliteSet: CoreIModelHiliteSet;\n      /** The set of currently selected elements taken from `IModelConnection.selectionSet`. */\n      readonly selectionSet: CoreIModelSelectionSet;\n    };\n\n  /** Selection storage to synchronize IModel's tool selection with. */\n  selectionStorage: SelectionStorage;\n\n  /** Active scope provider. */\n  activeScopeProvider: () => ComputeSelectionProps[\"scope\"];\n\n  /**\n   * A caching hilite set provider used to retrieve hilite sets for an iModel. If not provided, a new `CachingHiliteSetProvider`\n   * will be created for the given iModel using the provided `imodelAccess`.\n   * If the consuming application already has a `CachingHiliteSetProvider` defined, it should be provided instead\n   * to reuse the cache and avoid creating new providers for each iModel.\n   */\n  cachingHiliteSetProvider?: CachingHiliteSetProvider;\n}\n\n/**\n * Enables synchronization between iModel selection and unified selection.\n * @returns function for disposing the synchronization.\n * @beta\n */\nexport function enableUnifiedSelectionSyncWithIModel(props: EnableUnifiedSelectionSyncWithIModelProps): () => void {\n  const selectionHandler = new IModelSelectionHandler(props);\n  return () => selectionHandler.dispose();\n}\n\n/**\n * A handler that syncs selection between unified selection storage\n * (`SelectionStorage`) and an iModel (`iModel.selectionSet`, `iModel.hilited`).\n * @internal\n */\nexport class IModelSelectionHandler {\n  private _selectionSourceName = \"Tool\";\n\n  private _imodelAccess: EnableUnifiedSelectionSyncWithIModelProps[\"imodelAccess\"];\n  private _selectionStorage: SelectionStorage;\n  private _hiliteSetProvider: HiliteSetProvider;\n  private _cachingHiliteSetProvider: CachingHiliteSetProvider;\n  private _activeScopeProvider: () => ComputeSelectionProps[\"scope\"];\n\n  private _isSuspended: boolean;\n  private _cancelOngoingChanges = new Subject<void>();\n  private _unifiedSelectionListenerDisposeFunc: () => void;\n  private _imodelListenerDisposeFunc: () => void;\n  private _hasCustomCachingHiliteSetProvider: boolean;\n\n  public constructor(props: EnableUnifiedSelectionSyncWithIModelProps) {\n    this._imodelAccess = props.imodelAccess;\n    this._selectionStorage = props.selectionStorage;\n    this._activeScopeProvider = props.activeScopeProvider;\n    this._isSuspended = false;\n    this._hasCustomCachingHiliteSetProvider = !!props.cachingHiliteSetProvider;\n\n    this._cachingHiliteSetProvider =\n      props.cachingHiliteSetProvider ??\n      createCachingHiliteSetProvider({\n        selectionStorage: this._selectionStorage,\n        imodelProvider: () => this._imodelAccess,\n      });\n\n    this._hiliteSetProvider = createHiliteSetProvider({ imodelAccess: this._imodelAccess });\n    this._imodelListenerDisposeFunc = this._imodelAccess.selectionSet.onChanged.addListener(this.onIModelSelectionChanged);\n    this._unifiedSelectionListenerDisposeFunc = this._selectionStorage.selectionChangeEvent.addListener(this.onUnifiedSelectionChanged);\n\n    // stop imodel from syncing tool selection with hilited list - we want to override that behavior\n    this._imodelAccess.hiliteSet.wantSyncWithSelectionSet = false;\n    this.applyCurrentHiliteSet({ activeSelectionAction: \"clearAll\" });\n  }\n\n  public dispose() {\n    this._cancelOngoingChanges.next();\n    this._imodelListenerDisposeFunc();\n    this._unifiedSelectionListenerDisposeFunc();\n    if (!this._hasCustomCachingHiliteSetProvider) {\n      this._cachingHiliteSetProvider.dispose();\n    }\n  }\n\n  /** Temporarily suspends tool selection synchronization until the returned `IDisposable` is disposed. */\n  public suspendIModelToolSelectionSync() {\n    const wasSuspended = this._isSuspended;\n    this._isSuspended = true;\n    return {\n      dispose: () => (this._isSuspended = wasSuspended),\n    };\n  }\n\n  private handleUnifiedSelectionChange(changeType: StorageSelectionChangeType, selectables: Selectables, source: string) {\n    if (changeType === \"clear\" || changeType === \"replace\") {\n      this.applyCurrentHiliteSet({ activeSelectionAction: changeType === \"replace\" && source === \"Tool\" ? \"clearHilited\" : \"clearAll\" });\n      return;\n    }\n\n    if (changeType === \"add\") {\n      from(this._hiliteSetProvider.getHiliteSet({ selectables }))\n        .pipe(takeUntil(this._cancelOngoingChanges))\n        .subscribe({\n          next: (set) => {\n            this.addHiliteSet(set);\n          },\n        });\n      return;\n    }\n\n    from(this._hiliteSetProvider.getHiliteSet({ selectables }))\n      .pipe(takeUntil(this._cancelOngoingChanges))\n      .subscribe({\n        next: (set) => {\n          this.removeHiliteSet(set);\n        },\n        complete: () => {\n          this.applyCurrentHiliteSet({ activeSelectionAction: \"keep\" });\n        },\n      });\n  }\n\n  private onUnifiedSelectionChanged = (args: StorageSelectionChangeEventArgs) => {\n    // iModels are only interested in top-level selection changes\n    if (args.imodelKey !== this._imodelAccess.key || args.level !== 0) {\n      return;\n    }\n\n    if (args.changeType === \"replace\" || args.changeType === \"clear\") {\n      this._cancelOngoingChanges.next();\n    }\n    this.handleUnifiedSelectionChange(args.changeType, args.selectables, args.source);\n  };\n\n  private applyCurrentHiliteSet({ activeSelectionAction }: { activeSelectionAction: \"clearAll\" | \"clearHilited\" | \"keep\" }) {\n    if (activeSelectionAction !== \"keep\") {\n      using(this.suspendIModelToolSelectionSync(), (_) => {\n        this._imodelAccess.hiliteSet.clear();\n        if (activeSelectionAction === \"clearAll\") {\n          this._imodelAccess.selectionSet.emptyAll();\n        }\n      });\n    }\n\n    from(this._cachingHiliteSetProvider.getHiliteSet({ imodelKey: this._imodelAccess.key }))\n      .pipe(takeUntil(this._cancelOngoingChanges))\n      .subscribe({\n        next: (ids) => {\n          this.addHiliteSet(ids);\n        },\n      });\n  }\n\n  private addHiliteSet(set: HiliteSet) {\n    using(this.suspendIModelToolSelectionSync(), (_) => {\n      if (set.models && set.models.length) {\n        this._imodelAccess.hiliteSet.models.addIds(set.models);\n      }\n      if (set.subCategories && set.subCategories.length) {\n        this._imodelAccess.hiliteSet.subcategories.addIds(set.subCategories);\n      }\n      if (set.elements && set.elements.length) {\n        this._imodelAccess.hiliteSet.elements.addIds(set.elements);\n        this._imodelAccess.selectionSet.add(set.elements);\n      }\n    });\n  }\n\n  private removeHiliteSet(set: HiliteSet) {\n    using(this.suspendIModelToolSelectionSync(), (_) => {\n      if (set.models.length) {\n        this._imodelAccess.hiliteSet.models.deleteIds(set.models);\n      }\n      if (set.subCategories.length) {\n        this._imodelAccess.hiliteSet.subcategories.deleteIds(set.subCategories);\n      }\n      if (set.elements.length) {\n        this._imodelAccess.hiliteSet.elements.deleteIds(set.elements);\n        this._imodelAccess.selectionSet.remove(set.elements);\n      }\n    });\n  }\n\n  private onIModelSelectionChanged = async (event?: CoreSelectionSetEventUnsafe): Promise<void> => {\n    if (this._isSuspended) {\n      return;\n    }\n\n    if (CoreSelectionSetEventType.Clear === event!.type) {\n      this._selectionStorage.clearSelection({ imodelKey: this._imodelAccess.key, source: this._selectionSourceName });\n      return;\n    }\n\n    const elementIds = this.getSelectionSetChangeIds(event!);\n    const scopedSelection = computeSelection({ queryExecutor: this._imodelAccess, elementIds, scope: this._activeScopeProvider() });\n    await this.handleIModelSelectionChange(event!.type, scopedSelection);\n  };\n\n  private async handleIModelSelectionChange(type: CoreSelectionSetEventType, iterator: AsyncIterableIterator<SelectableInstanceKey>) {\n    if (type === CoreSelectionSetEventType.Remove) {\n      for await (const selectable of iterator) {\n        this._selectionStorage.removeFromSelection({ imodelKey: this._imodelAccess.key, source: this._selectionSourceName, selectables: [selectable] });\n      }\n      return;\n    }\n\n    if (type === CoreSelectionSetEventType.Replace) {\n      this._selectionStorage.clearSelection({ imodelKey: this._imodelAccess.key, source: this._selectionSourceName });\n    }\n\n    for await (const selectable of iterator) {\n      this._selectionStorage.addToSelection({ imodelKey: this._imodelAccess.key, source: this._selectionSourceName, selectables: [selectable] });\n    }\n  }\n\n  private getSelectionSetChangeIds(event: CoreSelectionSetEventUnsafe): string[] {\n    switch (event.type) {\n      case CoreSelectionSetEventType.Add:\n      case CoreSelectionSetEventType.Replace:\n        assert(!!event.added);\n        return this.idArgToIds(event.added);\n      default:\n        assert(!!event.removed);\n        return this.idArgToIds(event.removed);\n    }\n  }\n\n  private idArgToIds(ids: Id64Arg): string[] {\n    if (typeof ids === \"string\") {\n      return [ids];\n    }\n    if (ids instanceof Set) {\n      return [...ids];\n    }\n    return ids;\n  }\n}\n"]}

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

@@ -0,0 +1,37 @@
+import { ECSchemaProvider, ECSqlQueryExecutor } from "@itwin/presentation-shared";
+import { Selectables } from "./Selectable";
+/**
+ * A set of model, subcategory and element ids that can be used for specifying hilite.
+ * @see https://www.itwinjs.org/reference/core-frontend/selectionset/hiliteset/
+ * @beta
+ */
+export interface HiliteSet {
+    models: string[];
+    subCategories: string[];
+    elements: string[];
+}
+/**
+ * Props for creating a `HiliteSetProvider` instance.
+ * @internal Not exported through barrel, but used in public API as an argument. May be supplemented with optional attributes any time.
+ */
+export interface HiliteSetProviderProps {
+    imodelAccess: ECSchemaProvider & ECSqlQueryExecutor;
+}
+/**
+ * Defines return value of `createHiliteSetProvider`.
+ *
+ * @beta Used in public API as a return value. Not expected to be created / extended by package
+ * consumers, may be supplemented with required attributes any time.
+ */
+export interface HiliteSetProvider {
+    /** Get the current hilite set iterator for the specified imodel */
+    getHiliteSet(props: {
+        selectables: Selectables;
+    }): AsyncIterableIterator<HiliteSet>;
+}
+/**
+ * Creates a hilite set provider that returns a `HiliteSet` for given selectables.
+ * @beta
+ */
+export declare function createHiliteSetProvider(props: HiliteSetProviderProps): HiliteSetProvider;
+//# sourceMappingURL=HiliteSetProvider.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"HiliteSetProvider.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/HiliteSetProvider.ts"],"names":[],"mappings":"AAOA,OAAO,EAAM,gBAAgB,EAAgB,kBAAkB,EAAsB,MAAM,4BAA4B,CAAC;AACxH,OAAO,EAAyB,WAAW,EAAE,MAAM,cAAc,CAAC;AAKlE;;;;GAIG;AACH,MAAM,WAAW,SAAS;IACxB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB;AAED;;;GAGG;AACH,MAAM,WAAW,sBAAsB;IACrC,YAAY,EAAE,gBAAgB,GAAG,kBAAkB,CAAC;CACrD;AAED;;;;;GAKG;AACH,MAAM,WAAW,iBAAiB;IAChC,mEAAmE;IACnE,YAAY,CAAC,KAAK,EAAE;QAAE,WAAW,EAAE,WAAW,CAAA;KAAE,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC;CACrF;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,sBAAsB,GAAG,iBAAiB,CAExF"}

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

@@ -0,0 +1,344 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createHiliteSetProvider = void 0;
+const rxjs_1 = require("rxjs");
+const rxjs_for_await_1 = require("rxjs-for-await");
+const presentation_shared_1 = require("@itwin/presentation-shared");
+const Utils_1 = require("./Utils");
+const HILITE_SET_EMIT_FREQUENCY = 20;
+/**
+ * Creates a hilite set provider that returns a `HiliteSet` for given selectables.
+ * @beta
+ */
+function createHiliteSetProvider(props) {
+    return new HiliteSetProviderImpl(props);
+}
+exports.createHiliteSetProvider = createHiliteSetProvider;
+class HiliteSetProviderImpl {
+    _imodelAccess;
+    // Map between a class name and its type
+    _classRelationCache;
+    constructor(props) {
+        this._imodelAccess = props.imodelAccess;
+        this._classRelationCache = new Map();
+    }
+    /**
+     * Get hilite set iterator for supplied `Selectables`.
+     */
+    getHiliteSet(props) {
+        const obs = this.getHiliteSetObservable(props);
+        return (0, rxjs_for_await_1.eachValueFrom)(obs);
+    }
+    /**
+     * Returns a "hot" observable of hilite sets.
+     */
+    getHiliteSetObservable({ selectables }) {
+        const instancesByType = this.getInstancesByType(selectables);
+        const observables = {
+            models: this.getHilitedModels(instancesByType),
+            subCategories: this.getHilitedSubCategories(instancesByType),
+            elements: this.getHilitedElements(instancesByType),
+        };
+        let hiliteSet = { models: [], subCategories: [], elements: [] };
+        let lastEmitTime = performance.now();
+        const subject = new rxjs_1.Subject();
+        const subscriptions = ["models", "subCategories", "elements"].map((key) => observables[key].subscribe({
+            next(val) {
+                hiliteSet[key].push(val);
+                if (performance.now() - lastEmitTime < HILITE_SET_EMIT_FREQUENCY) {
+                    return;
+                }
+                subject.next(hiliteSet);
+                hiliteSet = { models: [], subCategories: [], elements: [] };
+                lastEmitTime = performance.now();
+            },
+            complete() {
+                observables[key] = undefined;
+                if (observables.models || observables.subCategories || observables.elements) {
+                    return;
+                }
+                // Emit last batch before completing the observable.
+                if (hiliteSet.elements.length || hiliteSet.models.length || hiliteSet.subCategories.length) {
+                    subject.next(hiliteSet);
+                }
+                subject.complete();
+            },
+            error(err) {
+                subscriptions.forEach((x) => x.unsubscribe());
+                subscriptions.length = 0;
+                subject.error(err);
+            },
+        }));
+        return subject;
+    }
+    async getType(key) {
+        const cachedType = this._classRelationCache.get(key.className.replace(".", ":"));
+        if (cachedType) {
+            return cachedType;
+        }
+        const promise = this.getTypeImpl(key).then((res) => {
+            // Update the cache with the result of the promise.
+            this._classRelationCache.set(key.className, res);
+            return res;
+        });
+        // Add the promise to cache to prevent `getTypeImpl` being called multiple times.
+        this._classRelationCache.set(key.className, promise);
+        return promise;
+    }
+    async getTypeImpl(key) {
+        const keyClass = await this.getClass(key.className);
+        return ((keyClass &&
+            ((await this.checkType(keyClass, "BisCore", "Subject", "subject")) ??
+                (await this.checkType(keyClass, "BisCore", "Model", "model")) ??
+                (await this.checkType(keyClass, "BisCore", "Category", "category")) ??
+                (await this.checkType(keyClass, "BisCore", "SubCategory", "subCategory")) ??
+                (await this.checkType(keyClass, "Functional", "FunctionalElement", "functionalElement")) ??
+                (await this.checkType(keyClass, "BisCore", "GroupInformationElement", "groupInformationElement")) ??
+                (await this.checkType(keyClass, "BisCore", "GeometricElement", "geometricElement")) ??
+                (await this.checkType(keyClass, "BisCore", "Element", "element")))) ??
+            "unknown");
+    }
+    async checkType(keyClass, schemaName, className, type) {
+        return (await keyClass.is(className, schemaName)) ? type : undefined;
+    }
+    getInstancesByType(selectables) {
+        const keyTypeObs = (0, rxjs_1.merge)((0, rxjs_1.from)(selectables.custom.values()).pipe((0, rxjs_1.mergeMap)((selectable) => selectable.loadInstanceKeys())), (0, rxjs_1.from)(selectables.instanceKeys).pipe((0, rxjs_1.mergeMap)(([className, idSet]) => (0, rxjs_1.from)(idSet).pipe((0, rxjs_1.map)((id) => ({ className, id })))))).pipe(
+        // Get types for each instance key
+        (0, rxjs_1.mergeMap)((instanceKey) => (0, rxjs_1.from)(this.getType(instanceKey)).pipe((0, rxjs_1.map)((instanceIdType) => ({ instanceId: instanceKey.id, instanceIdType })))), 
+        // Cache the results
+        (0, rxjs_1.shareReplay)());
+        return Object.fromEntries(INSTANCE_TYPES.map((type) => [
+            type,
+            keyTypeObs.pipe((0, rxjs_1.filter)(({ instanceIdType }) => instanceIdType === type), (0, rxjs_1.map)(({ instanceId }) => instanceId), unique()),
+        ]));
+    }
+    getHilitedModels(instancesByType) {
+        return (0, rxjs_1.forkJoin)({
+            modelKeys: instancesByType.model.pipe((0, rxjs_1.toArray)()),
+            subjectKeys: instancesByType.subject.pipe((0, rxjs_1.toArray)()),
+        }).pipe((0, rxjs_1.mergeMap)(({ modelKeys, subjectKeys }) => {
+            if (!modelKeys.length && !subjectKeys.length) {
+                return rxjs_1.EMPTY;
+            }
+            const bindings = [];
+            const ctes = [
+                `
+            ChildSubjects(ECInstanceId, JsonProperties) AS (
+              SELECT ECInstanceId, JsonProperties
+              FROM BisCore.Subject
+              WHERE ${(0, Utils_1.formIdBindings)("ECInstanceId", subjectKeys, bindings)}
+              UNION ALL
+              SELECT r.ECInstanceId, r.JsonProperties
+              FROM ChildSubjects s
+              JOIN BisCore.Subject r ON r.Parent.Id = s.ECInstanceId
+            )
+          `,
+                `
+            Models(ECInstanceId) AS (
+              SELECT s.ECInstanceId AS ECInstanceId
+              FROM BisCore.Model s
+              WHERE ${(0, Utils_1.formIdBindings)("ECInstanceId", modelKeys, bindings)}
+            )
+          `,
+            ];
+            const ecsql = [
+                `
+            SELECT r.ECInstanceId AS ECInstanceId
+            FROM ChildSubjects s
+            JOIN BisCore.PhysicalPartition r ON r.Parent.Id = s.ECInstanceId OR json_extract(s.JsonProperties,'$.Subject.Model.TargetPartition') = printf('0x%x', r.ECInstanceId)
+          `,
+                `
+            SELECT ECInstanceId FROM Models
+          `,
+            ].join(" UNION ");
+            return (0, rxjs_1.from)(this._imodelAccess.createQueryReader({ ctes, ecsql, bindings }));
+        }), (0, rxjs_1.map)((row) => row.ECInstanceId));
+    }
+    getHilitedSubCategories(instancesByType) {
+        return (0, rxjs_1.forkJoin)({
+            subCategoryKeys: instancesByType.subCategory.pipe((0, rxjs_1.toArray)()),
+            categoryKeys: instancesByType.category.pipe((0, rxjs_1.toArray)()),
+        }).pipe((0, rxjs_1.mergeMap)(({ subCategoryKeys, categoryKeys }) => {
+            if (!subCategoryKeys.length && !categoryKeys.length) {
+                return rxjs_1.EMPTY;
+            }
+            const bindings = [];
+            const ctes = [
+                `
+            CategorySubCategories(ECInstanceId) AS (
+              SELECT r.ECInstanceId AS ECInstanceId
+              FROM BisCore.Category s
+              JOIN BisCore.SubCategory r ON r.Parent.Id = s.ECInstanceId
+              WHERE ${(0, Utils_1.formIdBindings)("s.ECInstanceId", categoryKeys, bindings)}
+            )
+          `,
+                `
+            SubCategories(ECInstanceId) AS (
+              SELECT s.ECInstanceId AS ECInstanceId
+              FROM BisCore.SubCategory s
+              WHERE ${(0, Utils_1.formIdBindings)("s.ECInstanceId", subCategoryKeys, bindings)}
+            )
+          `,
+            ];
+            const ecsql = [`SELECT ECInstanceId FROM CategorySubCategories`, `SELECT ECInstanceId FROM SubCategories`].join(" UNION ");
+            return (0, rxjs_1.from)(this._imodelAccess.createQueryReader({ ctes, ecsql, bindings }));
+        }), (0, rxjs_1.map)((row) => row.ECInstanceId));
+    }
+    getHilitedElements(instancesByType) {
+        return (0, rxjs_1.forkJoin)({
+            groupInformationElementKeys: instancesByType.groupInformationElement.pipe((0, rxjs_1.toArray)()),
+            geometricElementKeys: instancesByType.geometricElement.pipe((0, rxjs_1.toArray)()),
+            functionalElements: instancesByType.functionalElement.pipe((0, rxjs_1.toArray)()),
+            elementKeys: instancesByType.element.pipe((0, rxjs_1.toArray)()),
+        }).pipe((0, rxjs_1.mergeMap)(({ groupInformationElementKeys, geometricElementKeys, functionalElements, elementKeys }) => {
+            const hasFunctionalElements = !!functionalElements.length;
+            if (!groupInformationElementKeys.length && !geometricElementKeys.length && !elementKeys.length && !hasFunctionalElements) {
+                return rxjs_1.EMPTY;
+            }
+            const bindings = [];
+            const ctes = [
+                ...(hasFunctionalElements ? this.getHilitedFunctionalElementsQueryCTEs(functionalElements, bindings) : []),
+                `
+            GroupMembers(ECInstanceId, ECClassId) AS (
+              SELECT TargetECInstanceId, TargetECClassId
+              FROM BisCore.ElementGroupsMembers
+              WHERE ${(0, Utils_1.formIdBindings)("SourceECInstanceId", groupInformationElementKeys, bindings)}
+            )
+          `,
+                `
+            GroupGeometricElements(ECInstanceId, ECClassId) AS (
+              SELECT ECInstanceId, ECClassId FROM GroupMembers
+              UNION ALL
+              SELECT r.ECInstanceId, r.ECClassId
+              FROM GroupGeometricElements s
+              JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
+            )
+          `,
+                `
+            ElementGeometricElements(ECInstanceId, ECClassId) AS (
+              SELECT ECInstanceId, ECClassId
+              FROM BisCore.Element
+              WHERE ${(0, Utils_1.formIdBindings)("ECInstanceId", elementKeys, bindings)}
+              UNION ALL
+              SELECT r.ECInstanceId, r.ECClassId
+              FROM ElementGeometricElements s
+              JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
+            )
+          `,
+                `
+            GeometricElementGeometricElements(ECInstanceId, ECClassId) AS (
+              SELECT ECInstanceId, ECClassId
+              FROM BisCore.GeometricElement
+              WHERE ${(0, Utils_1.formIdBindings)("ECInstanceId", geometricElementKeys, bindings)}
+              UNION ALL
+              SELECT r.ECInstanceId, r.ECClassId
+              FROM GeometricElementGeometricElements s
+              JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
+            )
+          `,
+            ];
+            const ecsql = [
+                ...(hasFunctionalElements ? ["SELECT ECInstanceId FROM FunctionalElementChildGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)"] : []),
+                "SELECT ECInstanceId FROM GeometricElementGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)",
+                "SELECT ECInstanceId FROM GroupGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)",
+                "SELECT ECInstanceId FROM ElementGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)",
+            ].join(" UNION ");
+            return (0, rxjs_1.from)(this._imodelAccess.createQueryReader({ ctes, ecsql, bindings }));
+        }), (0, rxjs_1.map)((row) => row.ECInstanceId));
+    }
+    getHilitedFunctionalElementsQueryCTEs(functionalElements, bindings) {
+        return [
+            `
+        ChildFunctionalElements(ECInstanceId, ECClassId) AS (
+          SELECT ECInstanceId, ECClassId
+          FROM Functional.FunctionalElement
+          WHERE ${(0, Utils_1.formIdBindings)("ECInstanceId", functionalElements, bindings)}
+          UNION ALL
+          SELECT r.ECInstanceId, r.ECClassId
+          FROM ChildFunctionalElements s
+          JOIN Functional.FunctionalElement r ON r.Parent.Id = s.ECInstanceId
+        )
+      `,
+            `
+        PhysicalElements(ECInstanceId, ECClassId) AS (
+          SELECT r.SourceECInstanceId, r.SourceECClassId
+          FROM ChildFunctionalElements s
+          JOIN Functional.PhysicalElementFulfillsFunction r ON r.TargetECInstanceId = s.ECInstanceId
+        )
+      `,
+            `
+        DrawingGraphicElements(ECInstanceId, ECClassId) AS (
+          SELECT r.SourceECInstanceId, r.SourceECClassId
+          FROM ChildFunctionalElements s
+          JOIN Functional.DrawingGraphicRepresentsFunctionalElement r ON r.TargetECInstanceId = s.ECInstanceId
+        )
+      `,
+            `
+        PhysicalElementGeometricElements(ECInstanceId, ECClassId) AS (
+          SELECT ECInstanceId, ECClassId FROM PhysicalElements
+          UNION ALL
+          SELECT r.ECInstanceId, r.ECClassId
+          FROM PhysicalElementGeometricElements s
+          JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
+        )
+      `,
+            `
+        DrawingGraphicElementGeometricElements(ECInstanceId, ECClassId) AS (
+          SELECT ECInstanceId, ECClassId FROM DrawingGraphicElements
+          UNION ALL
+          SELECT r.ECInstanceId, r.ECClassId
+          FROM DrawingGraphicElementGeometricElements s
+          JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
+        )
+      `,
+            `
+        FunctionalElementChildGeometricElements(ECInstanceId, ECClassId) AS (
+          SELECT ECInstanceId, ECClassId FROM PhysicalElementGeometricElements
+          UNION
+          SELECT ECInstanceId, ECClassId FROM DrawingGraphicElementGeometricElements
+        )
+      `,
+        ];
+    }
+    async getClass(fullClassName) {
+        const { schemaName, className } = (0, presentation_shared_1.parseFullClassName)(fullClassName);
+        const schema = await this._imodelAccess.getSchema(schemaName);
+        if (!schema) {
+            return undefined;
+        }
+        const schemaClass = await schema.getClass(className);
+        if (!schemaClass) {
+            return undefined;
+        }
+        return schemaClass;
+    }
+}
+const INSTANCE_TYPES = [
+    "subject",
+    "model",
+    "category",
+    "subCategory",
+    "functionalElement",
+    "groupInformationElement",
+    "geometricElement",
+    "element",
+    "unknown",
+];
+function unique() {
+    return function (obs) {
+        return obs.pipe((0, rxjs_1.scan)((acc, val) => {
+            if (acc.set.has(val)) {
+                delete acc.val;
+                return acc;
+            }
+            acc.set.add(val);
+            acc.val = val;
+            return acc;
+        }, { set: new Set() }), (0, rxjs_1.map)(({ val }) => val), (0, rxjs_1.filter)((x) => !!x));
+    };
+}
+//# sourceMappingURL=HiliteSetProvider.js.map