@itwin/tree-widget-react 4.0.0-alpha.2 → 4.0.0-alpha.20

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

@@ -0,0 +1,207 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { BehaviorSubject, bufferCount, debounceTime, defer, EMPTY, filter, first, forkJoin, from, fromEventPattern, map, mergeMap, of, reduce, scan, share, shareReplay, startWith, Subject, switchMap, take, takeUntil, tap, } from "rxjs";
+import { Guid, Id64 } from "@itwin/core-bentley";
+import { createECSqlQueryExecutor } from "@itwin/presentation-core-interop";
+import { getClassesByView, getIdsFromChildrenTree, getOptimalBatchSize, releaseMainThreadOnItemsCount, setDifference, updateChildrenTree } from "./Utils.js";
+/** @internal */
+export const SET_CHANGE_DEBOUNCE_TIME = 20;
+/** @internal */
+export class AlwaysAndNeverDrawnElementInfo {
+    #subscriptions;
+    #alwaysDrawn;
+    #neverDrawn;
+    #disposeSubject = new Subject();
+    #viewport;
+    #elementClassName;
+    #componentId;
+    #componentName;
+    #suppressors;
+    #suppress = new Subject();
+    constructor(props) {
+        this.#elementClassName = props.elementClassName;
+        this.#viewport = props.viewport;
+        this.#alwaysDrawn = { cacheEntryObs: this.createCacheEntryObservable("always") };
+        this.#neverDrawn = { cacheEntryObs: this.createCacheEntryObservable("never") };
+        this.#suppressors = this.#suppress.pipe(scan((acc, suppress) => acc + (suppress ? 1 : -1), 0), startWith(0), shareReplay(1));
+        this.#subscriptions = [this.#alwaysDrawn.cacheEntryObs.subscribe(), this.#neverDrawn.cacheEntryObs.subscribe()];
+        this.#componentId = props.componentId ?? Guid.createValue();
+        this.#componentName = "AlwaysAndNeverDrawnElementInfo";
+    }
+    suppressChangeEvents() {
+        this.#suppress.next(true);
+    }
+    resumeChangeEvents() {
+        this.#suppress.next(false);
+    }
+    getElementsTree({ setType, modelIds, ...props }) {
+        const cache = setType === "always" ? this.#alwaysDrawn : this.#neverDrawn;
+        const getElements = (rootTreeNodes) => {
+            if (!rootTreeNodes) {
+                return new Map();
+            }
+            const pathToElements = [modelIds];
+            if ("categoryIds" in props && props.categoryIds) {
+                pathToElements.push(props.categoryIds);
+                if ("parentElementIdsPath" in props && props.parentElementIdsPath) {
+                    props.parentElementIdsPath.forEach((parentElementIds) => pathToElements.push(parentElementIds));
+                }
+            }
+            return this.getChildrenTree({ currentChildrenTree: rootTreeNodes, pathToElements, currentIdsIndex: 0 });
+        };
+        return !cache.latestCacheEntryValue
+            ? cache.cacheEntryObs.pipe(map(getElements))
+            : this.#suppressors.pipe(take(1), mergeMap((suppressionCount) => suppressionCount > 0 ? of(cache.latestCacheEntryValue).pipe(map(getElements)) : cache.cacheEntryObs.pipe(map(getElements))));
+    }
+    getChildrenTree({ currentChildrenTree, pathToElements, currentIdsIndex, }) {
+        if (currentIdsIndex >= pathToElements.length) {
+            return currentChildrenTree;
+        }
+        const result = new Map();
+        const currentParentIds = pathToElements[currentIdsIndex];
+        // currentParentIds is undefined - it means that we are getting children for categories that don't have model as parent
+        if (!currentParentIds) {
+            for (const entry of currentChildrenTree.values()) {
+                if (entry.children) {
+                    const childrenTreeOfChildren = this.getChildrenTree({
+                        currentChildrenTree: entry.children,
+                        pathToElements,
+                        currentIdsIndex: currentIdsIndex + 1,
+                    });
+                    childrenTreeOfChildren.forEach((val, childId) => result.set(childId, val));
+                }
+            }
+            return result;
+        }
+        for (const parentId of Id64.iterable(currentParentIds)) {
+            const entry = currentChildrenTree.get(parentId);
+            if (entry?.children) {
+                const childrenTreeOfChildren = this.getChildrenTree({
+                    currentChildrenTree: entry.children,
+                    pathToElements,
+                    currentIdsIndex: currentIdsIndex + 1,
+                });
+                childrenTreeOfChildren.forEach((val, childId) => result.set(childId, val));
+            }
+        }
+        return result;
+    }
+    createCacheEntryObservable(setType) {
+        const event = setType === "always" ? this.#viewport.onAlwaysDrawnChanged : this.#viewport.onNeverDrawnChanged;
+        const getIds = setType === "always" ? () => this.#viewport.alwaysDrawn : () => this.#viewport.neverDrawn;
+        const resultSubject = new BehaviorSubject(undefined);
+        const obs = fromEventPattern((handler) => event.addListener(handler), (handler) => event.removeListener(handler)).pipe(
+        // Fire the observable once at the beginning
+        startWith(undefined), 
+        // Reset result subject as soon as a new event is emitted.
+        // This will make newly subscribed observers wait for the debounce period to pass
+        // instead of consuming the cached value which at this point becomes invalid.
+        tap(() => resultSubject.next(undefined)), 
+        // Check if cache updates are not suppressed.
+        switchMap(() => this.#suppressors.pipe(filter((suppressors) => suppressors === 0), take(1))), debounceTime(SET_CHANGE_DEBOUNCE_TIME), 
+        // Cancel pending request if dispose() is called.
+        takeUntil(this.#disposeSubject), 
+        // If multiple requests are sent at once, preserve only the result of the newest.
+        switchMap(() => this.queryAlwaysOrNeverDrawnElementInfo(getIds(), `${setType}Drawn`)), tap((cacheEntry) => {
+            const value = setType === "always" ? this.#alwaysDrawn : this.#neverDrawn;
+            value.latestCacheEntryValue = cacheEntry;
+        }), 
+        // Share the result by using a subject which always emits the saved result.
+        share({
+            connector: () => resultSubject,
+            resetOnRefCountZero: false,
+        }), 
+        // Wait until the result is available.
+        first((x) => !!x, new Map()));
+        return obs;
+    }
+    [Symbol.dispose]() {
+        this.#subscriptions.forEach((x) => x.unsubscribe());
+        this.#subscriptions = [];
+        this.#disposeSubject.next();
+    }
+    queryAlwaysOrNeverDrawnElementInfo(set, requestId) {
+        const elementInfo = set?.size
+            ? from(set).pipe(bufferCount(getOptimalBatchSize({ totalSize: set.size, maximumBatchSize: 5000 })), releaseMainThreadOnItemsCount(2), mergeMap((block, index) => this.queryElementInfo(block, `${requestId}-${index}`), 10))
+            : EMPTY;
+        return elementInfo.pipe(releaseMainThreadOnItemsCount(500), reduce((acc, { categoryId, rootCategoryId, modelId, elementsPath }) => {
+            const elementIdInList = elementsPath[elementsPath.length - 1];
+            const additionalPropsGetter = (id, additionalProps) => {
+                if (id === elementIdInList) {
+                    // Last element in elementsPath is in always/never drawn set. We want to mark, that it is in the set, and save it's categoryId.
+                    return { isInAlwaysOrNeverDrawnSet: true, categoryId };
+                }
+                // Existing entries can keep their value, if it's a new entry it's not in the list.
+                return additionalProps ?? { isInAlwaysOrNeverDrawnSet: false };
+            };
+            updateChildrenTree({ tree: acc, idsToAdd: [modelId, rootCategoryId, ...elementsPath], additionalPropsGetter });
+            return acc;
+        }, (() => new Map())()));
+    }
+    queryElementInfo(elementIds, requestId) {
+        return defer(() => {
+            const executor = createECSqlQueryExecutor(this.#viewport.iModel);
+            const { elementClass } = this.#elementClassName
+                ? { elementClass: this.#elementClassName }
+                : getClassesByView(this.#viewport.viewType === "2d" ? "2d" : "3d");
+            return executor.createQueryReader({
+                ctes: [
+                    `
+            ElementInfo(modelId, rootCategoryId, categoryId, parentId, elementsPath) AS (
+              SELECT
+                Model.Id modelId,
+                Category.Id rootCategoryId,
+                Category.Id categoryId,
+                Parent.Id parentId,
+                CAST(IdToHex(ECInstanceId) AS TEXT) elementsPath
+              FROM ${elementClass}
+              WHERE InVirtualSet(?, ECInstanceId)
+
+              UNION ALL
+
+              SELECT
+                e.modelId modelId,
+                p.Category.Id rootCategoryId,
+                e.categoryId categoryId,
+                p.Parent.Id parentId,
+                CAST(IdToHex(p.ECInstanceId) AS TEXT) || ';' || e.elementsPath
+              FROM ${elementClass} p
+              JOIN ElementInfo e ON p.ECInstanceId = e.parentId
+            )
+            `,
+                ],
+                ecsql: `
+            SELECT elementsPath elementsPath, modelId modelId, categoryId categoryId, rootCategoryId rootCategoryId
+            FROM ElementInfo
+            WHERE parentId IS NULL
+          `,
+                bindings: [{ type: "idset", value: elementIds }],
+            }, {
+                rowFormat: "ECSqlPropertyNames",
+                restartToken: `${this.#componentName}/${this.#componentId}/${requestId}`,
+            });
+        }).pipe(map((row) => {
+            return { elementsPath: row.elementsPath.split(";"), modelId: row.modelId, categoryId: row.categoryId, rootCategoryId: row.rootCategoryId };
+        }));
+    }
+    getAlwaysOrNeverDrawnElements(props) {
+        return this.getElementsTree(props).pipe(map((childrenTree) => getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ treeEntry }) => treeEntry.isInAlwaysOrNeverDrawnSet })));
+    }
+    clearAlwaysAndNeverDrawnElements(props) {
+        return forkJoin({
+            alwaysDrawn: this.getAlwaysOrNeverDrawnElements({ modelIds: props.modelId, categoryIds: props.categoryIds, setType: "always" }),
+            neverDrawn: this.getAlwaysOrNeverDrawnElements({ modelIds: props.modelId, categoryIds: props.categoryIds, setType: "never" }),
+        }).pipe(map(({ alwaysDrawn, neverDrawn }) => {
+            const viewport = this.#viewport;
+            if (viewport.alwaysDrawn?.size && alwaysDrawn.size) {
+                viewport.setAlwaysDrawn({ elementIds: setDifference(viewport.alwaysDrawn, alwaysDrawn) });
+            }
+            if (viewport.neverDrawn?.size && neverDrawn.size) {
+                viewport.setNeverDrawn({ elementIds: setDifference(viewport.neverDrawn, neverDrawn) });
+            }
+        }));
+    }
+}
+//# sourceMappingURL=AlwaysAndNeverDrawnElementInfo.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AlwaysAndNeverDrawnElementInfo.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/AlwaysAndNeverDrawnElementInfo.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EACL,eAAe,EACf,WAAW,EACX,YAAY,EACZ,KAAK,EACL,KAAK,EACL,MAAM,EACN,KAAK,EACL,QAAQ,EACR,IAAI,EACJ,gBAAgB,EAChB,GAAG,EACH,QAAQ,EACR,EAAE,EACF,MAAM,EACN,IAAI,EACJ,KAAK,EACL,WAAW,EACX,SAAS,EACT,OAAO,EACP,SAAS,EACT,IAAI,EACJ,SAAS,EACT,GAAG,GACJ,MAAM,MAAM,CAAC;AACd,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,6BAA6B,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAO7J,gBAAgB;AAChB,MAAM,CAAC,MAAM,wBAAwB,GAAG,EAAE,CAAC;AA8C3C,gBAAgB;AAChB,MAAM,OAAO,8BAA8B;IACzC,cAAc,CAAiB;IAC/B,YAAY,CAAwF;IACpG,WAAW,CAAwF;IACnG,eAAe,GAAG,IAAI,OAAO,EAAQ,CAAC;IAC7B,SAAS,CAAqB;IAC9B,iBAAiB,CAAU;IACpC,YAAY,CAAa;IACzB,cAAc,CAAS;IAEvB,YAAY,CAAqB;IACjC,SAAS,GAAG,IAAI,OAAO,EAAW,CAAC;IAEnC,YAAY,KAA4F;QACtG,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,YAAY,GAAG,EAAE,aAAa,EAAE,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjF,IAAI,CAAC,WAAW,GAAG,EAAE,aAAa,EAAE,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC,EAAE,CAAC;QAC/E,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CACrC,IAAI,CAAC,CAAC,GAAG,EAAE,QAAQ,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EACrD,SAAS,CAAC,CAAC,CAAC,EACZ,WAAW,CAAC,CAAC,CAAC,CACf,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,SAAS,EAAE,EAAE,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC,CAAC;QAChH,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QAC5D,IAAI,CAAC,cAAc,GAAG,gCAAgC,CAAC;IACzD,CAAC;IAEM,oBAAoB;QACzB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEM,kBAAkB;QACvB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IAEM,eAAe,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,KAAK,EAAwB;QAC1E,MAAM,KAAK,GAAG,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;QAC1E,MAAM,WAAW,GAAG,CAAC,aAAyC,EAAkB,EAAE;YAChF,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,OAAO,IAAI,GAAG,EAAE,CAAC;YACnB,CAAC;YACD,MAAM,cAAc,GAAG,CAAC,QAAQ,CAAC,CAAC;YAClC,IAAI,aAAa,IAAI,KAAK,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;gBAChD,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;gBACvC,IAAI,sBAAsB,IAAI,KAAK,IAAI,KAAK,CAAC,oBAAoB,EAAE,CAAC;oBAClE,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,gBAAgB,EAAE,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;gBAClG,CAAC;YACH,CAAC;YACD,OAAO,IAAI,CAAC,eAAe,CAAC,EAAE,mBAAmB,EAAE,aAAa,EAAE,cAAc,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC,CAAC;QAC1G,CAAC,CAAC;QAEF,OAAO,CAAC,KAAK,CAAC,qBAAqB;YACjC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAC5C,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CACpB,IAAI,CAAC,CAAC,CAAC,EACP,QAAQ,CAAC,CAAC,gBAAgB,EAAE,EAAE,CAC5B,gBAAgB,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAC3H,CACF,CAAC;IACR,CAAC;IAEO,eAAe,CAAmB,EACxC,mBAAmB,EACnB,cAAc,EACd,eAAe,GAKhB;QACC,IAAI,eAAe,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC;YAC7C,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QACD,MAAM,MAAM,GAAoB,IAAI,GAAG,EAAE,CAAC;QAC1C,MAAM,gBAAgB,GAAG,cAAc,CAAC,eAAe,CAAC,CAAC;QACzD,uHAAuH;QACvH,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtB,KAAK,MAAM,KAAK,IAAI,mBAAmB,CAAC,MAAM,EAAE,EAAE,CAAC;gBACjD,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;oBACnB,MAAM,sBAAsB,GAAG,IAAI,CAAC,eAAe,CAAC;wBAClD,mBAAmB,EAAE,KAAK,CAAC,QAAQ;wBACnC,cAAc;wBACd,eAAe,EAAE,eAAe,GAAG,CAAC;qBACrC,CAAC,CAAC;oBACH,sBAAsB,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;gBAC7E,CAAC;YACH,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACvD,MAAM,KAAK,GAAG,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAChD,IAAI,KAAK,EAAE,QAAQ,EAAE,CAAC;gBACpB,MAAM,sBAAsB,GAAG,IAAI,CAAC,eAAe,CAAC;oBAClD,mBAAmB,EAAE,KAAK,CAAC,QAAQ;oBACnC,cAAc;oBACd,eAAe,EAAE,eAAe,GAAG,CAAC;iBACrC,CAAC,CAAC;gBACH,sBAAsB,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,0BAA0B,CAAC,OAAgB;QACjD,MAAM,KAAK,GAAG,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC;QAC9G,MAAM,MAAM,GAAG,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;QAEzG,MAAM,aAAa,GAAG,IAAI,eAAe,CAA6B,SAAS,CAAC,CAAC;QAEjF,MAAM,GAAG,GAAG,gBAAgB,CAC1B,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,EACvC,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAC3C,CAAC,IAAI;QACJ,4CAA4C;QAC5C,SAAS,CAAC,SAAS,CAAC;QACpB,0DAA0D;QAC1D,iFAAiF;QACjF,6EAA6E;QAC7E,GAAG,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,6CAA6C;QAC7C,SAAS,CAAC,GAAG,EAAE,CACb,IAAI,CAAC,YAAY,CAAC,IAAI,CACpB,MAAM,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,KAAK,CAAC,CAAC,EAC1C,IAAI,CAAC,CAAC,CAAC,CACR,CACF,EACD,YAAY,CAAC,wBAAwB,CAAC;QACtC,iDAAiD;QACjD,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC;QAC/B,iFAAiF;QACjF,SAAS,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kCAAkC,CAAC,MAAM,EAAE,EAAE,GAAG,OAAO,OAAO,CAAC,CAAC,EACrF,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;YACjB,MAAM,KAAK,GAAG,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;YAC1E,KAAK,CAAC,qBAAqB,GAAG,UAAU,CAAC;QAC3C,CAAC,CAAC;QACF,2EAA2E;QAC3E,KAAK,CAAC;YACJ,SAAS,EAAE,GAAG,EAAE,CAAC,aAAa;YAC9B,mBAAmB,EAAE,KAAK;SAC3B,CAAC;QACF,sCAAsC;QACtC,KAAK,CAAC,CAAC,CAAC,EAAuB,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,GAAG,EAAE,CAAC,CAClD,CAAC;QACF,OAAO,GAAG,CAAC;IACb,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;IAC9B,CAAC;IAEO,kCAAkC,CAAC,GAAwC,EAAE,SAAiB;QACpG,MAAM,WAAW,GAAG,GAAG,EAAE,IAAI;YAC3B,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CACZ,WAAW,CAAC,mBAAmB,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,EACjF,6BAA6B,CAAC,CAAC,CAAC,EAChC,QAAQ,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,GAAG,SAAS,IAAI,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CACtF;YACH,CAAC,CAAC,KAAK,CAAC;QACV,OAAO,WAAW,CAAC,IAAI,CACrB,6BAA6B,CAAC,GAAG,CAAC,EAClC,MAAM,CACJ,CAAC,GAAG,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE,EAAE;YAC7D,MAAM,eAAe,GAAG,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC9D,MAAM,qBAAqB,GAAG,CAAC,EAAc,EAAE,eAA0B,EAAY,EAAE;gBACrF,IAAI,EAAE,KAAK,eAAe,EAAE,CAAC;oBAC3B,+HAA+H;oBAC/H,OAAO,EAAE,yBAAyB,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;gBACzD,CAAC;gBACD,mFAAmF;gBACnF,OAAO,eAAe,IAAI,EAAE,yBAAyB,EAAE,KAAK,EAAE,CAAC;YACjE,CAAC,CAAC;YACF,kBAAkB,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,OAAO,EAAE,cAAc,EAAE,GAAG,YAAY,CAAC,EAAE,qBAAqB,EAAE,CAAC,CAAC;YAC/G,OAAO,GAAG,CAAC;QACb,CAAC,EACD,CAAC,GAAmB,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,CACpC,CACF,CAAC;IACJ,CAAC;IAEO,gBAAgB,CACtB,UAAqB,EACrB,SAAiB;QAOjB,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,QAAQ,GAAG,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACjE,MAAM,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,iBAAiB;gBAC7C,CAAC,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,iBAAiB,EAAE;gBAC1C,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACrE,OAAO,QAAQ,CAAC,iBAAiB,CAC/B;gBACE,IAAI,EAAE;oBACJ;;;;;;;;qBAQS,YAAY;;;;;;;;;;;qBAWZ,YAAY;;;aAGpB;iBACF;gBACD,KAAK,EAAE;;;;WAIN;gBACD,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;aACjD,EACD;gBACE,SAAS,EAAE,oBAAoB;gBAC/B,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,IAAI,SAAS,EAAE;aACzE,CACF,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,YAAY,EAAE,GAAG,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,cAAc,EAAE,GAAG,CAAC,cAAc,EAAE,CAAC;QAC7I,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,6BAA6B,CAAC,KAA2B;QAC9D,OAAO,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,IAAI,CACrC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,sBAAsB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,yBAAyB,EAAE,CAAC,CAAC,CACzI,CAAC;IACJ,CAAC;IAEM,gCAAgC,CAAC,KAAgE;QACtG,OAAO,QAAQ,CAAC;YACd,WAAW,EAAE,IAAI,CAAC,6BAA6B,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;YAC/H,UAAU,EAAE,IAAI,CAAC,6BAA6B,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;SAC9H,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE;YAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;YAChC,IAAI,QAAQ,CAAC,WAAW,EAAE,IAAI,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC;gBACnD,QAAQ,CAAC,cAAc,CAAC,EAAE,UAAU,EAAE,aAAa,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;YAC5F,CAAC;YACD,IAAI,QAAQ,CAAC,UAAU,EAAE,IAAI,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;gBACjD,QAAQ,CAAC,aAAa,CAAC,EAAE,UAAU,EAAE,aAAa,CAAC,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC;YACzF,CAAC;QACH,CAAC,CAAC,CACH,CAAC;IACJ,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 {\n  BehaviorSubject,\n  bufferCount,\n  debounceTime,\n  defer,\n  EMPTY,\n  filter,\n  first,\n  forkJoin,\n  from,\n  fromEventPattern,\n  map,\n  mergeMap,\n  of,\n  reduce,\n  scan,\n  share,\n  shareReplay,\n  startWith,\n  Subject,\n  switchMap,\n  take,\n  takeUntil,\n  tap,\n} from \"rxjs\";\nimport { Guid, Id64 } from \"@itwin/core-bentley\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { getClassesByView, getIdsFromChildrenTree, getOptimalBatchSize, releaseMainThreadOnItemsCount, setDifference, updateChildrenTree } from \"./Utils.js\";\n\nimport type { Observable, Subscription } from \"rxjs\";\nimport type { GuidString, Id64Arg, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { TreeWidgetViewport } from \"../TreeWidgetViewport.js\";\nimport type { ChildrenTree } from \"./Utils.js\";\n\n/** @internal */\nexport const SET_CHANGE_DEBOUNCE_TIME = 20;\n\ntype SetType = \"always\" | \"never\";\n\ninterface GetElementsTreeByModelProps {\n  /** Only always/never drawn elements that have the specified models will be returned. */\n  modelIds: Id64Arg;\n  /**\n   * The type of set from which tree should be retrieved.\n   * `always` - ChildrenTree will be created from `alwaysDrawn` set.\n   * `never` - ChildrenTree will be created from `neverDrawn` set.\n   */\n  setType: SetType;\n}\ninterface GetElementsTreeByCategoryProps {\n  modelIds: Id64Arg | undefined;\n  /**\n   * Categories of root elements.\n   *\n   * Elements are filtered by given categories. Children of those elements are also included, no matter their category.\n   */\n  categoryIds: Id64Arg;\n  /**\n   * The type of set from which tree should be retrieved.\n   * `always` - ChildrenTree will be created from `alwaysDrawn` set.\n   * `never` - ChildrenTree will be created from `neverDrawn` set.\n   */\n  setType: SetType;\n}\ninterface GetElementsTreeByElementProps extends GetElementsTreeByCategoryProps {\n  /** Path to element for which to get its' child always/never drawn elements. When undefined, models and categories will be used to get the always/never drawn elements. */\n  parentElementIdsPath: Array<Id64Arg>;\n}\n\n/** @internal */\nexport type GetElementsTreeProps = GetElementsTreeByModelProps | GetElementsTreeByCategoryProps | GetElementsTreeByElementProps;\n\n/**\n * - `categoryId` is assigned only to the elements in always/never drawn set.\n * - `isInAlwaysOrNeverDrawnSet` flag determines if key (ECInstanceId) is in always/never set.\n * @internal\n */\nexport type MapEntry = { isInAlwaysOrNeverDrawnSet: true; categoryId: Id64String } | { isInAlwaysOrNeverDrawnSet: false };\n\ntype CachedNodesMap = ChildrenTree<MapEntry>;\n\n/** @internal */\nexport class AlwaysAndNeverDrawnElementInfo implements Disposable {\n  #subscriptions: Subscription[];\n  #alwaysDrawn: { cacheEntryObs: Observable<CachedNodesMap>; latestCacheEntryValue?: CachedNodesMap };\n  #neverDrawn: { cacheEntryObs: Observable<CachedNodesMap>; latestCacheEntryValue?: CachedNodesMap };\n  #disposeSubject = new Subject<void>();\n  readonly #viewport: TreeWidgetViewport;\n  readonly #elementClassName?: string;\n  #componentId: GuidString;\n  #componentName: string;\n\n  #suppressors: Observable<number>;\n  #suppress = new Subject<boolean>();\n\n  constructor(props: { viewport: TreeWidgetViewport; elementClassName?: string; componentId?: GuidString }) {\n    this.#elementClassName = props.elementClassName;\n    this.#viewport = props.viewport;\n    this.#alwaysDrawn = { cacheEntryObs: this.createCacheEntryObservable(\"always\") };\n    this.#neverDrawn = { cacheEntryObs: this.createCacheEntryObservable(\"never\") };\n    this.#suppressors = this.#suppress.pipe(\n      scan((acc, suppress) => acc + (suppress ? 1 : -1), 0),\n      startWith(0),\n      shareReplay(1),\n    );\n    this.#subscriptions = [this.#alwaysDrawn.cacheEntryObs.subscribe(), this.#neverDrawn.cacheEntryObs.subscribe()];\n    this.#componentId = props.componentId ?? Guid.createValue();\n    this.#componentName = \"AlwaysAndNeverDrawnElementInfo\";\n  }\n\n  public suppressChangeEvents() {\n    this.#suppress.next(true);\n  }\n\n  public resumeChangeEvents() {\n    this.#suppress.next(false);\n  }\n\n  public getElementsTree({ setType, modelIds, ...props }: GetElementsTreeProps): Observable<CachedNodesMap> {\n    const cache = setType === \"always\" ? this.#alwaysDrawn : this.#neverDrawn;\n    const getElements = (rootTreeNodes: CachedNodesMap | undefined): CachedNodesMap => {\n      if (!rootTreeNodes) {\n        return new Map();\n      }\n      const pathToElements = [modelIds];\n      if (\"categoryIds\" in props && props.categoryIds) {\n        pathToElements.push(props.categoryIds);\n        if (\"parentElementIdsPath\" in props && props.parentElementIdsPath) {\n          props.parentElementIdsPath.forEach((parentElementIds) => pathToElements.push(parentElementIds));\n        }\n      }\n      return this.getChildrenTree({ currentChildrenTree: rootTreeNodes, pathToElements, currentIdsIndex: 0 });\n    };\n\n    return !cache.latestCacheEntryValue\n      ? cache.cacheEntryObs.pipe(map(getElements))\n      : this.#suppressors.pipe(\n          take(1),\n          mergeMap((suppressionCount) =>\n            suppressionCount > 0 ? of(cache.latestCacheEntryValue).pipe(map(getElements)) : cache.cacheEntryObs.pipe(map(getElements)),\n          ),\n        );\n  }\n\n  private getChildrenTree<T extends object>({\n    currentChildrenTree,\n    pathToElements,\n    currentIdsIndex,\n  }: {\n    currentChildrenTree: ChildrenTree<T>;\n    pathToElements: Array<Id64Arg | undefined>;\n    currentIdsIndex: number;\n  }): ChildrenTree<T> {\n    if (currentIdsIndex >= pathToElements.length) {\n      return currentChildrenTree;\n    }\n    const result: ChildrenTree<T> = new Map();\n    const currentParentIds = pathToElements[currentIdsIndex];\n    // currentParentIds is undefined - it means that we are getting children for categories that don't have model as parent\n    if (!currentParentIds) {\n      for (const entry of currentChildrenTree.values()) {\n        if (entry.children) {\n          const childrenTreeOfChildren = this.getChildrenTree({\n            currentChildrenTree: entry.children,\n            pathToElements,\n            currentIdsIndex: currentIdsIndex + 1,\n          });\n          childrenTreeOfChildren.forEach((val, childId) => result.set(childId, val));\n        }\n      }\n      return result;\n    }\n    for (const parentId of Id64.iterable(currentParentIds)) {\n      const entry = currentChildrenTree.get(parentId);\n      if (entry?.children) {\n        const childrenTreeOfChildren = this.getChildrenTree({\n          currentChildrenTree: entry.children,\n          pathToElements,\n          currentIdsIndex: currentIdsIndex + 1,\n        });\n        childrenTreeOfChildren.forEach((val, childId) => result.set(childId, val));\n      }\n    }\n    return result;\n  }\n\n  private createCacheEntryObservable(setType: SetType): Observable<CachedNodesMap> {\n    const event = setType === \"always\" ? this.#viewport.onAlwaysDrawnChanged : this.#viewport.onNeverDrawnChanged;\n    const getIds = setType === \"always\" ? () => this.#viewport.alwaysDrawn : () => this.#viewport.neverDrawn;\n\n    const resultSubject = new BehaviorSubject<CachedNodesMap | undefined>(undefined);\n\n    const obs = fromEventPattern(\n      (handler) => event.addListener(handler),\n      (handler) => event.removeListener(handler),\n    ).pipe(\n      // Fire the observable once at the beginning\n      startWith(undefined),\n      // Reset result subject as soon as a new event is emitted.\n      // This will make newly subscribed observers wait for the debounce period to pass\n      // instead of consuming the cached value which at this point becomes invalid.\n      tap(() => resultSubject.next(undefined)),\n      // Check if cache updates are not suppressed.\n      switchMap(() =>\n        this.#suppressors.pipe(\n          filter((suppressors) => suppressors === 0),\n          take(1),\n        ),\n      ),\n      debounceTime(SET_CHANGE_DEBOUNCE_TIME),\n      // Cancel pending request if dispose() is called.\n      takeUntil(this.#disposeSubject),\n      // If multiple requests are sent at once, preserve only the result of the newest.\n      switchMap(() => this.queryAlwaysOrNeverDrawnElementInfo(getIds(), `${setType}Drawn`)),\n      tap((cacheEntry) => {\n        const value = setType === \"always\" ? this.#alwaysDrawn : this.#neverDrawn;\n        value.latestCacheEntryValue = cacheEntry;\n      }),\n      // Share the result by using a subject which always emits the saved result.\n      share({\n        connector: () => resultSubject,\n        resetOnRefCountZero: false,\n      }),\n      // Wait until the result is available.\n      first((x): x is CachedNodesMap => !!x, new Map()),\n    );\n    return obs;\n  }\n\n  public [Symbol.dispose](): void {\n    this.#subscriptions.forEach((x) => x.unsubscribe());\n    this.#subscriptions = [];\n    this.#disposeSubject.next();\n  }\n\n  private queryAlwaysOrNeverDrawnElementInfo(set: ReadonlySet<Id64String> | undefined, requestId: string): Observable<CachedNodesMap> {\n    const elementInfo = set?.size\n      ? from(set).pipe(\n          bufferCount(getOptimalBatchSize({ totalSize: set.size, maximumBatchSize: 5000 })),\n          releaseMainThreadOnItemsCount(2),\n          mergeMap((block, index) => this.queryElementInfo(block, `${requestId}-${index}`), 10),\n        )\n      : EMPTY;\n    return elementInfo.pipe(\n      releaseMainThreadOnItemsCount(500),\n      reduce(\n        (acc, { categoryId, rootCategoryId, modelId, elementsPath }) => {\n          const elementIdInList = elementsPath[elementsPath.length - 1];\n          const additionalPropsGetter = (id: Id64String, additionalProps?: MapEntry): MapEntry => {\n            if (id === elementIdInList) {\n              // Last element in elementsPath is in always/never drawn set. We want to mark, that it is in the set, and save it's categoryId.\n              return { isInAlwaysOrNeverDrawnSet: true, categoryId };\n            }\n            // Existing entries can keep their value, if it's a new entry it's not in the list.\n            return additionalProps ?? { isInAlwaysOrNeverDrawnSet: false };\n          };\n          updateChildrenTree({ tree: acc, idsToAdd: [modelId, rootCategoryId, ...elementsPath], additionalPropsGetter });\n          return acc;\n        },\n        ((): CachedNodesMap => new Map())(),\n      ),\n    );\n  }\n\n  private queryElementInfo(\n    elementIds: Id64Array,\n    requestId: string,\n  ): Observable<{\n    rootCategoryId: Id64String;\n    modelId: Id64String;\n    categoryId: Id64String;\n    elementsPath: Id64Array;\n  }> {\n    return defer(() => {\n      const executor = createECSqlQueryExecutor(this.#viewport.iModel);\n      const { elementClass } = this.#elementClassName\n        ? { elementClass: this.#elementClassName }\n        : getClassesByView(this.#viewport.viewType === \"2d\" ? \"2d\" : \"3d\");\n      return executor.createQueryReader(\n        {\n          ctes: [\n            `\n            ElementInfo(modelId, rootCategoryId, categoryId, parentId, elementsPath) AS (\n              SELECT\n                Model.Id modelId,\n                Category.Id rootCategoryId,\n                Category.Id categoryId,\n                Parent.Id parentId,\n                CAST(IdToHex(ECInstanceId) AS TEXT) elementsPath\n              FROM ${elementClass}\n              WHERE InVirtualSet(?, ECInstanceId)\n\n              UNION ALL\n\n              SELECT\n                e.modelId modelId,\n                p.Category.Id rootCategoryId,\n                e.categoryId categoryId,\n                p.Parent.Id parentId,\n                CAST(IdToHex(p.ECInstanceId) AS TEXT) || ';' || e.elementsPath\n              FROM ${elementClass} p\n              JOIN ElementInfo e ON p.ECInstanceId = e.parentId\n            )\n            `,\n          ],\n          ecsql: `\n            SELECT elementsPath elementsPath, modelId modelId, categoryId categoryId, rootCategoryId rootCategoryId\n            FROM ElementInfo\n            WHERE parentId IS NULL\n          `,\n          bindings: [{ type: \"idset\", value: elementIds }],\n        },\n        {\n          rowFormat: \"ECSqlPropertyNames\",\n          restartToken: `${this.#componentName}/${this.#componentId}/${requestId}`,\n        },\n      );\n    }).pipe(\n      map((row) => {\n        return { elementsPath: row.elementsPath.split(\";\"), modelId: row.modelId, categoryId: row.categoryId, rootCategoryId: row.rootCategoryId };\n      }),\n    );\n  }\n\n  public getAlwaysOrNeverDrawnElements(props: GetElementsTreeProps) {\n    return this.getElementsTree(props).pipe(\n      map((childrenTree) => getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ treeEntry }) => treeEntry.isInAlwaysOrNeverDrawnSet })),\n    );\n  }\n\n  public clearAlwaysAndNeverDrawnElements(props: { categoryIds: Id64Arg; modelId: Id64String | undefined }) {\n    return forkJoin({\n      alwaysDrawn: this.getAlwaysOrNeverDrawnElements({ modelIds: props.modelId, categoryIds: props.categoryIds, setType: \"always\" }),\n      neverDrawn: this.getAlwaysOrNeverDrawnElements({ modelIds: props.modelId, categoryIds: props.categoryIds, setType: \"never\" }),\n    }).pipe(\n      map(({ alwaysDrawn, neverDrawn }) => {\n        const viewport = this.#viewport;\n        if (viewport.alwaysDrawn?.size && alwaysDrawn.size) {\n          viewport.setAlwaysDrawn({ elementIds: setDifference(viewport.alwaysDrawn, alwaysDrawn) });\n        }\n        if (viewport.neverDrawn?.size && neverDrawn.size) {\n          viewport.setNeverDrawn({ elementIds: setDifference(viewport.neverDrawn, neverDrawn) });\n        }\n      }),\n    );\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/internal/ClassNameDefinitions.d.ts

@@ -0,0 +1,43 @@
+/** @internal */
+export declare const CLASS_NAME_Subject = "BisCore.Subject";
+/** @internal */
+export declare const CLASS_NAME_Category = "BisCore.Category";
+/** @internal */
+export declare const CLASS_NAME_SpatialCategory = "BisCore.SpatialCategory";
+/** @internal */
+export declare const CLASS_NAME_DrawingCategory = "BisCore.DrawingCategory";
+/** @internal */
+export declare const CLASS_NAME_SubCategory = "BisCore.SubCategory";
+/** @internal */
+export declare const CLASS_NAME_DefinitionContainer = "BisCore.DefinitionContainer";
+/** @internal */
+export declare const CLASS_NAME_DefinitionElement = "BisCore.DefinitionElement";
+/** @internal */
+export declare const CLASS_NAME_DefinitionModel = "BisCore.DefinitionModel";
+/** @internal */
+export declare const CLASS_NAME_Element = "BisCore.Element";
+/** @internal */
+export declare const CLASS_NAME_InformationPartitionElement = "BisCore.InformationPartitionElement";
+/** @internal */
+export declare const CLASS_NAME_GeometricElement = "BisCore.GeometricElement";
+/** @internal */
+export declare const CLASS_NAME_GeometricElement2d = "BisCore.GeometricElement2d";
+/** @internal */
+export declare const CLASS_NAME_GeometricElement3d = "BisCore.GeometricElement3d";
+/** @internal */
+export declare const CLASS_NAME_ISubModeledElement = "BisCore.ISubModeledElement";
+/** @internal */
+export declare const CLASS_NAME_Model = "BisCore.Model";
+/** @internal */
+export declare const CLASS_NAME_GeometricModel2d = "BisCore.GeometricModel2d";
+/** @internal */
+export declare const CLASS_NAME_GeometricModel3d = "BisCore.GeometricModel3d";
+/** @internal */
+export declare const CLASS_NAME_ClassificationSystem = "ClassificationSystems.ClassificationSystem";
+/** @internal */
+export declare const CLASS_NAME_ClassificationTable = "ClassificationSystems.ClassificationTable";
+/** @internal */
+export declare const CLASS_NAME_Classification = "ClassificationSystems.Classification";
+/** @internal */
+export declare const CLASS_NAME_ElementHasClassifications = "ClassificationSystems.ElementHasClassifications";
+//# sourceMappingURL=ClassNameDefinitions.d.ts.map

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

@@ -0,0 +1,47 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+/** @internal */
+export const CLASS_NAME_Subject = "BisCore.Subject";
+/** @internal */
+export const CLASS_NAME_Category = "BisCore.Category";
+/** @internal */
+export const CLASS_NAME_SpatialCategory = "BisCore.SpatialCategory";
+/** @internal */
+export const CLASS_NAME_DrawingCategory = "BisCore.DrawingCategory";
+/** @internal */
+export const CLASS_NAME_SubCategory = "BisCore.SubCategory";
+/** @internal */
+export const CLASS_NAME_DefinitionContainer = "BisCore.DefinitionContainer";
+/** @internal */
+export const CLASS_NAME_DefinitionElement = "BisCore.DefinitionElement";
+/** @internal */
+export const CLASS_NAME_DefinitionModel = "BisCore.DefinitionModel";
+/** @internal */
+export const CLASS_NAME_Element = "BisCore.Element";
+/** @internal */
+export const CLASS_NAME_InformationPartitionElement = "BisCore.InformationPartitionElement";
+/** @internal */
+export const CLASS_NAME_GeometricElement = "BisCore.GeometricElement";
+/** @internal */
+export const CLASS_NAME_GeometricElement2d = "BisCore.GeometricElement2d";
+/** @internal */
+export const CLASS_NAME_GeometricElement3d = "BisCore.GeometricElement3d";
+/** @internal */
+export const CLASS_NAME_ISubModeledElement = "BisCore.ISubModeledElement";
+/** @internal */
+export const CLASS_NAME_Model = "BisCore.Model";
+/** @internal */
+export const CLASS_NAME_GeometricModel2d = "BisCore.GeometricModel2d";
+/** @internal */
+export const CLASS_NAME_GeometricModel3d = "BisCore.GeometricModel3d";
+/** @internal */
+export const CLASS_NAME_ClassificationSystem = "ClassificationSystems.ClassificationSystem";
+/** @internal */
+export const CLASS_NAME_ClassificationTable = "ClassificationSystems.ClassificationTable";
+/** @internal */
+export const CLASS_NAME_Classification = "ClassificationSystems.Classification";
+/** @internal */
+export const CLASS_NAME_ElementHasClassifications = "ClassificationSystems.ElementHasClassifications";
+//# sourceMappingURL=ClassNameDefinitions.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ClassNameDefinitions.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/ClassNameDefinitions.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,gBAAgB;AAChB,MAAM,CAAC,MAAM,kBAAkB,GAAG,iBAAiB,CAAC;AAEpD,gBAAgB;AAChB,MAAM,CAAC,MAAM,mBAAmB,GAAG,kBAAkB,CAAC;AACtD,gBAAgB;AAChB,MAAM,CAAC,MAAM,0BAA0B,GAAG,yBAAyB,CAAC;AACpE,gBAAgB;AAChB,MAAM,CAAC,MAAM,0BAA0B,GAAG,yBAAyB,CAAC;AACpE,gBAAgB;AAChB,MAAM,CAAC,MAAM,sBAAsB,GAAG,qBAAqB,CAAC;AAE5D,gBAAgB;AAChB,MAAM,CAAC,MAAM,8BAA8B,GAAG,6BAA6B,CAAC;AAC5E,gBAAgB;AAChB,MAAM,CAAC,MAAM,4BAA4B,GAAG,2BAA2B,CAAC;AACxE,gBAAgB;AAChB,MAAM,CAAC,MAAM,0BAA0B,GAAG,yBAAyB,CAAC;AAEpE,gBAAgB;AAChB,MAAM,CAAC,MAAM,kBAAkB,GAAG,iBAAiB,CAAC;AACpD,gBAAgB;AAChB,MAAM,CAAC,MAAM,sCAAsC,GAAG,qCAAqC,CAAC;AAC5F,gBAAgB;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAAG,0BAA0B,CAAC;AACtE,gBAAgB;AAChB,MAAM,CAAC,MAAM,6BAA6B,GAAG,4BAA4B,CAAC;AAC1E,gBAAgB;AAChB,MAAM,CAAC,MAAM,6BAA6B,GAAG,4BAA4B,CAAC;AAE1E,gBAAgB;AAChB,MAAM,CAAC,MAAM,6BAA6B,GAAG,4BAA4B,CAAC;AAE1E,gBAAgB;AAChB,MAAM,CAAC,MAAM,gBAAgB,GAAG,eAAe,CAAC;AAChD,gBAAgB;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAAG,0BAA0B,CAAC;AACtE,gBAAgB;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAAG,0BAA0B,CAAC;AAEtE,gBAAgB;AAChB,MAAM,CAAC,MAAM,+BAA+B,GAAG,4CAA4C,CAAC;AAC5F,gBAAgB;AAChB,MAAM,CAAC,MAAM,8BAA8B,GAAG,2CAA2C,CAAC;AAC1F,gBAAgB;AAChB,MAAM,CAAC,MAAM,yBAAyB,GAAG,sCAAsC,CAAC;AAChF,gBAAgB;AAChB,MAAM,CAAC,MAAM,oCAAoC,GAAG,iDAAiD,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\n/** @internal */\nexport const CLASS_NAME_Subject = \"BisCore.Subject\";\n\n/** @internal */\nexport const CLASS_NAME_Category = \"BisCore.Category\";\n/** @internal */\nexport const CLASS_NAME_SpatialCategory = \"BisCore.SpatialCategory\";\n/** @internal */\nexport const CLASS_NAME_DrawingCategory = \"BisCore.DrawingCategory\";\n/** @internal */\nexport const CLASS_NAME_SubCategory = \"BisCore.SubCategory\";\n\n/** @internal */\nexport const CLASS_NAME_DefinitionContainer = \"BisCore.DefinitionContainer\";\n/** @internal */\nexport const CLASS_NAME_DefinitionElement = \"BisCore.DefinitionElement\";\n/** @internal */\nexport const CLASS_NAME_DefinitionModel = \"BisCore.DefinitionModel\";\n\n/** @internal */\nexport const CLASS_NAME_Element = \"BisCore.Element\";\n/** @internal */\nexport const CLASS_NAME_InformationPartitionElement = \"BisCore.InformationPartitionElement\";\n/** @internal */\nexport const CLASS_NAME_GeometricElement = \"BisCore.GeometricElement\";\n/** @internal */\nexport const CLASS_NAME_GeometricElement2d = \"BisCore.GeometricElement2d\";\n/** @internal */\nexport const CLASS_NAME_GeometricElement3d = \"BisCore.GeometricElement3d\";\n\n/** @internal */\nexport const CLASS_NAME_ISubModeledElement = \"BisCore.ISubModeledElement\";\n\n/** @internal */\nexport const CLASS_NAME_Model = \"BisCore.Model\";\n/** @internal */\nexport const CLASS_NAME_GeometricModel2d = \"BisCore.GeometricModel2d\";\n/** @internal */\nexport const CLASS_NAME_GeometricModel3d = \"BisCore.GeometricModel3d\";\n\n/** @internal */\nexport const CLASS_NAME_ClassificationSystem = \"ClassificationSystems.ClassificationSystem\";\n/** @internal */\nexport const CLASS_NAME_ClassificationTable = \"ClassificationSystems.ClassificationTable\";\n/** @internal */\nexport const CLASS_NAME_Classification = \"ClassificationSystems.Classification\";\n/** @internal */\nexport const CLASS_NAME_ElementHasClassifications = \"ClassificationSystems.ElementHasClassifications\";\n"]}

package/lib/esm/tree-widget-react/components/trees/common/internal/ModelCategoryElementsCountCache.d.ts

@@ -0,0 +1,12 @@
+import type { Observable } from "rxjs";
+import type { GuidString, Id64String } from "@itwin/core-bentley";
+import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+/** @internal */
+export declare class ModelCategoryElementsCountCache implements Disposable {
+    #private;
+    constructor(queryExecutor: LimitingECSqlQueryExecutor, elementsClassNames: string[], componentId: GuidString);
+    private queryCategoryElementCounts;
+    [Symbol.dispose](): void;
+    getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Observable<number>;
+}
+//# sourceMappingURL=ModelCategoryElementsCountCache.d.ts.map

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

@@ -0,0 +1,112 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { bufferCount, bufferTime, defer, filter, from, map, mergeAll, mergeMap, reduce, ReplaySubject, Subject, take, toArray } from "rxjs";
+import { assert, Guid } from "@itwin/core-bentley";
+import { releaseMainThreadOnItemsCount } from "./Utils.js";
+/** @internal */
+export class ModelCategoryElementsCountCache {
+    #cache = new Map();
+    #requestsStream = new Subject();
+    #subscription;
+    #queryExecutor;
+    #elementsClassNames;
+    #componentId;
+    #componentName;
+    constructor(queryExecutor, elementsClassNames, componentId) {
+        this.#componentId = componentId;
+        this.#queryExecutor = queryExecutor;
+        this.#elementsClassNames = elementsClassNames;
+        this.#componentName = "ModelCategoryElementsCountCache";
+        this.#subscription = this.#requestsStream
+            .pipe(bufferTime(20), filter((requests) => requests.length > 0), mergeMap((requests) => this.queryCategoryElementCounts(requests)), mergeAll())
+            .subscribe({
+            next: ({ modelId, categoryId, elementsCount }) => {
+                const subject = this.#cache.get(`${modelId}-${categoryId}`);
+                assert(!!subject);
+                subject.next(elementsCount);
+            },
+        });
+    }
+    queryCategoryElementCounts(input) {
+        return from(input).pipe(reduce((acc, { modelId, categoryId }) => {
+            const entry = acc.get(modelId);
+            if (!entry) {
+                acc.set(modelId, new Set([categoryId]));
+            }
+            else {
+                entry.add(categoryId);
+            }
+            return acc;
+        }, new Map()), mergeMap((modelCategoryMap) => modelCategoryMap.entries()), map(([modelId, categoryIds]) => `Model.Id = ${modelId} AND Category.Id IN (${[...categoryIds].join(", ")})`), 
+        // we may have thousands of where clauses here, and sending a single query with all of them could take a
+        // long time - instead, split it into smaller chunks
+        bufferCount(100), mergeMap((whereClauses) => defer(() => this.#queryExecutor.createQueryReader({
+            ctes: this.#elementsClassNames.map((elementsClassName, index) => `
+                  CategoryElements${index}(id, modelId, categoryId) AS (
+                    SELECT ECInstanceId, Model.Id, Category.Id
+                    FROM ${elementsClassName}
+                    WHERE
+                      Parent.Id IS NULL
+                      AND (
+                        ${whereClauses.join(" OR ")}
+                      )
+
+                    UNION ALL
+
+                    SELECT c.ECInstanceId, p.modelId, p.categoryId
+                    FROM ${elementsClassName} c
+                    JOIN CategoryElements${index} p ON c.Parent.Id = p.id
+                  )
+                `),
+            ecsql: `
+                SELECT modelId, categoryId, COUNT(id) elementsCount
+                FROM (
+                  ${this.#elementsClassNames
+                .map((_, index) => `
+                      SELECT * FROM CategoryElements${index}
+                    `)
+                .join(" UNION ALL ")}
+                )
+                GROUP BY modelId, categoryId
+              `,
+        }, {
+            rowFormat: "ECSqlPropertyNames",
+            limit: "unbounded",
+            restartToken: `${this.#componentName}/${this.#componentId}/category-element-counts/${Guid.createValue()}`,
+        }))), releaseMainThreadOnItemsCount(500), reduce(({ acc, createKey }, row) => {
+            acc.set(createKey({ modelId: row.modelId, categoryId: row.categoryId }), {
+                modelId: row.modelId,
+                categoryId: row.categoryId,
+                elementsCount: row.elementsCount,
+            });
+            return { acc, createKey };
+        }, {
+            acc: new Map(),
+            createKey: (keyProps) => `${keyProps.modelId}-${keyProps.categoryId}`,
+        }), mergeMap(({ acc: result, createKey }) => {
+            input.forEach(({ modelId, categoryId }) => {
+                if (!result.has(createKey({ modelId, categoryId }))) {
+                    result.set(createKey({ modelId, categoryId }), { categoryId, modelId, elementsCount: 0 });
+                }
+            });
+            return from(result.values());
+        }), toArray());
+    }
+    [Symbol.dispose]() {
+        this.#subscription.unsubscribe();
+    }
+    getCategoryElementsCount(modelId, categoryId) {
+        const cacheKey = `${modelId}-${categoryId}`;
+        let result = this.#cache.get(cacheKey);
+        if (result !== undefined) {
+            return from(result).pipe(take(1));
+        }
+        result = new ReplaySubject(1);
+        this.#cache.set(cacheKey, result);
+        this.#requestsStream.next({ modelId, categoryId });
+        return from(result).pipe(take(1));
+    }
+}
+//# sourceMappingURL=ModelCategoryElementsCountCache.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelCategoryElementsCountCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/ModelCategoryElementsCountCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAC5I,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,6BAA6B,EAAE,MAAM,YAAY,CAAC;AAS3D,gBAAgB;AAChB,MAAM,OAAO,+BAA+B;IAC1C,MAAM,GAAG,IAAI,GAAG,EAAqC,CAAC;IACtD,eAAe,GAAG,IAAI,OAAO,EAAmD,CAAC;IACjF,aAAa,CAAe;IAC5B,cAAc,CAA6B;IAC3C,mBAAmB,CAAW;IAC9B,YAAY,CAAa;IACzB,cAAc,CAAS;IAEvB,YAAmB,aAAyC,EAAE,kBAA4B,EAAE,WAAuB;QACjH,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;QAChC,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,mBAAmB,GAAG,kBAAkB,CAAC;QAC9C,IAAI,CAAC,cAAc,GAAG,iCAAiC,CAAC;QACxD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,eAAe;aACtC,IAAI,CACH,UAAU,CAAC,EAAE,CAAC,EACd,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EACzC,QAAQ,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,CAAC,EACjE,QAAQ,EAAE,CACX;aACA,SAAS,CAAC;YACT,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;gBAC5D,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9B,CAAC;SACF,CAAC,CAAC;IACP,CAAC;IAEO,0BAA0B,CAChC,KAA6D;QAE7D,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CACrB,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE;YACtC,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC/B,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC1C,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACxB,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAuB,CAAC,EAClC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,EAC1D,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,cAAc,OAAO,wBAAwB,CAAC,GAAG,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;QAC5G,wGAAwG;QACxG,oDAAoD;QACpD,WAAW,CAAC,GAAG,CAAC,EAChB,QAAQ,CAAC,CAAC,YAAY,EAAE,EAAE,CACxB,KAAK,CAAC,GAAG,EAAE,CACT,IAAI,CAAC,cAAc,CAAC,iBAAiB,CACnC;YACE,IAAI,EAAE,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAChC,CAAC,iBAAiB,EAAE,KAAK,EAAE,EAAE,CAAC;oCACV,KAAK;;2BAEd,iBAAiB;;;;0BAIlB,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC;;;;;;2BAMxB,iBAAiB;2CACD,KAAK;;iBAE/B,CACF;YACD,KAAK,EAAE;;;oBAGD,IAAI,CAAC,mBAAmB;iBACvB,GAAG,CACF,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;sDACkB,KAAK;qBACtC,CACA;iBACA,IAAI,CAAC,aAAa,CAAC;;;eAGzB;SACF,EACD;YACE,SAAS,EAAE,oBAAoB;YAC/B,KAAK,EAAE,WAAW;YAClB,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,4BAA4B,IAAI,CAAC,WAAW,EAAE,EAAE;SAC1G,CACF,CACF,CACF,EACD,6BAA6B,CAAC,GAAG,CAAC,EAClC,MAAM,CACJ,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,EAAE;YAC1B,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,EAAE;gBACvE,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,aAAa,EAAE,GAAG,CAAC,aAAa;aACjC,CAAC,CAAC;YACH,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC;QAC5B,CAAC,EACD;YACE,GAAG,EAAE,IAAI,GAAG,EAAkF;YAC9F,SAAS,EAAE,CAAC,QAAyD,EAAE,EAAE,CAAC,GAAG,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,UAAU,EAAE;SACvH,CACF,EACD,QAAQ,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE;YACtC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE;gBACxC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,EAAE,CAAC;oBACpD,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;gBAC5F,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/B,CAAC,CAAC,EACF,OAAO,EAAE,CACV,CAAC;IACJ,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;IACnC,CAAC;IAEM,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QACzE,MAAM,QAAQ,GAAqB,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;QAC9D,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACpC,CAAC;QAED,MAAM,GAAG,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAClC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACpC,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { bufferCount, bufferTime, defer, filter, from, map, mergeAll, mergeMap, reduce, ReplaySubject, Subject, take, toArray } from \"rxjs\";\nimport { assert, Guid } from \"@itwin/core-bentley\";\nimport { releaseMainThreadOnItemsCount } from \"./Utils.js\";\n\nimport type { Observable, Subscription } from \"rxjs\";\nimport type { GuidString, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { CategoryId, ModelId } from \"./Types.js\";\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\n/** @internal */\nexport class ModelCategoryElementsCountCache implements Disposable {\n  #cache = new Map<ModelCategoryKey, Subject<number>>();\n  #requestsStream = new Subject<{ modelId: Id64String; categoryId: Id64String }>();\n  #subscription: Subscription;\n  #queryExecutor: LimitingECSqlQueryExecutor;\n  #elementsClassNames: string[];\n  #componentId: GuidString;\n  #componentName: string;\n\n  public constructor(queryExecutor: LimitingECSqlQueryExecutor, elementsClassNames: string[], componentId: GuidString) {\n    this.#componentId = componentId;\n    this.#queryExecutor = queryExecutor;\n    this.#elementsClassNames = elementsClassNames;\n    this.#componentName = \"ModelCategoryElementsCountCache\";\n    this.#subscription = this.#requestsStream\n      .pipe(\n        bufferTime(20),\n        filter((requests) => requests.length > 0),\n        mergeMap((requests) => this.queryCategoryElementCounts(requests)),\n        mergeAll(),\n      )\n      .subscribe({\n        next: ({ modelId, categoryId, elementsCount }) => {\n          const subject = this.#cache.get(`${modelId}-${categoryId}`);\n          assert(!!subject);\n          subject.next(elementsCount);\n        },\n      });\n  }\n\n  private queryCategoryElementCounts(\n    input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n  ): Observable<Array<{ modelId: Id64String; categoryId: Id64String; elementsCount: number }>> {\n    return from(input).pipe(\n      reduce((acc, { modelId, categoryId }) => {\n        const entry = acc.get(modelId);\n        if (!entry) {\n          acc.set(modelId, new Set([categoryId]));\n        } else {\n          entry.add(categoryId);\n        }\n        return acc;\n      }, new Map<Id64String, Id64Set>()),\n      mergeMap((modelCategoryMap) => modelCategoryMap.entries()),\n      map(([modelId, categoryIds]) => `Model.Id = ${modelId} AND Category.Id IN (${[...categoryIds].join(\", \")})`),\n      // we may have thousands of where clauses here, and sending a single query with all of them could take a\n      // long time - instead, split it into smaller chunks\n      bufferCount(100),\n      mergeMap((whereClauses) =>\n        defer(() =>\n          this.#queryExecutor.createQueryReader(\n            {\n              ctes: this.#elementsClassNames.map(\n                (elementsClassName, index) => `\n                  CategoryElements${index}(id, modelId, categoryId) AS (\n                    SELECT ECInstanceId, Model.Id, Category.Id\n                    FROM ${elementsClassName}\n                    WHERE\n                      Parent.Id IS NULL\n                      AND (\n                        ${whereClauses.join(\" OR \")}\n                      )\n\n                    UNION ALL\n\n                    SELECT c.ECInstanceId, p.modelId, p.categoryId\n                    FROM ${elementsClassName} c\n                    JOIN CategoryElements${index} p ON c.Parent.Id = p.id\n                  )\n                `,\n              ),\n              ecsql: `\n                SELECT modelId, categoryId, COUNT(id) elementsCount\n                FROM (\n                  ${this.#elementsClassNames\n                    .map(\n                      (_, index) => `\n                      SELECT * FROM CategoryElements${index}\n                    `,\n                    )\n                    .join(\" UNION ALL \")}\n                )\n                GROUP BY modelId, categoryId\n              `,\n            },\n            {\n              rowFormat: \"ECSqlPropertyNames\",\n              limit: \"unbounded\",\n              restartToken: `${this.#componentName}/${this.#componentId}/category-element-counts/${Guid.createValue()}`,\n            },\n          ),\n        ),\n      ),\n      releaseMainThreadOnItemsCount(500),\n      reduce(\n        ({ acc, createKey }, row) => {\n          acc.set(createKey({ modelId: row.modelId, categoryId: row.categoryId }), {\n            modelId: row.modelId,\n            categoryId: row.categoryId,\n            elementsCount: row.elementsCount,\n          });\n          return { acc, createKey };\n        },\n        {\n          acc: new Map<string, { modelId: Id64String; categoryId: Id64String; elementsCount: number }>(),\n          createKey: (keyProps: { modelId: Id64String; categoryId: Id64String }) => `${keyProps.modelId}-${keyProps.categoryId}`,\n        },\n      ),\n      mergeMap(({ acc: result, createKey }) => {\n        input.forEach(({ modelId, categoryId }) => {\n          if (!result.has(createKey({ modelId, categoryId }))) {\n            result.set(createKey({ modelId, categoryId }), { categoryId, modelId, elementsCount: 0 });\n          }\n        });\n\n        return from(result.values());\n      }),\n      toArray(),\n    );\n  }\n\n  public [Symbol.dispose]() {\n    this.#subscription.unsubscribe();\n  }\n\n  public getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Observable<number> {\n    const cacheKey: ModelCategoryKey = `${modelId}-${categoryId}`;\n    let result = this.#cache.get(cacheKey);\n    if (result !== undefined) {\n      return from(result).pipe(take(1));\n    }\n\n    result = new ReplaySubject(1);\n    this.#cache.set(cacheKey, result);\n    this.#requestsStream.next({ modelId, categoryId });\n    return from(result).pipe(take(1));\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/{Rxjs.d.ts → internal/Rxjs.d.ts}

@@ -1,10 +1,17 @@
 import type { Observable, ObservableInput, OperatorFunction } from "rxjs";
 /**
  * Applies reduce function and "returns" early if the predicate returns `false` for the accumulator.
+ * @internal
  */
 export declare function reduceWhile<TValue, TAccumulator>(predicate: (x: TAccumulator) => boolean, reduceFunc: (acc: TAccumulator, val: TValue) => TAccumulator, initial: TAccumulator): OperatorFunction<TValue, TAccumulator | undefined>;
-/** Same as `firstValueFrom` except it won't throw if the observable emits no values. */
+/**
+ * Same as `firstValueFrom` except it won't throw if the observable emits no values.
+ * @internal
+ */
 export declare function toVoidPromise(obs: Observable<any>): Promise<void>;
-/** Returns observable results in an array */
+/**
+ * Returns observable results in an array
+ * @internal
+ */
 export declare function collect<T>(obs: ObservableInput<T>): Promise<T[]>;
 //# sourceMappingURL=Rxjs.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/{Rxjs.js → internal/Rxjs.js}

@@ -5,13 +5,17 @@
 import { defaultIfEmpty, from, last, scan, takeWhile } from "rxjs";
 /**
  * Applies reduce function and "returns" early if the predicate returns `false` for the accumulator.
+ * @internal
  */
 export function reduceWhile(predicate, reduceFunc, initial) {
     return (obs) => {
         return obs.pipe(scan(reduceFunc, initial), takeWhile(predicate, true), defaultIfEmpty(undefined), last());
     };
 }
-/** Same as `firstValueFrom` except it won't throw if the observable emits no values. */
+/**
+ * Same as `firstValueFrom` except it won't throw if the observable emits no values.
+ * @internal
+ */
 export async function toVoidPromise(obs) {
     return new Promise((resolve, reject) => {
         obs.subscribe({
@@ -20,7 +24,10 @@ export async function toVoidPromise(obs) {
         });
     });
 }
-/** Returns observable results in an array */
+/**
+ * Returns observable results in an array
+ * @internal
+ */
 export async function collect(obs) {
     return new Promise((resolve, reject) => {
         const arr = new Array();
@@ -32,6 +39,7 @@ export async function collect(obs) {
                 resolve(arr);
             },
             error(reason) {
+                // eslint-disable-next-line @typescript-eslint/prefer-promise-reject-errors
                 reject(reason);
             },
         });

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

@@ -0,0 +1 @@
+{"version":3,"file":"Rxjs.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/Rxjs.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,MAAM,CAAC;AAInE;;;GAGG;AACH,MAAM,UAAU,WAAW,CACzB,SAAuC,EACvC,UAA4D,EAC5D,OAAqB;IAErB,OAAO,CAAC,GAAG,EAAE,EAAE;QACb,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,EAAE,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,cAAc,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5G,CAAC,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,GAAoB;IACtD,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC3C,GAAG,CAAC,SAAS,CAAC;YACZ,QAAQ,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE;YACzB,KAAK,EAAE,MAAM;SACd,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,OAAO,CAAI,GAAuB;IACtD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,GAAG,GAAG,IAAI,KAAK,EAAK,CAAC;QAC3B,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC;YAClB,IAAI,CAAC,IAAO;gBACV,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjB,CAAC;YACD,QAAQ;gBACN,OAAO,CAAC,GAAG,CAAC,CAAC;YACf,CAAC;YACD,KAAK,CAAC,MAAM;gBACV,2EAA2E;gBAC3E,MAAM,CAAC,MAAM,CAAC,CAAC;YACjB,CAAC;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { defaultIfEmpty, from, last, scan, takeWhile } from \"rxjs\";\n\nimport type { Observable, ObservableInput, OperatorFunction } from \"rxjs\";\n\n/**\n * Applies reduce function and \"returns\" early if the predicate returns `false` for the accumulator.\n * @internal\n */\nexport function reduceWhile<TValue, TAccumulator>(\n  predicate: (x: TAccumulator) => boolean,\n  reduceFunc: (acc: TAccumulator, val: TValue) => TAccumulator,\n  initial: TAccumulator,\n): OperatorFunction<TValue, TAccumulator | undefined> {\n  return (obs) => {\n    return obs.pipe(scan(reduceFunc, initial), takeWhile(predicate, true), defaultIfEmpty(undefined), last());\n  };\n}\n\n/**\n * Same as `firstValueFrom` except it won't throw if the observable emits no values.\n * @internal\n */\nexport async function toVoidPromise(obs: Observable<any>): Promise<void> {\n  return new Promise<void>((resolve, reject) => {\n    obs.subscribe({\n      complete: () => resolve(),\n      error: reject,\n    });\n  });\n}\n\n/**\n * Returns observable results in an array\n * @internal\n */\nexport async function collect<T>(obs: ObservableInput<T>): Promise<T[]> {\n  return new Promise((resolve, reject) => {\n    const arr = new Array<T>();\n    from(obs).subscribe({\n      next(item: T) {\n        arr.push(item);\n      },\n      complete() {\n        resolve(arr);\n      },\n      error(reason) {\n        // eslint-disable-next-line @typescript-eslint/prefer-promise-reject-errors\n        reject(reason);\n      },\n    });\n  });\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/{Tooltip.d.ts → internal/Tooltip.d.ts}

@@ -1,17 +1,13 @@
-import type { VisibilityStatus } from "./UseHierarchyVisibility.js";
+import type { VisibilityStatus } from "../UseHierarchyVisibility.js";
 /** @internal */
 export type Visibility = "visible" | "hidden" | "partial";
 /** @internal */
 export type NonPartialVisibilityStatus = Omit<VisibilityStatus, "state"> & {
     state: "visible" | "hidden";
 };
-interface VisibilityStatusOptions {
-    useTooltip?: string | false;
-}
 /** @internal */
-export declare function createVisibilityStatus(status: "visible" | "hidden", options?: VisibilityStatusOptions): NonPartialVisibilityStatus;
-export declare function createVisibilityStatus(status: "visible" | "hidden" | "partial" | "disabled", options?: VisibilityStatusOptions): VisibilityStatus;
+export declare function createVisibilityStatus(status: "visible" | "hidden"): NonPartialVisibilityStatus;
+export declare function createVisibilityStatus(status: "visible" | "hidden" | "partial" | "disabled"): VisibilityStatus;
 /** @internal */
-export declare function createTooltip(status: Visibility | "disabled", tooltipStringId: string | undefined): string;
-export {};
+export declare function createTooltip(status: Visibility | "disabled" | "determining"): string;
 //# sourceMappingURL=Tooltip.d.ts.map

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

@@ -0,0 +1,18 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { TreeWidget } from "../../../../TreeWidget.js";
+export function createVisibilityStatus(status) {
+    return {
+        state: status === "disabled" ? "visible" : status,
+        isDisabled: status === "disabled",
+    };
+}
+/** @internal */
+export function createTooltip(status) {
+    const statusStringId = `visibilityTooltips.status.${status}`;
+    const statusString = TreeWidget.translate(statusStringId);
+    return statusString;
+}
+//# sourceMappingURL=Tooltip.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"Tooltip.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/Tooltip.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAavD,MAAM,UAAU,sBAAsB,CAAC,MAA+B;IACpE,OAAO;QACL,KAAK,EAAE,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM;QACjD,UAAU,EAAE,MAAM,KAAK,UAAU;KAClC,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,aAAa,CAAC,MAA+C;IAC3E,MAAM,cAAc,GAAG,6BAA6B,MAAM,EAAE,CAAC;IAC7D,MAAM,YAAY,GAAG,UAAU,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IAC1D,OAAO,YAAY,CAAC;AACtB,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 { TreeWidget } from \"../../../../TreeWidget.js\";\n\nimport type { VisibilityStatus } from \"../UseHierarchyVisibility.js\";\n\n/** @internal */\nexport type Visibility = \"visible\" | \"hidden\" | \"partial\";\n\n/** @internal */\nexport type NonPartialVisibilityStatus = Omit<VisibilityStatus, \"state\"> & { state: \"visible\" | \"hidden\" };\n\n/** @internal */\nexport function createVisibilityStatus(status: \"visible\" | \"hidden\"): NonPartialVisibilityStatus;\nexport function createVisibilityStatus(status: \"visible\" | \"hidden\" | \"partial\" | \"disabled\"): VisibilityStatus;\nexport function createVisibilityStatus(status: Visibility | \"disabled\"): VisibilityStatus {\n  return {\n    state: status === \"disabled\" ? \"visible\" : status,\n    isDisabled: status === \"disabled\",\n  };\n}\n\n/** @internal */\nexport function createTooltip(status: Visibility | \"disabled\" | \"determining\"): string {\n  const statusStringId = `visibilityTooltips.status.${status}`;\n  const statusString = TreeWidget.translate(statusStringId);\n  return statusString;\n}\n"]}