@itwin/unified-selection 1.8.6 → 2.0.0-alpha.1

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

@@ -1,382 +0,0 @@
-"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 = createHiliteSetProvider;
-const rxjs_1 = require("rxjs");
-const rxjs_for_await_1 = require("rxjs-for-await");
-const core_bentley_1 = require("@itwin/core-bentley");
-const presentation_shared_1 = require("@itwin/presentation-shared");
-const Selectable_js_1 = require("./Selectable.js");
-const Utils_js_1 = require("./Utils.js");
-const HILITE_SET_EMIT_FREQUENCY = 20;
-/**
- * Creates a hilite set provider that returns a `HiliteSet` for given selectables.
- * @public
- */
-function createHiliteSetProvider(props) {
-    return new HiliteSetProviderImpl(props);
-}
-class HiliteSetProviderImpl {
-    _imodelAccess;
-    // Map between a class name and its type
-    _classRelationCache;
-    #componentId;
-    #componentName;
-    constructor(props) {
-        this._imodelAccess = props.imodelAccess;
-        this._classRelationCache = new Map();
-        this.#componentId = core_bentley_1.Guid.createValue();
-        this.#componentName = "HiliteSetProvider";
-    }
-    /**
-     * 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) {
-        if (key.className === Selectable_js_1.TRANSIENT_ELEMENT_CLASSNAME) {
-            return "transient";
-        }
-        const normalizedClassName = (0, presentation_shared_1.normalizeFullClassName)(key.className);
-        const cachedType = this._classRelationCache.get(normalizedClassName);
-        if (cachedType) {
-            return cachedType;
-        }
-        const promise = this.getTypeImpl(normalizedClassName).then((res) => {
-            // Update the cache with the result of the promise.
-            this._classRelationCache.set(normalizedClassName, res);
-            return res;
-        });
-        // Add the promise to cache to prevent `getTypeImpl` being called multiple times.
-        this._classRelationCache.set(normalizedClassName, promise);
-        return promise;
-    }
-    async getTypeImpl(fullClassName) {
-        return ((await this.checkType(fullClassName, "BisCore.Subject", "subject")) ??
-            (await this.checkType(fullClassName, "BisCore.Model", "model")) ??
-            (await this.checkType(fullClassName, "BisCore.Category", "category")) ??
-            (await this.checkType(fullClassName, "BisCore.SubCategory", "subCategory")) ??
-            (await this.checkType(fullClassName, "Functional.FunctionalElement", "functionalElement")) ??
-            (await this.checkType(fullClassName, "BisCore.GroupInformationElement", "groupInformationElement")) ??
-            (await this.checkType(fullClassName, "BisCore.GeometricElement", "geometricElement")) ??
-            (await this.checkType(fullClassName, "BisCore.Element", "element")) ??
-            "unknown");
-    }
-    async checkType(keyClassName, checkClassName, type) {
-        try {
-            const res = this._imodelAccess.classDerivesFrom(keyClassName, checkClassName);
-            const isOfType = typeof res === "boolean" ? res : await res;
-            return isOfType ? type : undefined;
-        }
-        catch (e) {
-            // we may be checking against a non-existing schema (e.g. Functional), in which case we should
-            // return undefined instead of throwing an error
-            if (e instanceof Error && e.message.match(/Schema "[\w\d_]+" not found/)) {
-                return undefined;
-            }
-            throw e;
-        }
-    }
-    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((0, Utils_js_1.releaseMainThreadOnItemsCount)(500), 
-        // 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_js_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_js_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)(executeQuery({
-                queryExecutor: this._imodelAccess,
-                query: { ctes, ecsql, bindings },
-                config: { restartToken: `${this.#componentName}/${this.#componentId}/models/${core_bentley_1.Guid.createValue()}` },
-            }));
-        }));
-    }
-    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_js_1.formIdBindings)("s.ECInstanceId", categoryKeys, bindings)}
-            )
-          `,
-                `
-            SubCategories(ECInstanceId) AS (
-              SELECT s.ECInstanceId AS ECInstanceId
-              FROM BisCore.SubCategory s
-              WHERE ${(0, Utils_js_1.formIdBindings)("s.ECInstanceId", subCategoryKeys, bindings)}
-            )
-          `,
-            ];
-            const ecsql = [`SELECT ECInstanceId FROM CategorySubCategories`, `SELECT ECInstanceId FROM SubCategories`].join(" UNION ");
-            return (0, rxjs_1.from)(executeQuery({
-                queryExecutor: this._imodelAccess,
-                query: { ctes, ecsql, bindings },
-                config: {
-                    restartToken: `${this.#componentName}/${this.#componentId}/sub-categories/${core_bentley_1.Guid.createValue()}`,
-                },
-            }));
-        }));
-    }
-    getHilitedElements(instancesByType) {
-        return (0, rxjs_1.forkJoin)({
-            groupInformationElementIds: instancesByType.groupInformationElement.pipe((0, rxjs_1.toArray)()),
-            geometricElementIds: instancesByType.geometricElement.pipe((0, rxjs_1.toArray)()),
-            functionalElementIds: instancesByType.functionalElement.pipe((0, rxjs_1.toArray)()),
-            elementIds: instancesByType.element.pipe((0, rxjs_1.toArray)()),
-            transientIds: instancesByType.transient.pipe((0, rxjs_1.toArray)()),
-        }).pipe((0, rxjs_1.mergeMap)(({ groupInformationElementIds, geometricElementIds, functionalElementIds, elementIds, transientIds }) => {
-            const hasFunctionalElementIds = !!functionalElementIds.length;
-            const hasPersistentElementIds = hasFunctionalElementIds ||
-                !!(groupInformationElementIds.length || geometricElementIds.length || elementIds.length);
-            const hasTransientElementIds = !!transientIds.length;
-            if (!hasPersistentElementIds && !hasTransientElementIds) {
-                return rxjs_1.EMPTY;
-            }
-            if (!hasPersistentElementIds) {
-                return (0, rxjs_1.from)(transientIds);
-            }
-            const bindings = [];
-            const ctes = [
-                ...(hasFunctionalElementIds
-                    ? this.getHilitedFunctionalElementsQueryCTEs(functionalElementIds, bindings)
-                    : []),
-                `
-              GroupMembers(ECInstanceId, ECClassId) AS (
-                SELECT TargetECInstanceId, TargetECClassId
-                FROM BisCore.ElementGroupsMembers
-                WHERE ${(0, Utils_js_1.formIdBindings)("SourceECInstanceId", groupInformationElementIds, 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_js_1.formIdBindings)("ECInstanceId", elementIds, 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_js_1.formIdBindings)("ECInstanceId", geometricElementIds, bindings)}
-                UNION ALL
-                SELECT r.ECInstanceId, r.ECClassId
-                FROM GeometricElementGeometricElements s
-                JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
-              )
-            `,
-            ];
-            const ecsql = [
-                ...(hasFunctionalElementIds
-                    ? [
-                        "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.merge)((0, rxjs_1.from)(executeQuery({
-                queryExecutor: this._imodelAccess,
-                query: { ctes, ecsql, bindings },
-                config: { restartToken: `${this.#componentName}/${this.#componentId}/elements/${core_bentley_1.Guid.createValue()}` },
-            })), (0, rxjs_1.from)(transientIds));
-        }));
-    }
-    getHilitedFunctionalElementsQueryCTEs(functionalElements, bindings) {
-        return [
-            `
-        ChildFunctionalElements(ECInstanceId, ECClassId) AS (
-          SELECT ECInstanceId, ECClassId
-          FROM Functional.FunctionalElement
-          WHERE ${(0, Utils_js_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
-        )
-      `,
-        ];
-    }
-}
-const INSTANCE_TYPES = [
-    "subject",
-    "model",
-    "category",
-    "subCategory",
-    "functionalElement",
-    "groupInformationElement",
-    "geometricElement",
-    "element",
-    "transient",
-    "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));
-    };
-}
-async function* executeQuery(props) {
-    yield* (0, Utils_js_1.genericExecuteQuery)({ ...props, parseQueryRow: (row) => row.ECInstanceId });
-}
-//# sourceMappingURL=HiliteSetProvider.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"HiliteSetProvider.js","sourceRoot":"","sources":["../../../src/unified-selection/HiliteSetProvider.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,+BAac;AACd,mDAA+C;AAC/C,sDAAmE;AACnE,oEAQoC;AACpC,mDAAkG;AAClG,yCAAgG;AAEhG,MAAM,yBAAyB,GAAG,EAAE,CAAC;AA0CrC;;;GAGG;AACH,iCAAwC,KAA6B;IACnE,OAAO,IAAI,qBAAqB,CAAC,KAAK,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,qBAAqB;IACjB,aAAa,CAAiD;IACtE,wCAAwC;IAChC,mBAAmB,CAAwD;IACnF,YAAY,CAAa;IACzB,cAAc,CAAS;IAEvB,YAAY,KAA6B;QACvC,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,mBAAmB,GAAG,IAAI,GAAG,EAAE,CAAC;QACrC,IAAI,CAAC,YAAY,GAAG,mBAAI,CAAC,WAAW,EAAE,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,mBAAmB,CAAC;IAC5C,CAAC;IAED;;OAEG;IACI,YAAY,CAAC,KAAmC;QACrD,MAAM,GAAG,GAAG,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QAC/C,OAAO,IAAA,8BAAa,EAAC,GAAG,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACK,sBAAsB,CAAC,EAAE,WAAW,EAAgC;QAC1E,MAAM,eAAe,GAAG,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;QAC7D,MAAM,WAAW,GAAsD;YACrE,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC;YAC9C,aAAa,EAAE,IAAI,CAAC,uBAAuB,CAAC,eAAe,CAAC;YAC5D,QAAQ,EAAE,IAAI,CAAC,kBAAkB,CAAC,eAAe,CAAC;SACnD,CAAC;QAEF,IAAI,SAAS,GAAc,EAAE,MAAM,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;QAC3E,IAAI,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,IAAI,cAAO,EAAa,CAAC;QACzC,MAAM,aAAa,GAAI,CAAC,QAAQ,EAAE,eAAe,EAAE,UAAU,CAAW,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACnF,WAAW,CAAC,GAAG,CAAE,CAAC,SAAS,CAAC;YAC1B,IAAI,CAAC,GAAG;gBACN,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACzB,IAAI,WAAW,CAAC,GAAG,EAAE,GAAG,YAAY,GAAG,yBAAyB,EAAE,CAAC;oBACjE,OAAO;gBACT,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACxB,SAAS,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;gBAC5D,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YACnC,CAAC;YACD,QAAQ;gBACN,WAAW,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;gBAC7B,IAAI,WAAW,CAAC,MAAM,IAAI,WAAW,CAAC,aAAa,IAAI,WAAW,CAAC,QAAQ,EAAE,CAAC;oBAC5E,OAAO;gBACT,CAAC;gBACD,oDAAoD;gBACpD,IAAI,SAAS,CAAC,QAAQ,CAAC,MAAM,IAAI,SAAS,CAAC,MAAM,CAAC,MAAM,IAAI,SAAS,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;oBAC3F,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC1B,CAAC;gBACD,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,CAAC;YACD,KAAK,CAAC,GAAG;gBACP,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;gBAC9C,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC;gBACzB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;SACF,CAAC,CACH,CAAC;QAEF,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,GAA0B;QAC9C,IAAI,GAAG,CAAC,SAAS,KAAK,2CAA2B,EAAE,CAAC;YAClD,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,MAAM,mBAAmB,GAAG,IAAA,4CAAsB,EAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAClE,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACrE,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE;YACjE,mDAAmD;YACnD,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC;YACvD,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,iFAAiF;QACjF,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;QAC3D,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,aAAqB;QAC7C,OAAO,CACL,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,iBAAiB,EAAE,SAAS,CAAC,CAAC;YACnE,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;YAC/D,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,kBAAkB,EAAE,UAAU,CAAC,CAAC;YACrE,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,qBAAqB,EAAE,aAAa,CAAC,CAAC;YAC3E,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,8BAA8B,EAAE,mBAAmB,CAAC,CAAC;YAC1F,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,iCAAiC,EAAE,yBAAyB,CAAC,CAAC;YACnG,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,0BAA0B,EAAE,kBAAkB,CAAC,CAAC;YACrF,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,iBAAiB,EAAE,SAAS,CAAC,CAAC;YACnE,SAAS,CACV,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,YAAoB,EAAE,cAAsB,EAAE,IAAoB;QACxF,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;YAC9E,MAAM,QAAQ,GAAG,OAAO,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC;YAC5D,OAAO,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QACrC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,8FAA8F;YAC9F,gDAAgD;YAChD,IAAI,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,6BAA6B,CAAC,EAAE,CAAC;gBACzE,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,WAAwB;QACjD,MAAM,UAAU,GAAG,IAAA,YAAK,EACtB,IAAA,WAAI,EAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,IAAA,eAAQ,EAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,gBAAgB,EAAE,CAAC,CAAC,EAC/F,IAAA,WAAI,EAAC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CACjC,IAAA,eAAQ,EAAC,CAAC,CAAC,SAAS,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,IAAA,WAAI,EAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAA,UAAG,EAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CACrF,CACF,CAAC,IAAI,CACJ,IAAA,wCAA6B,EAAC,GAAG,CAAC;QAClC,kCAAkC;QAClC,IAAA,eAAQ,EAAC,CAAC,WAAW,EAAE,EAAE,CACvB,IAAA,WAAI,EAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAA,UAAG,EAAC,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,WAAW,CAAC,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,CAChH;QACD,oBAAoB;QACpB,IAAA,kBAAW,GAAE,CACd,CAAC;QAEF,OAAO,MAAM,CAAC,WAAW,CACvB,cAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;YAC3B,IAAI;YACJ,UAAU,CAAC,IAAI,CACb,IAAA,aAAM,EAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,cAAc,KAAK,IAAI,CAAC,EACvD,IAAA,UAAG,EAAC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,EACnC,MAAM,EAAE,CACT;SACF,CAAC,CACgB,CAAC;IACvB,CAAC;IAEO,gBAAgB,CAAC,eAAgC;QACvD,OAAO,IAAA,eAAQ,EAAC;YACd,SAAS,EAAE,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,IAAA,cAAO,GAAE,CAAC;YAChD,WAAW,EAAE,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,IAAA,cAAO,GAAE,CAAC;SACrD,CAAC,CAAC,IAAI,CACL,IAAA,eAAQ,EAAC,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE;YACtC,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;gBAC7C,OAAO,YAAK,CAAC;YACf,CAAC;YAED,MAAM,QAAQ,GAAmB,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG;gBACX;;;;sBAIY,IAAA,yBAAc,EAAC,cAAc,EAAE,WAAW,EAAE,QAAQ,CAAC;;;;;;WAMhE;gBACD;;;;sBAIY,IAAA,yBAAc,EAAC,cAAc,EAAE,SAAS,EAAE,QAAQ,CAAC;;WAE9D;aACF,CAAC;YACF,MAAM,KAAK,GAAG;gBACZ;;;;WAIC;gBACD;;WAEC;aACF,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClB,OAAO,IAAA,WAAI,EACT,YAAY,CAAC;gBACX,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;gBAChC,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,WAAW,mBAAI,CAAC,WAAW,EAAE,EAAE,EAAE;aACrG,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,uBAAuB,CAAC,eAAgC;QAC9D,OAAO,IAAA,eAAQ,EAAC;YACd,eAAe,EAAE,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,IAAA,cAAO,GAAE,CAAC;YAC5D,YAAY,EAAE,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAA,cAAO,GAAE,CAAC;SACvD,CAAC,CAAC,IAAI,CACL,IAAA,eAAQ,EAAC,CAAC,EAAE,eAAe,EAAE,YAAY,EAAE,EAAE,EAAE;YAC7C,IAAI,CAAC,eAAe,CAAC,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;gBACpD,OAAO,YAAK,CAAC;YACf,CAAC;YAED,MAAM,QAAQ,GAAmB,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG;gBACX;;;;;sBAKY,IAAA,yBAAc,EAAC,gBAAgB,EAAE,YAAY,EAAE,QAAQ,CAAC;;WAEnE;gBACD;;;;sBAIY,IAAA,yBAAc,EAAC,gBAAgB,EAAE,eAAe,EAAE,QAAQ,CAAC;;WAEtE;aACF,CAAC;YACF,MAAM,KAAK,GAAG,CAAC,gDAAgD,EAAE,wCAAwC,CAAC,CAAC,IAAI,CAC7G,SAAS,CACV,CAAC;YACF,OAAO,IAAA,WAAI,EACT,YAAY,CAAC;gBACX,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;gBAChC,MAAM,EAAE;oBACN,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,mBAAmB,mBAAI,CAAC,WAAW,EAAE,EAAE;iBACjG;aACF,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,kBAAkB,CAAC,eAAgC;QACzD,OAAO,IAAA,eAAQ,EAAC;YACd,0BAA0B,EAAE,eAAe,CAAC,uBAAuB,CAAC,IAAI,CAAC,IAAA,cAAO,GAAE,CAAC;YACnF,mBAAmB,EAAE,eAAe,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAA,cAAO,GAAE,CAAC;YACrE,oBAAoB,EAAE,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAA,cAAO,GAAE,CAAC;YACvE,UAAU,EAAE,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,IAAA,cAAO,GAAE,CAAC;YACnD,YAAY,EAAE,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,IAAA,cAAO,GAAE,CAAC;SACxD,CAAC,CAAC,IAAI,CACL,IAAA,eAAQ,EACN,CAAC,EAAE,0BAA0B,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE;YACtG,MAAM,uBAAuB,GAAG,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC;YAC9D,MAAM,uBAAuB,GAC3B,uBAAuB;gBACvB,CAAC,CAAC,CAAC,0BAA0B,CAAC,MAAM,IAAI,mBAAmB,CAAC,MAAM,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;YAC3F,MAAM,sBAAsB,GAAG,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC;YACrD,IAAI,CAAC,uBAAuB,IAAI,CAAC,sBAAsB,EAAE,CAAC;gBACxD,OAAO,YAAK,CAAC;YACf,CAAC;YACD,IAAI,CAAC,uBAAuB,EAAE,CAAC;gBAC7B,OAAO,IAAA,WAAI,EAAC,YAAY,CAAC,CAAC;YAC5B,CAAC;YAED,MAAM,QAAQ,GAAmB,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG;gBACX,GAAG,CAAC,uBAAuB;oBACzB,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,oBAAoB,EAAE,QAAQ,CAAC;oBAC5E,CAAC,CAAC,EAAE,CAAC;gBACP;;;;wBAIY,IAAA,yBAAc,EAAC,oBAAoB,EAAE,0BAA0B,EAAE,QAAQ,CAAC;;aAErF;gBACD;;;;;;;;aAQC;gBACD;;;;wBAIY,IAAA,yBAAc,EAAC,cAAc,EAAE,UAAU,EAAE,QAAQ,CAAC;;;;;;aAM/D;gBACD;;;;wBAIY,IAAA,yBAAc,EAAC,cAAc,EAAE,mBAAmB,EAAE,QAAQ,CAAC;;;;;;aAMxE;aACF,CAAC;YACF,MAAM,KAAK,GAAG;gBACZ,GAAG,CAAC,uBAAuB;oBACzB,CAAC,CAAC;wBACE,gHAAgH;qBACjH;oBACH,CAAC,CAAC,EAAE,CAAC;gBACP,0GAA0G;gBAC1G,+FAA+F;gBAC/F,iGAAiG;aAClG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClB,OAAO,IAAA,YAAK,EACV,IAAA,WAAI,EACF,YAAY,CAAC;gBACX,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE;gBAChC,MAAM,EAAE,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,aAAa,mBAAI,CAAC,WAAW,EAAE,EAAE,EAAE;aACvG,CAAC,CACH,EACD,IAAA,WAAI,EAAC,YAAY,CAAC,CACnB,CAAC;QACJ,CAAC,CACF,CACF,CAAC;IACJ,CAAC;IAEO,qCAAqC,CAAC,kBAA4B,EAAE,QAAwB;QAClG,OAAO;YACL;;;;kBAIY,IAAA,yBAAc,EAAC,cAAc,EAAE,kBAAkB,EAAE,QAAQ,CAAC;;;;;;OAMvE;YACD;;;;;;OAMC;YACD;;;;;;OAMC;YACD;;;;;;;;OAQC;YACD;;;;;;;;OAQC;YACD;;;;;;OAMC;SACF,CAAC;IACJ,CAAC;CACF;AAED,MAAM,cAAc,GAAG;IACrB,SAAS;IACT,OAAO;IACP,UAAU;IACV,aAAa;IACb,mBAAmB;IACnB,yBAAyB;IACzB,kBAAkB;IAClB,SAAS;IACT,WAAW;IACX,SAAS;CACD,CAAC;AAQX,SAAS,MAAM;IACb,OAAO,UAAU,GAAkB;QACjC,OAAO,GAAG,CAAC,IAAI,CACb,IAAA,WAAI,EACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACX,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrB,OAAO,GAAG,CAAC,GAAG,CAAC;gBACf,OAAO,GAAG,CAAC;YACb,CAAC;YAED,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACjB,GAAG,CAAC,GAAG,GAAG,GAAG,CAAC;YACd,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,GAAG,EAAE,IAAI,GAAG,EAAK,EAA8B,CAClD,EACD,IAAA,UAAG,EAAC,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,EACrB,IAAA,aAAM,EAAC,CAAC,CAAC,EAAU,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAC3B,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,YAAY,CAAC,KAI5B;IACC,KAAK,CAAC,CAAC,IAAA,8BAAmB,EAAC,EAAE,GAAG,KAAK,EAAE,aAAa,EAAE,CAAC,GAAkB,EAAE,EAAE,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;AACpG,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport {\n  EMPTY,\n  filter,\n  forkJoin,\n  from,\n  map,\n  merge,\n  mergeMap,\n  Observable,\n  scan,\n  shareReplay,\n  Subject,\n  toArray,\n} from \"rxjs\";\nimport { eachValueFrom } from \"rxjs-for-await\";\nimport { Guid, GuidString, Id64String } from \"@itwin/core-bentley\";\nimport {\n  ECClassHierarchyInspector,\n  ECSqlBinding,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n  normalizeFullClassName,\n} from \"@itwin/presentation-shared\";\nimport { SelectableInstanceKey, Selectables, TRANSIENT_ELEMENT_CLASSNAME } from \"./Selectable.js\";\nimport { formIdBindings, genericExecuteQuery, releaseMainThreadOnItemsCount } from \"./Utils.js\";\n\nconst HILITE_SET_EMIT_FREQUENCY = 20;\n\n/**\n * A set of model, subcategory and element ids that can be used for specifying hilite.\n *\n * **Warning:** Used in public API as a return value. Not expected to be created / extended by package\n * consumers, may be supplemented with required attributes any time.\n *\n * @see https://www.itwinjs.org/reference/core-frontend/selectionset/hiliteset/\n * @see `HiliteSetProvider`\n * @see `createHiliteSetProvider`\n * @public\n */\nexport interface HiliteSet {\n  models: Id64String[];\n  subCategories: Id64String[];\n  elements: Id64String[];\n}\n\n/**\n * Props for creating a `HiliteSetProvider` instance.\n * @public\n */\nexport interface HiliteSetProviderProps {\n  /** An object that provides access to iModel's metadata and allows running ECSQL queries on it. */\n  imodelAccess: ECClassHierarchyInspector & ECSqlQueryExecutor;\n}\n\n/**\n * Defines return value of `createHiliteSetProvider`.\n *\n * **Warning:** Used in public API as a return value. Not expected to be created / extended by package\n * consumers, may be supplemented with required attributes any time.\n *\n * @see `createHiliteSetProvider`\n * @public\n */\nexport interface HiliteSetProvider {\n  /** Get the current hilite set iterator for the specified imodel */\n  getHiliteSet(props: { selectables: Selectables }): AsyncIterableIterator<HiliteSet>;\n}\n\n/**\n * Creates a hilite set provider that returns a `HiliteSet` for given selectables.\n * @public\n */\nexport function createHiliteSetProvider(props: HiliteSetProviderProps): HiliteSetProvider {\n  return new HiliteSetProviderImpl(props);\n}\n\nclass HiliteSetProviderImpl implements HiliteSetProvider {\n  private _imodelAccess: ECClassHierarchyInspector & ECSqlQueryExecutor;\n  // Map between a class name and its type\n  private _classRelationCache: Map<string, InstanceIdType | Promise<InstanceIdType>>;\n  #componentId: GuidString;\n  #componentName: string;\n\n  constructor(props: HiliteSetProviderProps) {\n    this._imodelAccess = props.imodelAccess;\n    this._classRelationCache = new Map();\n    this.#componentId = Guid.createValue();\n    this.#componentName = \"HiliteSetProvider\";\n  }\n\n  /**\n   * Get hilite set iterator for supplied `Selectables`.\n   */\n  public getHiliteSet(props: { selectables: Selectables }): AsyncIterableIterator<HiliteSet> {\n    const obs = this.getHiliteSetObservable(props);\n    return eachValueFrom(obs);\n  }\n\n  /**\n   * Returns a \"hot\" observable of hilite sets.\n   */\n  private getHiliteSetObservable({ selectables }: { selectables: Selectables }): Subject<HiliteSet> {\n    const instancesByType = this.getInstancesByType(selectables);\n    const observables: { [key in keyof HiliteSet]?: Observable<string> } = {\n      models: this.getHilitedModels(instancesByType),\n      subCategories: this.getHilitedSubCategories(instancesByType),\n      elements: this.getHilitedElements(instancesByType),\n    };\n\n    let hiliteSet: HiliteSet = { models: [], subCategories: [], elements: [] };\n    let lastEmitTime = performance.now();\n    const subject = new Subject<HiliteSet>();\n    const subscriptions = ([\"models\", \"subCategories\", \"elements\"] as const).map((key) =>\n      observables[key]!.subscribe({\n        next(val) {\n          hiliteSet[key].push(val);\n          if (performance.now() - lastEmitTime < HILITE_SET_EMIT_FREQUENCY) {\n            return;\n          }\n          subject.next(hiliteSet);\n          hiliteSet = { models: [], subCategories: [], elements: [] };\n          lastEmitTime = performance.now();\n        },\n        complete() {\n          observables[key] = undefined;\n          if (observables.models || observables.subCategories || observables.elements) {\n            return;\n          }\n          // Emit last batch before completing the observable.\n          if (hiliteSet.elements.length || hiliteSet.models.length || hiliteSet.subCategories.length) {\n            subject.next(hiliteSet);\n          }\n          subject.complete();\n        },\n        error(err) {\n          subscriptions.forEach((x) => x.unsubscribe());\n          subscriptions.length = 0;\n          subject.error(err);\n        },\n      }),\n    );\n\n    return subject;\n  }\n\n  private async getType(key: SelectableInstanceKey): Promise<InstanceIdType> {\n    if (key.className === TRANSIENT_ELEMENT_CLASSNAME) {\n      return \"transient\";\n    }\n\n    const normalizedClassName = normalizeFullClassName(key.className);\n    const cachedType = this._classRelationCache.get(normalizedClassName);\n    if (cachedType) {\n      return cachedType;\n    }\n\n    const promise = this.getTypeImpl(normalizedClassName).then((res) => {\n      // Update the cache with the result of the promise.\n      this._classRelationCache.set(normalizedClassName, res);\n      return res;\n    });\n\n    // Add the promise to cache to prevent `getTypeImpl` being called multiple times.\n    this._classRelationCache.set(normalizedClassName, promise);\n    return promise;\n  }\n\n  private async getTypeImpl(fullClassName: string): Promise<InstanceIdType> {\n    return (\n      (await this.checkType(fullClassName, \"BisCore.Subject\", \"subject\")) ??\n      (await this.checkType(fullClassName, \"BisCore.Model\", \"model\")) ??\n      (await this.checkType(fullClassName, \"BisCore.Category\", \"category\")) ??\n      (await this.checkType(fullClassName, \"BisCore.SubCategory\", \"subCategory\")) ??\n      (await this.checkType(fullClassName, \"Functional.FunctionalElement\", \"functionalElement\")) ??\n      (await this.checkType(fullClassName, \"BisCore.GroupInformationElement\", \"groupInformationElement\")) ??\n      (await this.checkType(fullClassName, \"BisCore.GeometricElement\", \"geometricElement\")) ??\n      (await this.checkType(fullClassName, \"BisCore.Element\", \"element\")) ??\n      \"unknown\"\n    );\n  }\n\n  private async checkType(keyClassName: string, checkClassName: string, type: InstanceIdType) {\n    try {\n      const res = this._imodelAccess.classDerivesFrom(keyClassName, checkClassName);\n      const isOfType = typeof res === \"boolean\" ? res : await res;\n      return isOfType ? type : undefined;\n    } catch (e) {\n      // we may be checking against a non-existing schema (e.g. Functional), in which case we should\n      // return undefined instead of throwing an error\n      if (e instanceof Error && e.message.match(/Schema \"[\\w\\d_]+\" not found/)) {\n        return undefined;\n      }\n      throw e;\n    }\n  }\n\n  private getInstancesByType(selectables: Selectables): InstancesByType {\n    const keyTypeObs = merge(\n      from(selectables.custom.values()).pipe(mergeMap((selectable) => selectable.loadInstanceKeys())),\n      from(selectables.instanceKeys).pipe(\n        mergeMap(([className, idSet]) => from(idSet).pipe(map((id) => ({ className, id })))),\n      ),\n    ).pipe(\n      releaseMainThreadOnItemsCount(500),\n      // Get types for each instance key\n      mergeMap((instanceKey) =>\n        from(this.getType(instanceKey)).pipe(map((instanceIdType) => ({ instanceId: instanceKey.id, instanceIdType }))),\n      ),\n      // Cache the results\n      shareReplay(),\n    );\n\n    return Object.fromEntries(\n      INSTANCE_TYPES.map((type) => [\n        type,\n        keyTypeObs.pipe(\n          filter(({ instanceIdType }) => instanceIdType === type),\n          map(({ instanceId }) => instanceId),\n          unique(),\n        ),\n      ]),\n    ) as InstancesByType;\n  }\n\n  private getHilitedModels(instancesByType: InstancesByType): Observable<string> {\n    return forkJoin({\n      modelKeys: instancesByType.model.pipe(toArray()),\n      subjectKeys: instancesByType.subject.pipe(toArray()),\n    }).pipe(\n      mergeMap(({ modelKeys, subjectKeys }) => {\n        if (!modelKeys.length && !subjectKeys.length) {\n          return EMPTY;\n        }\n\n        const bindings: ECSqlBinding[] = [];\n        const ctes = [\n          `\n            ChildSubjects(ECInstanceId, JsonProperties) AS (\n              SELECT ECInstanceId, JsonProperties\n              FROM BisCore.Subject\n              WHERE ${formIdBindings(\"ECInstanceId\", subjectKeys, bindings)}\n              UNION ALL\n              SELECT r.ECInstanceId, r.JsonProperties\n              FROM ChildSubjects s\n              JOIN BisCore.Subject r ON r.Parent.Id = s.ECInstanceId\n            )\n          `,\n          `\n            Models(ECInstanceId) AS (\n              SELECT s.ECInstanceId AS ECInstanceId\n              FROM BisCore.Model s\n              WHERE ${formIdBindings(\"ECInstanceId\", modelKeys, bindings)}\n            )\n          `,\n        ];\n        const ecsql = [\n          `\n            SELECT r.ECInstanceId AS ECInstanceId\n            FROM ChildSubjects s\n            JOIN BisCore.PhysicalPartition r ON r.Parent.Id = s.ECInstanceId OR json_extract(s.JsonProperties,'$.Subject.Model.TargetPartition') = printf('0x%x', r.ECInstanceId)\n          `,\n          `\n            SELECT ECInstanceId FROM Models\n          `,\n        ].join(\" UNION \");\n        return from(\n          executeQuery({\n            queryExecutor: this._imodelAccess,\n            query: { ctes, ecsql, bindings },\n            config: { restartToken: `${this.#componentName}/${this.#componentId}/models/${Guid.createValue()}` },\n          }),\n        );\n      }),\n    );\n  }\n\n  private getHilitedSubCategories(instancesByType: InstancesByType): Observable<string> {\n    return forkJoin({\n      subCategoryKeys: instancesByType.subCategory.pipe(toArray()),\n      categoryKeys: instancesByType.category.pipe(toArray()),\n    }).pipe(\n      mergeMap(({ subCategoryKeys, categoryKeys }) => {\n        if (!subCategoryKeys.length && !categoryKeys.length) {\n          return EMPTY;\n        }\n\n        const bindings: ECSqlBinding[] = [];\n        const ctes = [\n          `\n            CategorySubCategories(ECInstanceId) AS (\n              SELECT r.ECInstanceId AS ECInstanceId\n              FROM BisCore.Category s\n              JOIN BisCore.SubCategory r ON r.Parent.Id = s.ECInstanceId\n              WHERE ${formIdBindings(\"s.ECInstanceId\", categoryKeys, bindings)}\n            )\n          `,\n          `\n            SubCategories(ECInstanceId) AS (\n              SELECT s.ECInstanceId AS ECInstanceId\n              FROM BisCore.SubCategory s\n              WHERE ${formIdBindings(\"s.ECInstanceId\", subCategoryKeys, bindings)}\n            )\n          `,\n        ];\n        const ecsql = [`SELECT ECInstanceId FROM CategorySubCategories`, `SELECT ECInstanceId FROM SubCategories`].join(\n          \" UNION \",\n        );\n        return from(\n          executeQuery({\n            queryExecutor: this._imodelAccess,\n            query: { ctes, ecsql, bindings },\n            config: {\n              restartToken: `${this.#componentName}/${this.#componentId}/sub-categories/${Guid.createValue()}`,\n            },\n          }),\n        );\n      }),\n    );\n  }\n\n  private getHilitedElements(instancesByType: InstancesByType): Observable<string> {\n    return forkJoin({\n      groupInformationElementIds: instancesByType.groupInformationElement.pipe(toArray()),\n      geometricElementIds: instancesByType.geometricElement.pipe(toArray()),\n      functionalElementIds: instancesByType.functionalElement.pipe(toArray()),\n      elementIds: instancesByType.element.pipe(toArray()),\n      transientIds: instancesByType.transient.pipe(toArray()),\n    }).pipe(\n      mergeMap(\n        ({ groupInformationElementIds, geometricElementIds, functionalElementIds, elementIds, transientIds }) => {\n          const hasFunctionalElementIds = !!functionalElementIds.length;\n          const hasPersistentElementIds =\n            hasFunctionalElementIds ||\n            !!(groupInformationElementIds.length || geometricElementIds.length || elementIds.length);\n          const hasTransientElementIds = !!transientIds.length;\n          if (!hasPersistentElementIds && !hasTransientElementIds) {\n            return EMPTY;\n          }\n          if (!hasPersistentElementIds) {\n            return from(transientIds);\n          }\n\n          const bindings: ECSqlBinding[] = [];\n          const ctes = [\n            ...(hasFunctionalElementIds\n              ? this.getHilitedFunctionalElementsQueryCTEs(functionalElementIds, bindings)\n              : []),\n            `\n              GroupMembers(ECInstanceId, ECClassId) AS (\n                SELECT TargetECInstanceId, TargetECClassId\n                FROM BisCore.ElementGroupsMembers\n                WHERE ${formIdBindings(\"SourceECInstanceId\", groupInformationElementIds, bindings)}\n              )\n            `,\n            `\n              GroupGeometricElements(ECInstanceId, ECClassId) AS (\n                SELECT ECInstanceId, ECClassId FROM GroupMembers\n                UNION ALL\n                SELECT r.ECInstanceId, r.ECClassId\n                FROM GroupGeometricElements s\n                JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId\n              )\n            `,\n            `\n              ElementGeometricElements(ECInstanceId, ECClassId) AS (\n                SELECT ECInstanceId, ECClassId\n                FROM BisCore.Element\n                WHERE ${formIdBindings(\"ECInstanceId\", elementIds, bindings)}\n                UNION ALL\n                SELECT r.ECInstanceId, r.ECClassId\n                FROM ElementGeometricElements s\n                JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId\n              )\n            `,\n            `\n              GeometricElementGeometricElements(ECInstanceId, ECClassId) AS (\n                SELECT ECInstanceId, ECClassId\n                FROM BisCore.GeometricElement\n                WHERE ${formIdBindings(\"ECInstanceId\", geometricElementIds, bindings)}\n                UNION ALL\n                SELECT r.ECInstanceId, r.ECClassId\n                FROM GeometricElementGeometricElements s\n                JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId\n              )\n            `,\n          ];\n          const ecsql = [\n            ...(hasFunctionalElementIds\n              ? [\n                  \"SELECT ECInstanceId FROM FunctionalElementChildGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)\",\n                ]\n              : []),\n            \"SELECT ECInstanceId FROM GeometricElementGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)\",\n            \"SELECT ECInstanceId FROM GroupGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)\",\n            \"SELECT ECInstanceId FROM ElementGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)\",\n          ].join(\" UNION \");\n          return merge(\n            from(\n              executeQuery({\n                queryExecutor: this._imodelAccess,\n                query: { ctes, ecsql, bindings },\n                config: { restartToken: `${this.#componentName}/${this.#componentId}/elements/${Guid.createValue()}` },\n              }),\n            ),\n            from(transientIds),\n          );\n        },\n      ),\n    );\n  }\n\n  private getHilitedFunctionalElementsQueryCTEs(functionalElements: string[], bindings: ECSqlBinding[]): string[] {\n    return [\n      `\n        ChildFunctionalElements(ECInstanceId, ECClassId) AS (\n          SELECT ECInstanceId, ECClassId\n          FROM Functional.FunctionalElement\n          WHERE ${formIdBindings(\"ECInstanceId\", functionalElements, bindings)}\n          UNION ALL\n          SELECT r.ECInstanceId, r.ECClassId\n          FROM ChildFunctionalElements s\n          JOIN Functional.FunctionalElement r ON r.Parent.Id = s.ECInstanceId\n        )\n      `,\n      `\n        PhysicalElements(ECInstanceId, ECClassId) AS (\n          SELECT r.SourceECInstanceId, r.SourceECClassId\n          FROM ChildFunctionalElements s\n          JOIN Functional.PhysicalElementFulfillsFunction r ON r.TargetECInstanceId = s.ECInstanceId\n        )\n      `,\n      `\n        DrawingGraphicElements(ECInstanceId, ECClassId) AS (\n          SELECT r.SourceECInstanceId, r.SourceECClassId\n          FROM ChildFunctionalElements s\n          JOIN Functional.DrawingGraphicRepresentsFunctionalElement r ON r.TargetECInstanceId = s.ECInstanceId\n        )\n      `,\n      `\n        PhysicalElementGeometricElements(ECInstanceId, ECClassId) AS (\n          SELECT ECInstanceId, ECClassId FROM PhysicalElements\n          UNION ALL\n          SELECT r.ECInstanceId, r.ECClassId\n          FROM PhysicalElementGeometricElements s\n          JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId\n        )\n      `,\n      `\n        DrawingGraphicElementGeometricElements(ECInstanceId, ECClassId) AS (\n          SELECT ECInstanceId, ECClassId FROM DrawingGraphicElements\n          UNION ALL\n          SELECT r.ECInstanceId, r.ECClassId\n          FROM DrawingGraphicElementGeometricElements s\n          JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId\n        )\n      `,\n      `\n        FunctionalElementChildGeometricElements(ECInstanceId, ECClassId) AS (\n          SELECT ECInstanceId, ECClassId FROM PhysicalElementGeometricElements\n          UNION\n          SELECT ECInstanceId, ECClassId FROM DrawingGraphicElementGeometricElements\n        )\n      `,\n    ];\n  }\n}\n\nconst INSTANCE_TYPES = [\n  \"subject\",\n  \"model\",\n  \"category\",\n  \"subCategory\",\n  \"functionalElement\",\n  \"groupInformationElement\",\n  \"geometricElement\",\n  \"element\",\n  \"transient\",\n  \"unknown\",\n] as const;\n\ntype InstanceIdType = (typeof INSTANCE_TYPES)[number];\n\ntype InstancesByType = {\n  [idType in InstanceIdType]: Observable<string>;\n};\n\nfunction unique<T>() {\n  return function (obs: Observable<T>): Observable<T> {\n    return obs.pipe(\n      scan(\n        (acc, val) => {\n          if (acc.set.has(val)) {\n            delete acc.val;\n            return acc;\n          }\n\n          acc.set.add(val);\n          acc.val = val;\n          return acc;\n        },\n        { set: new Set<T>() } as { set: Set<T>; val?: T },\n      ),\n      map(({ val }) => val),\n      filter((x): x is T => !!x),\n    );\n  };\n}\n\nasync function* executeQuery(props: {\n  queryExecutor: ECSqlQueryExecutor;\n  query: ECSqlQueryDef;\n  config?: ECSqlQueryReaderOptions;\n}): AsyncIterableIterator<string> {\n  yield* genericExecuteQuery({ ...props, parseQueryRow: (row: ECSqlQueryRow) => row.ECInstanceId });\n}\n"]}

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

@@ -1,51 +0,0 @@
-import "./DisposePolyfill.js";
-import { ECClassHierarchyInspector, ECSqlQueryExecutor } from "@itwin/presentation-shared";
-import { createHiliteSetProvider, HiliteSet, HiliteSetProvider } from "./HiliteSetProvider.js";
-import { SelectionStorage } from "./SelectionStorage.js";
-/**
- * Props for creating an `IModelHiliteSetProvider` instance.
- * @public
- */
-export interface IModelHiliteSetProviderProps {
-    /** Selection storage to use for retrieving the hilite set. */
-    selectionStorage: SelectionStorage;
-    /** A callback that should return iModel access by iModel key. */
-    imodelProvider: (imodelKey: string) => ECClassHierarchyInspector & ECSqlQueryExecutor;
-    /** An optional hilite set provider factory. If not provided, defaults to `createHiliteSetProvider` from this package. */
-    createHiliteSetProvider?: typeof createHiliteSetProvider;
-}
-/**
- * Defines return value of `createIModelHiliteSetProvider`.
- *
- * **Warning:** 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.
- *
- * @see `createIModelHiliteSetProvider`
- * @public
- */
-export interface IModelHiliteSetProvider {
-    /** Get the hilite set provider for the specified iModel. */
-    getHiliteSetProvider(props: {
-        /** iModel to get hilite set provider for */
-        imodelKey: string;
-    }): HiliteSetProvider;
-    /** Get the hilite set iterator for active selection in the specified iModel. */
-    getCurrentHiliteSet(props: {
-        /** iModel to get hilite set for */
-        imodelKey: string;
-    }): AsyncIterableIterator<HiliteSet>;
-    /**
-     * Disposes the cache.
-     */
-    [Symbol.dispose]: () => void;
-}
-/**
- * Creates a hilite set provider that can efficiently get a hilite set for the "active" selection in an iModel.
- *
- * This specific implementation caches the hilite set for the current selection, so subsequent hilite set
- * requests for the same iModel, don't cost until selection changes.
- *
- * @public
- */
-export declare function createIModelHiliteSetProvider(props: IModelHiliteSetProviderProps): IModelHiliteSetProvider;
-//# sourceMappingURL=IModelHiliteSetProvider.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"IModelHiliteSetProvider.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/IModelHiliteSetProvider.ts"],"names":[],"mappings":"AAKA,OAAO,sBAAsB,CAAC;AAI9B,OAAO,EAAE,yBAAyB,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAC3F,OAAO,EAAE,uBAAuB,EAAE,SAAS,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAE/F,OAAO,EAAkC,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzF;;;GAGG;AACH,MAAM,WAAW,4BAA4B;IAC3C,8DAA8D;IAC9D,gBAAgB,EAAE,gBAAgB,CAAC;IAEnC,iEAAiE;IACjE,cAAc,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,yBAAyB,GAAG,kBAAkB,CAAC;IAEtF,yHAAyH;IACzH,uBAAuB,CAAC,EAAE,OAAO,uBAAuB,CAAC;CAC1D;AAED;;;;;;;;GAQG;AACH,MAAM,WAAW,uBAAuB;IACtC,4DAA4D;IAC5D,oBAAoB,CAAC,KAAK,EAAE;QAC1B,4CAA4C;QAC5C,SAAS,EAAE,MAAM,CAAC;KACnB,GAAG,iBAAiB,CAAC;IAEtB,gFAAgF;IAChF,mBAAmB,CAAC,KAAK,EAAE;QACzB,mCAAmC;QACnC,SAAS,EAAE,MAAM,CAAC;KACnB,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC;IAErC;;OAEG;IACH,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;CAC9B;AAED;;;;;;;GAOG;AACH,wBAAgB,6BAA6B,CAAC,KAAK,EAAE,4BAA4B,GAAG,uBAAuB,CAE1G"}

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

@@ -1,66 +0,0 @@
-"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.createIModelHiliteSetProvider = createIModelHiliteSetProvider;
-require("./DisposePolyfill.js");
-const rxjs_1 = require("rxjs");
-const rxjs_for_await_1 = require("rxjs-for-await");
-const HiliteSetProvider_js_1 = require("./HiliteSetProvider.js");
-const SelectionStorage_js_1 = require("./SelectionStorage.js");
-/**
- * Creates a hilite set provider that can efficiently get a hilite set for the "active" selection in an iModel.
- *
- * This specific implementation caches the hilite set for the current selection, so subsequent hilite set
- * requests for the same iModel, don't cost until selection changes.
- *
- * @public
- */
-function createIModelHiliteSetProvider(props) {
-    return new IModelHiliteSetProviderImpl(props);
-}
-class IModelHiliteSetProviderImpl {
-    _selectionStorage;
-    _hiliteSetProviders = new Map();
-    _cache = new Map();
-    _removeListener;
-    _imodelProvider;
-    _createHiliteSetProvider;
-    constructor(props) {
-        this._selectionStorage = props.selectionStorage;
-        this._imodelProvider = props.imodelProvider;
-        this._removeListener = this._selectionStorage.selectionChangeEvent.addListener((args) => {
-            this._cache.delete(args.imodelKey);
-            if (args.changeType === "clear" && args.source === SelectionStorage_js_1.CLEAR_SELECTION_STORAGE_SOURCE) {
-                this._hiliteSetProviders.delete(args.imodelKey);
-            }
-        });
-        /* v8 ignore next -- @preserve */
-        this._createHiliteSetProvider = props.createHiliteSetProvider ?? HiliteSetProvider_js_1.createHiliteSetProvider;
-    }
-    getHiliteSetProvider({ imodelKey }) {
-        let provider = this._hiliteSetProviders.get(imodelKey);
-        if (!provider) {
-            provider = this._createHiliteSetProvider({ imodelAccess: this._imodelProvider(imodelKey) });
-            this._hiliteSetProviders.set(imodelKey, provider);
-        }
-        return provider;
-    }
-    getCurrentHiliteSet({ imodelKey }) {
-        let hiliteSet = this._cache.get(imodelKey);
-        if (!hiliteSet) {
-            const selectables = this._selectionStorage.getSelection({ imodelKey });
-            hiliteSet = (0, rxjs_1.from)(this.getHiliteSetProvider({ imodelKey }).getHiliteSet({ selectables })).pipe((0, rxjs_1.shareReplay)({ refCount: true }));
-            this._cache.set(imodelKey, hiliteSet);
-        }
-        return (0, rxjs_for_await_1.eachValueFrom)(hiliteSet);
-    }
-    [Symbol.dispose]() {
-        this._removeListener();
-        this._hiliteSetProviders = new Map();
-        this._cache = new Map();
-    }
-}
-//# sourceMappingURL=IModelHiliteSetProvider.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"IModelHiliteSetProvider.js","sourceRoot":"","sources":["../../../src/unified-selection/IModelHiliteSetProvider.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,gCAA8B;AAE9B,+BAAqD;AACrD,mDAA+C;AAE/C,iEAA+F;AAE/F,+DAAyF;AA6CzF;;;;;;;GAOG;AACH,uCAA8C,KAAmC;IAC/E,OAAO,IAAI,2BAA2B,CAAC,KAAK,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,2BAA2B;IACvB,iBAAiB,CAAmB;IACpC,mBAAmB,GAAG,IAAI,GAAG,EAA6B,CAAC;IAC3D,MAAM,GAAG,IAAI,GAAG,EAAiC,CAAC;IAClD,eAAe,CAAa;IAC5B,eAAe,CAAwE;IACvF,wBAAwB,CAAiC;IAEjE,YAAY,KAAmC;QAC7C,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC,cAAc,CAAC;QAC5C,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,WAAW,CAC5E,CAAC,IAAqC,EAAE,EAAE;YACxC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACnC,IAAI,IAAI,CAAC,UAAU,KAAK,OAAO,IAAI,IAAI,CAAC,MAAM,KAAK,oDAA8B,EAAE,CAAC;gBAClF,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClD,CAAC;QACH,CAAC,CACF,CAAC;QACF,iCAAiC;QACjC,IAAI,CAAC,wBAAwB,GAAG,KAAK,CAAC,uBAAuB,IAAI,8CAAuB,CAAC;IAC3F,CAAC;IAEM,oBAAoB,CAAC,EAAE,SAAS,EAAyB;QAC9D,IAAI,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACvD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,QAAQ,GAAG,IAAI,CAAC,wBAAwB,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAC5F,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEM,mBAAmB,CAAC,EAAE,SAAS,EAAyB;QAC7D,IAAI,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;YACvE,SAAS,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,oBAAoB,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,CAC3F,IAAA,kBAAW,EAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAChC,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACxC,CAAC;QACD,OAAO,IAAA,8BAAa,EAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,CAAC,mBAAmB,GAAG,IAAI,GAAG,EAAE,CAAC;QACrC,IAAI,CAAC,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;IAC1B,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport \"./DisposePolyfill.js\";\n\nimport { from, Observable, shareReplay } from \"rxjs\";\nimport { eachValueFrom } from \"rxjs-for-await\";\nimport { ECClassHierarchyInspector, ECSqlQueryExecutor } from \"@itwin/presentation-shared\";\nimport { createHiliteSetProvider, HiliteSet, HiliteSetProvider } from \"./HiliteSetProvider.js\";\nimport { StorageSelectionChangeEventArgs } from \"./SelectionChangeEvent.js\";\nimport { CLEAR_SELECTION_STORAGE_SOURCE, SelectionStorage } from \"./SelectionStorage.js\";\n\n/**\n * Props for creating an `IModelHiliteSetProvider` instance.\n * @public\n */\nexport interface IModelHiliteSetProviderProps {\n  /** Selection storage to use for retrieving the hilite set. */\n  selectionStorage: SelectionStorage;\n\n  /** A callback that should return iModel access by iModel key. */\n  imodelProvider: (imodelKey: string) => ECClassHierarchyInspector & ECSqlQueryExecutor;\n\n  /** An optional hilite set provider factory. If not provided, defaults to `createHiliteSetProvider` from this package. */\n  createHiliteSetProvider?: typeof createHiliteSetProvider;\n}\n\n/**\n * Defines return value of `createIModelHiliteSetProvider`.\n *\n * **Warning:** Used in public API as a return value. Not expected to be created / extended by package\n * consumers, may be supplemented with required attributes any time.\n *\n * @see `createIModelHiliteSetProvider`\n * @public\n */\nexport interface IModelHiliteSetProvider {\n  /** Get the hilite set provider for the specified iModel. */\n  getHiliteSetProvider(props: {\n    /** iModel to get hilite set provider for */\n    imodelKey: string;\n  }): HiliteSetProvider;\n\n  /** Get the hilite set iterator for active selection in the specified iModel. */\n  getCurrentHiliteSet(props: {\n    /** iModel to get hilite set for */\n    imodelKey: string;\n  }): AsyncIterableIterator<HiliteSet>;\n\n  /**\n   * Disposes the cache.\n   */\n  [Symbol.dispose]: () => void;\n}\n\n/**\n * Creates a hilite set provider that can efficiently get a hilite set for the \"active\" selection in an iModel.\n *\n * This specific implementation caches the hilite set for the current selection, so subsequent hilite set\n * requests for the same iModel, don't cost until selection changes.\n *\n * @public\n */\nexport function createIModelHiliteSetProvider(props: IModelHiliteSetProviderProps): IModelHiliteSetProvider {\n  return new IModelHiliteSetProviderImpl(props);\n}\n\nclass IModelHiliteSetProviderImpl implements IModelHiliteSetProvider {\n  private _selectionStorage: SelectionStorage;\n  private _hiliteSetProviders = new Map<string, HiliteSetProvider>();\n  private _cache = new Map<string, Observable<HiliteSet>>();\n  private _removeListener: () => void;\n  private _imodelProvider: (imodelKey: string) => ECClassHierarchyInspector & ECSqlQueryExecutor;\n  private _createHiliteSetProvider: typeof createHiliteSetProvider;\n\n  constructor(props: IModelHiliteSetProviderProps) {\n    this._selectionStorage = props.selectionStorage;\n    this._imodelProvider = props.imodelProvider;\n    this._removeListener = this._selectionStorage.selectionChangeEvent.addListener(\n      (args: StorageSelectionChangeEventArgs) => {\n        this._cache.delete(args.imodelKey);\n        if (args.changeType === \"clear\" && args.source === CLEAR_SELECTION_STORAGE_SOURCE) {\n          this._hiliteSetProviders.delete(args.imodelKey);\n        }\n      },\n    );\n    /* v8 ignore next -- @preserve */\n    this._createHiliteSetProvider = props.createHiliteSetProvider ?? createHiliteSetProvider;\n  }\n\n  public getHiliteSetProvider({ imodelKey }: { imodelKey: string }): HiliteSetProvider {\n    let provider = this._hiliteSetProviders.get(imodelKey);\n    if (!provider) {\n      provider = this._createHiliteSetProvider({ imodelAccess: this._imodelProvider(imodelKey) });\n      this._hiliteSetProviders.set(imodelKey, provider);\n    }\n    return provider;\n  }\n\n  public getCurrentHiliteSet({ imodelKey }: { imodelKey: string }): AsyncIterableIterator<HiliteSet> {\n    let hiliteSet = this._cache.get(imodelKey);\n    if (!hiliteSet) {\n      const selectables = this._selectionStorage.getSelection({ imodelKey });\n      hiliteSet = from(this.getHiliteSetProvider({ imodelKey }).getHiliteSet({ selectables })).pipe(\n        shareReplay({ refCount: true }),\n      );\n      this._cache.set(imodelKey, hiliteSet);\n    }\n    return eachValueFrom(hiliteSet);\n  }\n\n  public [Symbol.dispose](): void {\n    this._removeListener();\n    this._hiliteSetProviders = new Map();\n    this._cache = new Map();\n  }\n}\n"]}