@itwin/tree-widget-react 4.0.0-alpha.25 → 4.0.0-alpha.26

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

@@ -2,15 +2,16 @@
  * 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 { BehaviorSubject, bufferCount, debounceTime, defer, EMPTY, filter, first, forkJoin, from, fromEventPattern, map, mergeMap, of, race, 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 { catchBeSQLiteInterrupts } from "./UseErrorState.js";
-import { getClassesByView, getIdsFromChildrenTree, getOptimalBatchSize, releaseMainThreadOnItemsCount, setDifference, updateChildrenTree } from "./Utils.js";
+import { catchBeSQLiteInterrupts } from "../UseErrorState.js";
+import { getClassesByView, getIdsFromChildrenTree, getOptimalBatchSize, releaseMainThreadOnItemsCount, setDifference, updateChildrenTree } from "../Utils.js";
 /** @internal */
 export const SET_CHANGE_DEBOUNCE_TIME = 20;
+const ALWAYS_NEVER_BUFFER_THRESHOLD = 5000;
 /** @internal */
-export class AlwaysAndNeverDrawnElementInfo {
+export class AlwaysAndNeverDrawnElementInfoCache {
     #subscriptions;
     #alwaysDrawn;
     #neverDrawn;
@@ -22,7 +23,6 @@ export class AlwaysAndNeverDrawnElementInfo {
     #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") };
@@ -30,6 +30,7 @@ export class AlwaysAndNeverDrawnElementInfo {
         this.#subscriptions = [this.#alwaysDrawn.cacheEntryObs.subscribe(), this.#neverDrawn.cacheEntryObs.subscribe()];
         this.#componentId = props.componentId ?? Guid.createValue();
         this.#componentName = "AlwaysAndNeverDrawnElementInfo";
+        this.#elementClassName = props.elementClassName ? props.elementClassName : getClassesByView(this.#viewport.viewType === "2d" ? "2d" : "3d").elementClass;
     }
     suppressChangeEvents() {
         this.#suppress.next(true);
@@ -37,13 +38,13 @@ export class AlwaysAndNeverDrawnElementInfo {
     resumeChangeEvents() {
         this.#suppress.next(false);
     }
-    getElementsTree({ setType, modelIds, ...props }) {
+    getElementsTree({ setType, modelId, ...props }) {
         const cache = setType === "always" ? this.#alwaysDrawn : this.#neverDrawn;
         const getElements = (rootTreeNodes) => {
             if (!rootTreeNodes) {
                 return new Map();
             }
-            const pathToElements = [modelIds];
+            const pathToElements = [modelId];
             if ("categoryIds" in props && props.categoryIds) {
                 pathToElements.push(props.categoryIds);
                 if ("parentElementIdsPath" in props && props.parentElementIdsPath) {
@@ -62,20 +63,6 @@ export class AlwaysAndNeverDrawnElementInfo {
         }
         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) {
@@ -84,7 +71,9 @@ export class AlwaysAndNeverDrawnElementInfo {
                     pathToElements,
                     currentIdsIndex: currentIdsIndex + 1,
                 });
-                childrenTreeOfChildren.forEach((val, childId) => result.set(childId, val));
+                for (const [childId, childEntry] of childrenTreeOfChildren) {
+                    result.set(childId, childEntry);
+                }
             }
         }
         return result;
@@ -93,7 +82,11 @@ export class AlwaysAndNeverDrawnElementInfo {
         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(
+        // Observable listens to viewport always/never drawn set change events.
+        const sharedObs = fromEventPattern((handler) => event.addListener(handler), (handler) => event.removeListener(handler)).pipe(
+        // Return undefined when event is raised.
+        map(() => undefined), share());
+        const obs = sharedObs.pipe(
         // Fire the observable once at the beginning
         startWith(undefined), 
         // Reset result subject as soon as a new event is emitted.
@@ -105,9 +98,14 @@ export class AlwaysAndNeverDrawnElementInfo {
         // 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;
+        switchMap(() => 
+        // Race between the event and the query.
+        // In cases where event is raised before query returns, the query result is discarded.
+        race(sharedObs, defer(() => this.queryAlwaysOrNeverDrawnElementInfo(getIds(), setType)))), tap((cacheEntry) => {
+            if (cacheEntry !== undefined) {
+                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({
@@ -115,7 +113,7 @@ export class AlwaysAndNeverDrawnElementInfo {
             resetOnRefCountZero: false,
         }), 
         // Wait until the result is available.
-        first((x) => !!x, new Map()));
+        first((x) => !!x));
         return obs;
     }
     [Symbol.dispose]() {
@@ -123,9 +121,12 @@ export class AlwaysAndNeverDrawnElementInfo {
         this.#subscriptions = [];
         this.#disposeSubject.next();
     }
-    queryAlwaysOrNeverDrawnElementInfo(set, requestId) {
+    queryAlwaysOrNeverDrawnElementInfo(set, setType) {
         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))
+            ? set.size > ALWAYS_NEVER_BUFFER_THRESHOLD
+                ? // When set is larger, buffer helps to not block main thread for long periods of time
+                    from(set).pipe(bufferCount(getOptimalBatchSize({ totalSize: set.size, maximumBatchSize: ALWAYS_NEVER_BUFFER_THRESHOLD })), releaseMainThreadOnItemsCount(2), mergeMap((block, index) => this.queryElementInfo(block, `${setType}-${index}`), 10))
+                : this.queryElementInfo([...set], `${setType}-0`)
             : EMPTY;
         return elementInfo.pipe(releaseMainThreadOnItemsCount(500), reduce((acc, { categoryId, rootCategoryId, modelId, elementsPath }) => {
             const elementIdInList = elementsPath[elementsPath.length - 1];
@@ -139,14 +140,11 @@ export class AlwaysAndNeverDrawnElementInfo {
             };
             updateChildrenTree({ tree: acc, idsToAdd: [modelId, rootCategoryId, ...elementsPath], additionalPropsGetter });
             return acc;
-        }, (() => new Map())()));
+        }, 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: [
                     `
@@ -157,7 +155,7 @@ export class AlwaysAndNeverDrawnElementInfo {
                 Category.Id categoryId,
                 Parent.Id parentId,
                 CAST(IdToHex(ECInstanceId) AS TEXT) elementsPath
-              FROM ${elementClass}
+              FROM ${this.#elementClassName}
               WHERE InVirtualSet(?, ECInstanceId)
 
               UNION ALL
@@ -168,7 +166,7 @@ export class AlwaysAndNeverDrawnElementInfo {
                 e.categoryId categoryId,
                 p.Parent.Id parentId,
                 CAST(IdToHex(p.ECInstanceId) AS TEXT) || ';' || e.elementsPath
-              FROM ${elementClass} p
+              FROM ${this.#elementClassName} p
               JOIN ElementInfo e ON p.ECInstanceId = e.parentId
             )
             `,
@@ -192,8 +190,12 @@ export class AlwaysAndNeverDrawnElementInfo {
     }
     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" }),
+            alwaysDrawn: this.#viewport.alwaysDrawn?.size
+                ? this.getAlwaysOrNeverDrawnElements({ modelId: props.modelId, categoryIds: props.categoryIds, setType: "always" })
+                : of(new Set()),
+            neverDrawn: this.#viewport.neverDrawn?.size
+                ? this.getAlwaysOrNeverDrawnElements({ modelId: props.modelId, categoryIds: props.categoryIds, setType: "never" })
+                : of(new Set()),
         }).pipe(map(({ alwaysDrawn, neverDrawn }) => {
             const viewport = this.#viewport;
             if (viewport.alwaysDrawn?.size && alwaysDrawn.size) {
@@ -205,4 +207,4 @@ export class AlwaysAndNeverDrawnElementInfo {
         }));
     }
 }
-//# sourceMappingURL=AlwaysAndNeverDrawnElementInfo.js.map
+//# sourceMappingURL=AlwaysAndNeverDrawnElementInfoCache.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"AlwaysAndNeverDrawnElementInfoCache.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/caches/AlwaysAndNeverDrawnElementInfoCache.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,IAAI,EACJ,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,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,6BAA6B,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAO9J,gBAAgB;AAChB,MAAM,CAAC,MAAM,wBAAwB,GAAG,EAAE,CAAC;AAC3C,MAAM,6BAA6B,GAAG,IAAI,CAAC;AA8C3C,gBAAgB;AAChB,MAAM,OAAO,mCAAmC;IAC9C,cAAc,CAAiB;IAC/B,YAAY,CAAwF;IACpG,WAAW,CAAwF;IACnG,eAAe,GAAG,IAAI,OAAO,EAAQ,CAAC;IAC7B,SAAS,CAAqB;IAC9B,iBAAiB,CAAS;IACnC,YAAY,CAAa;IACzB,cAAc,CAAS;IAEvB,YAAY,CAAqB;IACjC,SAAS,GAAG,IAAI,OAAO,EAAW,CAAC;IAEnC,YAAY,KAA+C;QACzD,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;QACvD,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC;IAC3J,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,OAAO,EAAE,GAAG,KAAK,EAAwB;QACzE,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,GAAmB,CAAC,OAAO,CAAC,CAAC;YACjD,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,CAAC,EACtB,mBAAmB,EACnB,cAAc,EACd,eAAe,GAKhB;QACC,IAAI,eAAe,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC;YAC7C,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QACD,MAAM,MAAM,GAA2B,IAAI,GAAG,EAAE,CAAC;QACjD,MAAM,gBAAgB,GAAG,cAAc,CAAC,eAAe,CAAC,CAAC;QACzD,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,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,sBAAsB,EAAE,CAAC;oBAC3D,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;gBAClC,CAAC;YACH,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;QACjF,uEAAuE;QACvE,MAAM,SAAS,GAAG,gBAAgB,CAChC,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,yCAAyC;QACzC,GAAG,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,EACpB,KAAK,EAAE,CACR,CAAC;QACF,MAAM,GAAG,GAAG,SAAS,CAAC,IAAI;QACxB,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;QACb,wCAAwC;QACxC,sFAAsF;QACtF,IAAI,CACF,SAAS,EACT,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kCAAkC,CAAC,MAAM,EAAE,EAAE,OAAO,CAAC,CAAC,CACxE,CACF,EACD,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;YACjB,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;gBAC7B,MAAM,KAAK,GAAG,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;gBAC1E,KAAK,CAAC,qBAAqB,GAAG,UAAU,CAAC;YAC3C,CAAC;QACH,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,CAAC,CACvC,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,OAAgB;QACnG,MAAM,WAAW,GAAG,GAAG,EAAE,IAAI;YAC3B,CAAC,CAAC,GAAG,CAAC,IAAI,GAAG,6BAA6B;gBACxC,CAAC,CAAC,qFAAqF;oBACrF,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CACZ,WAAW,CAAC,mBAAmB,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,IAAI,EAAE,gBAAgB,EAAE,6BAA6B,EAAE,CAAC,CAAC,EAC1G,6BAA6B,CAAC,CAAC,CAAC,EAChC,QAAQ,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,GAAG,OAAO,IAAI,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CACpF;gBACH,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,GAAG,CAAC,EAAE,GAAG,OAAO,IAAI,CAAC;YACnD,CAAC,CAAC,KAAK,CAAC;QACV,OAAO,WAAW,CAAC,IAAI,CACrB,6BAA6B,CAAC,GAAG,CAAC,EAClC,MAAM,CAAC,CAAC,GAAmB,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE,EAAE;YACpF,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,EAAE,IAAI,GAAG,EAAE,CAAC,CACd,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,OAAO,QAAQ,CAAC,iBAAiB,CAC/B;gBACE,IAAI,EAAE;oBACJ;;;;;;;;qBAQS,IAAI,CAAC,iBAAiB;;;;;;;;;;;qBAWtB,IAAI,CAAC,iBAAiB;;;aAG9B;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,uBAAuB,EACvB,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,KAAoD;QAC1F,OAAO,QAAQ,CAAC;YACd,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI;gBAC3C,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;gBACnH,CAAC,CAAC,EAAE,CAAC,IAAI,GAAG,EAAc,CAAC;YAC7B,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI;gBACzC,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;gBAClH,CAAC,CAAC,EAAE,CAAC,IAAI,GAAG,EAAc,CAAC;SAC9B,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  race,\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 { catchBeSQLiteInterrupts } from \"../UseErrorState.js\";\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;\nconst ALWAYS_NEVER_BUFFER_THRESHOLD = 5000;\n\ntype SetType = \"always\" | \"never\";\n\ninterface GetElementsTreeByModelProps {\n  /** Only always/never drawn elements that have the specified model will be returned. */\n  modelId: Id64String;\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 extends GetElementsTreeByModelProps {\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\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\ninterface AlwaysAndNeverDrawnElementInfoCacheProps {\n  viewport: TreeWidgetViewport;\n  elementClassName?: string;\n  componentId?: GuidString;\n}\n\n/** @internal */\nexport class AlwaysAndNeverDrawnElementInfoCache 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: AlwaysAndNeverDrawnElementInfoCacheProps) {\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    this.#elementClassName = props.elementClassName ? props.elementClassName : getClassesByView(this.#viewport.viewType === \"2d\" ? \"2d\" : \"3d\").elementClass;\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, modelId, ...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: Array<Id64Arg> = [modelId];\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({\n    currentChildrenTree,\n    pathToElements,\n    currentIdsIndex,\n  }: {\n    currentChildrenTree: ChildrenTree<MapEntry>;\n    pathToElements: Array<Id64Arg>;\n    currentIdsIndex: number;\n  }): ChildrenTree<MapEntry> {\n    if (currentIdsIndex >= pathToElements.length) {\n      return currentChildrenTree;\n    }\n    const result: ChildrenTree<MapEntry> = new Map();\n    const currentParentIds = pathToElements[currentIdsIndex];\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        for (const [childId, childEntry] of childrenTreeOfChildren) {\n          result.set(childId, childEntry);\n        }\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    // Observable listens to viewport always/never drawn set change events.\n    const sharedObs = fromEventPattern(\n      (handler) => event.addListener(handler),\n      (handler) => event.removeListener(handler),\n    ).pipe(\n      // Return undefined when event is raised.\n      map(() => undefined),\n      share(),\n    );\n    const obs = sharedObs.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(() =>\n        // Race between the event and the query.\n        // In cases where event is raised before query returns, the query result is discarded.\n        race(\n          sharedObs,\n          defer(() => this.queryAlwaysOrNeverDrawnElementInfo(getIds(), setType)),\n        ),\n      ),\n      tap((cacheEntry) => {\n        if (cacheEntry !== undefined) {\n          const value = setType === \"always\" ? this.#alwaysDrawn : this.#neverDrawn;\n          value.latestCacheEntryValue = cacheEntry;\n        }\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),\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, setType: SetType): Observable<CachedNodesMap> {\n    const elementInfo = set?.size\n      ? set.size > ALWAYS_NEVER_BUFFER_THRESHOLD\n        ? // When set is larger, buffer helps to not block main thread for long periods of time\n          from(set).pipe(\n            bufferCount(getOptimalBatchSize({ totalSize: set.size, maximumBatchSize: ALWAYS_NEVER_BUFFER_THRESHOLD })),\n            releaseMainThreadOnItemsCount(2),\n            mergeMap((block, index) => this.queryElementInfo(block, `${setType}-${index}`), 10),\n          )\n        : this.queryElementInfo([...set], `${setType}-0`)\n      : EMPTY;\n    return elementInfo.pipe(\n      releaseMainThreadOnItemsCount(500),\n      reduce((acc: CachedNodesMap, { 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      }, new Map()),\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      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 ${this.#elementClassName}\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 ${this.#elementClassName} 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      catchBeSQLiteInterrupts,\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 }) {\n    return forkJoin({\n      alwaysDrawn: this.#viewport.alwaysDrawn?.size\n        ? this.getAlwaysOrNeverDrawnElements({ modelId: props.modelId, categoryIds: props.categoryIds, setType: \"always\" })\n        : of(new Set<Id64String>()),\n      neverDrawn: this.#viewport.neverDrawn?.size\n        ? this.getAlwaysOrNeverDrawnElements({ modelId: props.modelId, categoryIds: props.categoryIds, setType: \"never\" })\n        : of(new Set<Id64String>()),\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/caches/BaseIdsCache.d.ts

@@ -0,0 +1,68 @@
+import { ElementChildrenCache } from "./ElementChildrenCache.js";
+import { ElementModelCategoriesCache } from "./ElementModelCategoriesCache.js";
+import { ModelCategoryElementsCountCache } from "./ModelCategoryElementsCountCache.js";
+import { ModeledElementsCache } from "./ModeledElementsCache.js";
+import { SubCategoriesCache } from "./SubCategoriesCache.js";
+import type { Observable } from "rxjs";
+import type { Id64Array, Id64String } from "@itwin/core-bentley";
+import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+import type { Props } from "@itwin/presentation-shared";
+/** @internal */
+export interface BaseIdsCacheProps {
+    queryExecutor: LimitingECSqlQueryExecutor;
+    elementClassName: string;
+    type: "2d" | "3d";
+}
+/** @internal */
+export declare class BaseIdsCache implements Disposable {
+    #private;
+    constructor(props: BaseIdsCacheProps);
+    [Symbol.dispose](): void;
+    getSubModels(props: {
+        modelId: Id64String;
+        categoryId?: Id64String;
+    } | {
+        categoryId: Id64String;
+        modelId: Id64String | undefined;
+    }): Observable<Id64Array>;
+    getAllModels(): ReturnType<ElementModelCategoriesCache["getAllModels"]>;
+    getCategories(props: Props<ElementModelCategoriesCache["getModelCategoryIds"]>): ReturnType<ElementModelCategoriesCache["getModelCategoryIds"]>;
+    getAllCategoriesOfElements(): ReturnType<ElementModelCategoriesCache["getAllCategoriesOfElements"]>;
+    getModels(props: Props<ElementModelCategoriesCache["getCategoryElementModels"]>): ReturnType<ElementModelCategoriesCache["getCategoryElementModels"]>;
+    getCategoriesOfModelsTopMostElements(props: Props<ElementModelCategoriesCache["getCategoriesOfModelsTopMostElements"]>): ReturnType<ElementModelCategoriesCache["getCategoriesOfModelsTopMostElements"]>;
+    getElementsCount(props: Props<ModelCategoryElementsCountCache["getCategoryElementsCount"]>): ReturnType<ModelCategoryElementsCountCache["getCategoryElementsCount"]>;
+    getChildElementsTree(props: Props<ElementChildrenCache["getChildElementsTree"]>): ReturnType<ElementChildrenCache["getChildElementsTree"]>;
+    getAllChildElementsCount(props: Props<ElementChildrenCache["getAllChildElementsCount"]>): ReturnType<ElementChildrenCache["getAllChildElementsCount"]>;
+    getSubCategories(props: Props<SubCategoriesCache["getSubCategories"]>): ReturnType<SubCategoriesCache["getSubCategories"]>;
+    getSubCategoriesInfo(): ReturnType<SubCategoriesCache["getSubCategoriesInfo"]>;
+    getSubModelsUnderElement(props: Props<ModeledElementsCache["getSubModelsUnderElement"]>): ReturnType<ModeledElementsCache["getSubModelsUnderElement"]>;
+    getModeledElementsInfo(): ReturnType<ModeledElementsCache["getModeledElementsInfo"]>;
+}
+/** @internal */
+export interface BaseIdsCacheImplProps {
+    baseIdsCache: BaseIdsCache;
+}
+/** @internal */
+export declare class BaseIdsCacheImpl implements Disposable {
+    #private;
+    constructor(props: BaseIdsCacheImplProps);
+    [Symbol.dispose](): void;
+    getChildElementsTree(props: Props<ElementChildrenCache["getChildElementsTree"]>): ReturnType<ElementChildrenCache["getChildElementsTree"]>;
+    getAllChildElementsCount(props: Props<ElementChildrenCache["getAllChildElementsCount"]>): ReturnType<ElementChildrenCache["getAllChildElementsCount"]>;
+    getSubCategories(props: Props<SubCategoriesCache["getSubCategories"]>): ReturnType<SubCategoriesCache["getSubCategories"]>;
+    getSubModels(props: {
+        modelId: Id64String;
+        categoryId?: Id64String;
+    } | {
+        categoryId: Id64String;
+        modelId: Id64String | undefined;
+    }): Observable<Id64Array>;
+    getSubModelsUnderElement(props: Props<ModeledElementsCache["getSubModelsUnderElement"]>): ReturnType<ModeledElementsCache["getSubModelsUnderElement"]>;
+    getElementsCount(props: Props<ModelCategoryElementsCountCache["getCategoryElementsCount"]>): ReturnType<ModelCategoryElementsCountCache["getCategoryElementsCount"]>;
+    getCategories(props: Props<ElementModelCategoriesCache["getModelCategoryIds"]>): ReturnType<ElementModelCategoriesCache["getModelCategoryIds"]>;
+    getModels(props: Props<ElementModelCategoriesCache["getCategoryElementModels"]>): ReturnType<ElementModelCategoriesCache["getCategoryElementModels"]>;
+    getAllCategoriesOfElements(): ReturnType<ElementModelCategoriesCache["getAllCategoriesOfElements"]>;
+    getCategoriesOfModelsTopMostElements(props: Props<BaseIdsCache["getCategoriesOfModelsTopMostElements"]>): ReturnType<BaseIdsCache["getCategoriesOfModelsTopMostElements"]>;
+    getSubCategoriesInfo(): ReturnType<SubCategoriesCache["getSubCategoriesInfo"]>;
+}
+//# sourceMappingURL=BaseIdsCache.d.ts.map

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

@@ -0,0 +1,148 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { mergeAll, mergeMap, toArray } from "rxjs";
+import { Guid } from "@itwin/core-bentley";
+import { ElementChildrenCache } from "./ElementChildrenCache.js";
+import { ElementModelCategoriesCache } from "./ElementModelCategoriesCache.js";
+import { ModelCategoryElementsCountCache } from "./ModelCategoryElementsCountCache.js";
+import { ModeledElementsCache } from "./ModeledElementsCache.js";
+import { SubCategoriesCache } from "./SubCategoriesCache.js";
+/** @internal */
+export class BaseIdsCache {
+    #queryExecutor;
+    #componentId;
+    #categoryElementCounts;
+    #elementChildrenCache;
+    #subCategoriesCache;
+    #modeledElementsCache;
+    #elementModelCategoriesCache;
+    constructor(props) {
+        this.#queryExecutor = props.queryExecutor;
+        this.#componentId = Guid.createValue();
+        this.#categoryElementCounts = new ModelCategoryElementsCountCache({
+            elementClassName: props.elementClassName,
+            componentId: this.#componentId,
+            queryExecutor: this.#queryExecutor,
+        });
+        this.#elementChildrenCache = new ElementChildrenCache({
+            queryExecutor: this.#queryExecutor,
+            elementClassName: props.elementClassName,
+            componentId: this.#componentId,
+        });
+        this.#subCategoriesCache = new SubCategoriesCache({
+            queryExecutor: this.#queryExecutor,
+            componentId: this.#componentId,
+        });
+        this.#modeledElementsCache = new ModeledElementsCache({
+            queryExecutor: this.#queryExecutor,
+            componentId: this.#componentId,
+            elementClassName: props.elementClassName,
+        });
+        this.#elementModelCategoriesCache = new ElementModelCategoriesCache({
+            queryExecutor: this.#queryExecutor,
+            componentId: this.#componentId,
+            elementClassName: props.elementClassName,
+            modeledElementsCache: this.#modeledElementsCache,
+        });
+    }
+    [Symbol.dispose]() {
+        this.#categoryElementCounts[Symbol.dispose]();
+    }
+    // Implement get sub-models method
+    getSubModels(props) {
+        if (props.modelId) {
+            if (props.categoryId) {
+                return this.#modeledElementsCache.getCategoryModeledElements({ modelId: props.modelId, categoryId: props.categoryId }).pipe(toArray());
+            }
+            return this.#elementModelCategoriesCache.getModelCategoryIds({ modelId: props.modelId }).pipe(mergeAll(), mergeMap((modelCategoryId) => this.#modeledElementsCache.getCategoryModeledElements({ modelId: props.modelId, categoryId: modelCategoryId })), toArray());
+        }
+        return this.#elementModelCategoriesCache.getCategoryElementModels({ categoryId: props.categoryId }).pipe(mergeAll(), mergeMap((modelId) => this.#modeledElementsCache.getCategoryModeledElements({ modelId, categoryId: props.categoryId })), toArray());
+    }
+    // Re-export cache methods
+    // ElementModelCategoriesCache methods
+    getAllModels() {
+        return this.#elementModelCategoriesCache.getAllModels();
+    }
+    getCategories(props) {
+        return this.#elementModelCategoriesCache.getModelCategoryIds(props);
+    }
+    getAllCategoriesOfElements() {
+        return this.#elementModelCategoriesCache.getAllCategoriesOfElements();
+    }
+    getModels(props) {
+        return this.#elementModelCategoriesCache.getCategoryElementModels(props);
+    }
+    getCategoriesOfModelsTopMostElements(props) {
+        return this.#elementModelCategoriesCache.getCategoriesOfModelsTopMostElements(props);
+    }
+    // ModelCategoryElementsCountCache methods
+    getElementsCount(props) {
+        return this.#categoryElementCounts.getCategoryElementsCount(props);
+    }
+    // ElementChildrenCache methods
+    getChildElementsTree(props) {
+        return this.#elementChildrenCache.getChildElementsTree(props);
+    }
+    getAllChildElementsCount(props) {
+        return this.#elementChildrenCache.getAllChildElementsCount(props);
+    }
+    // SubCategoriesCache methods
+    getSubCategories(props) {
+        return this.#subCategoriesCache.getSubCategories(props);
+    }
+    getSubCategoriesInfo() {
+        return this.#subCategoriesCache.getSubCategoriesInfo();
+    }
+    // ModeledElementsCache methods
+    getSubModelsUnderElement(props) {
+        return this.#modeledElementsCache.getSubModelsUnderElement(props);
+    }
+    getModeledElementsInfo() {
+        return this.#modeledElementsCache.getModeledElementsInfo();
+    }
+}
+/** @internal */
+export class BaseIdsCacheImpl {
+    #baseIdsCache;
+    constructor(props) {
+        this.#baseIdsCache = props.baseIdsCache;
+    }
+    [Symbol.dispose]() { }
+    // Implement IBaseIdsCache by re-exporting BaseIdsCache methods
+    getChildElementsTree(props) {
+        return this.#baseIdsCache.getChildElementsTree(props);
+    }
+    getAllChildElementsCount(props) {
+        return this.#baseIdsCache.getAllChildElementsCount(props);
+    }
+    getSubCategories(props) {
+        return this.#baseIdsCache.getSubCategories(props);
+    }
+    getSubModels(props) {
+        return this.#baseIdsCache.getSubModels(props);
+    }
+    getSubModelsUnderElement(props) {
+        return this.#baseIdsCache.getSubModelsUnderElement(props);
+    }
+    getElementsCount(props) {
+        return this.#baseIdsCache.getElementsCount(props);
+    }
+    getCategories(props) {
+        return this.#baseIdsCache.getCategories(props);
+    }
+    getModels(props) {
+        return this.#baseIdsCache.getModels(props);
+    }
+    getAllCategoriesOfElements() {
+        return this.#baseIdsCache.getAllCategoriesOfElements();
+    }
+    getCategoriesOfModelsTopMostElements(props) {
+        return this.#baseIdsCache.getCategoriesOfModelsTopMostElements(props);
+    }
+    getSubCategoriesInfo() {
+        return this.#baseIdsCache.getSubCategoriesInfo();
+    }
+}
+//# sourceMappingURL=BaseIdsCache.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"BaseIdsCache.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/caches/BaseIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACnD,OAAO,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AAC3C,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AACjE,OAAO,EAAE,2BAA2B,EAAE,MAAM,kCAAkC,CAAC;AAC/E,OAAO,EAAE,+BAA+B,EAAE,MAAM,sCAAsC,CAAC;AACvF,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AACjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAc7D,gBAAgB;AAChB,MAAM,OAAO,YAAY;IACvB,cAAc,CAA6B;IAC3C,YAAY,CAAa;IAChB,sBAAsB,CAAkC;IACxD,qBAAqB,CAAuB;IAC5C,mBAAmB,CAAqB;IACxC,qBAAqB,CAAuB;IAC5C,4BAA4B,CAA8B;IAEnE,YAAY,KAAwB;QAClC,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACvC,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC;YAChE,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;YACxC,WAAW,EAAE,IAAI,CAAC,YAAY;YAC9B,aAAa,EAAE,IAAI,CAAC,cAAc;SACnC,CAAC,CAAC;QACH,IAAI,CAAC,qBAAqB,GAAG,IAAI,oBAAoB,CAAC;YACpD,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;YACxC,WAAW,EAAE,IAAI,CAAC,YAAY;SAC/B,CAAC,CAAC;QACH,IAAI,CAAC,mBAAmB,GAAG,IAAI,kBAAkB,CAAC;YAChD,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,WAAW,EAAE,IAAI,CAAC,YAAY;SAC/B,CAAC,CAAC;QACH,IAAI,CAAC,qBAAqB,GAAG,IAAI,oBAAoB,CAAC;YACpD,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,WAAW,EAAE,IAAI,CAAC,YAAY;YAC9B,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;SACzC,CAAC,CAAC;QACH,IAAI,CAAC,4BAA4B,GAAG,IAAI,2BAA2B,CAAC;YAClE,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,WAAW,EAAE,IAAI,CAAC,YAAY;YAC9B,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;YACxC,oBAAoB,EAAE,IAAI,CAAC,qBAAqB;SACjD,CAAC,CAAC;IACL,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAED,kCAAkC;IAE3B,YAAY,CACjB,KAAqH;QAErH,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;gBACrB,OAAO,IAAI,CAAC,qBAAqB,CAAC,0BAA0B,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YACzI,CAAC;YAED,OAAO,IAAI,CAAC,4BAA4B,CAAC,mBAAmB,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAC3F,QAAQ,EAAE,EACV,QAAQ,CAAC,CAAC,eAAe,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,0BAA0B,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAQ,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC,EAC9I,OAAO,EAAE,CACV,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,4BAA4B,CAAC,wBAAwB,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,UAAW,EAAE,CAAC,CAAC,IAAI,CACvG,QAAQ,EAAE,EACV,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,0BAA0B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,UAAW,EAAE,CAAC,CAAC,EACxH,OAAO,EAAE,CACV,CAAC;IACJ,CAAC;IAED,0BAA0B;IAE1B,sCAAsC;IAE/B,YAAY;QACjB,OAAO,IAAI,CAAC,4BAA4B,CAAC,YAAY,EAAE,CAAC;IAC1D,CAAC;IAEM,aAAa,CAAC,KAAgE;QACnF,OAAO,IAAI,CAAC,4BAA4B,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;IACtE,CAAC;IAEM,0BAA0B;QAC/B,OAAO,IAAI,CAAC,4BAA4B,CAAC,0BAA0B,EAAE,CAAC;IACxE,CAAC;IAEM,SAAS,CAAC,KAAqE;QACpF,OAAO,IAAI,CAAC,4BAA4B,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;IAC3E,CAAC;IAEM,oCAAoC,CACzC,KAAiF;QAEjF,OAAO,IAAI,CAAC,4BAA4B,CAAC,oCAAoC,CAAC,KAAK,CAAC,CAAC;IACvF,CAAC;IAED,0CAA0C;IAEnC,gBAAgB,CACrB,KAAyE;QAEzE,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;IACrE,CAAC;IAED,+BAA+B;IAExB,oBAAoB,CAAC,KAA0D;QACpF,OAAO,IAAI,CAAC,qBAAqB,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IAChE,CAAC;IAEM,wBAAwB,CAC7B,KAA8D;QAE9D,OAAO,IAAI,CAAC,qBAAqB,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;IACpE,CAAC;IAED,6BAA6B;IAEtB,gBAAgB,CAAC,KAAoD;QAC1E,OAAO,IAAI,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;IAC1D,CAAC;IAEM,oBAAoB;QACzB,OAAO,IAAI,CAAC,mBAAmB,CAAC,oBAAoB,EAAE,CAAC;IACzD,CAAC;IAED,+BAA+B;IAExB,wBAAwB,CAC7B,KAA8D;QAE9D,OAAO,IAAI,CAAC,qBAAqB,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;IACpE,CAAC;IAEM,sBAAsB;QAC3B,OAAO,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,EAAE,CAAC;IAC7D,CAAC;CACF;AAOD,gBAAgB;AAChB,MAAM,OAAO,gBAAgB;IAC3B,aAAa,CAAe;IAC5B,YAAY,KAA4B;QACtC,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,YAAY,CAAC;IAC1C,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAI,CAAC;IAC5B,+DAA+D;IAExD,oBAAoB,CAAC,KAA0D;QACpF,OAAO,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC;IAEM,wBAAwB,CAC7B,KAA8D;QAE9D,OAAO,IAAI,CAAC,aAAa,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;IAC5D,CAAC;IAEM,gBAAgB,CAAC,KAAoD;QAC1E,OAAO,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACpD,CAAC;IAEM,YAAY,CACjB,KAAqH;QAErH,OAAO,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IAChD,CAAC;IAEM,wBAAwB,CAC7B,KAA8D;QAE9D,OAAO,IAAI,CAAC,aAAa,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;IAC5D,CAAC;IAEM,gBAAgB,CACrB,KAAyE;QAEzE,OAAO,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACpD,CAAC;IAEM,aAAa,CAAC,KAAgE;QACnF,OAAO,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;IACjD,CAAC;IAEM,SAAS,CAAC,KAAqE;QACpF,OAAO,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC7C,CAAC;IAEM,0BAA0B;QAC/B,OAAO,IAAI,CAAC,aAAa,CAAC,0BAA0B,EAAE,CAAC;IACzD,CAAC;IAEM,oCAAoC,CACzC,KAAkE;QAElE,OAAO,IAAI,CAAC,aAAa,CAAC,oCAAoC,CAAC,KAAK,CAAC,CAAC;IACxE,CAAC;IACM,oBAAoB;QACzB,OAAO,IAAI,CAAC,aAAa,CAAC,oBAAoB,EAAE,CAAC;IACnD,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 { mergeAll, mergeMap, toArray } from \"rxjs\";\nimport { Guid } from \"@itwin/core-bentley\";\nimport { ElementChildrenCache } from \"./ElementChildrenCache.js\";\nimport { ElementModelCategoriesCache } from \"./ElementModelCategoriesCache.js\";\nimport { ModelCategoryElementsCountCache } from \"./ModelCategoryElementsCountCache.js\";\nimport { ModeledElementsCache } from \"./ModeledElementsCache.js\";\nimport { SubCategoriesCache } from \"./SubCategoriesCache.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { Props } from \"@itwin/presentation-shared\";\n\n/** @internal */\nexport interface BaseIdsCacheProps {\n  queryExecutor: LimitingECSqlQueryExecutor;\n  elementClassName: string;\n  type: \"2d\" | \"3d\";\n}\n\n/** @internal */\nexport class BaseIdsCache implements Disposable {\n  #queryExecutor: LimitingECSqlQueryExecutor;\n  #componentId: GuidString;\n  readonly #categoryElementCounts: ModelCategoryElementsCountCache;\n  readonly #elementChildrenCache: ElementChildrenCache;\n  readonly #subCategoriesCache: SubCategoriesCache;\n  readonly #modeledElementsCache: ModeledElementsCache;\n  readonly #elementModelCategoriesCache: ElementModelCategoriesCache;\n\n  constructor(props: BaseIdsCacheProps) {\n    this.#queryExecutor = props.queryExecutor;\n    this.#componentId = Guid.createValue();\n    this.#categoryElementCounts = new ModelCategoryElementsCountCache({\n      elementClassName: props.elementClassName,\n      componentId: this.#componentId,\n      queryExecutor: this.#queryExecutor,\n    });\n    this.#elementChildrenCache = new ElementChildrenCache({\n      queryExecutor: this.#queryExecutor,\n      elementClassName: props.elementClassName,\n      componentId: this.#componentId,\n    });\n    this.#subCategoriesCache = new SubCategoriesCache({\n      queryExecutor: this.#queryExecutor,\n      componentId: this.#componentId,\n    });\n    this.#modeledElementsCache = new ModeledElementsCache({\n      queryExecutor: this.#queryExecutor,\n      componentId: this.#componentId,\n      elementClassName: props.elementClassName,\n    });\n    this.#elementModelCategoriesCache = new ElementModelCategoriesCache({\n      queryExecutor: this.#queryExecutor,\n      componentId: this.#componentId,\n      elementClassName: props.elementClassName,\n      modeledElementsCache: this.#modeledElementsCache,\n    });\n  }\n\n  public [Symbol.dispose]() {\n    this.#categoryElementCounts[Symbol.dispose]();\n  }\n\n  // Implement get sub-models method\n\n  public getSubModels(\n    props: { modelId: Id64String; categoryId?: Id64String } | { categoryId: Id64String; modelId: Id64String | undefined },\n  ): Observable<Id64Array> {\n    if (props.modelId) {\n      if (props.categoryId) {\n        return this.#modeledElementsCache.getCategoryModeledElements({ modelId: props.modelId, categoryId: props.categoryId }).pipe(toArray());\n      }\n\n      return this.#elementModelCategoriesCache.getModelCategoryIds({ modelId: props.modelId }).pipe(\n        mergeAll(),\n        mergeMap((modelCategoryId) => this.#modeledElementsCache.getCategoryModeledElements({ modelId: props.modelId!, categoryId: modelCategoryId })),\n        toArray(),\n      );\n    }\n\n    return this.#elementModelCategoriesCache.getCategoryElementModels({ categoryId: props.categoryId! }).pipe(\n      mergeAll(),\n      mergeMap((modelId) => this.#modeledElementsCache.getCategoryModeledElements({ modelId, categoryId: props.categoryId! })),\n      toArray(),\n    );\n  }\n\n  // Re-export cache methods\n\n  // ElementModelCategoriesCache methods\n\n  public getAllModels(): ReturnType<ElementModelCategoriesCache[\"getAllModels\"]> {\n    return this.#elementModelCategoriesCache.getAllModels();\n  }\n\n  public getCategories(props: Props<ElementModelCategoriesCache[\"getModelCategoryIds\"]>): ReturnType<ElementModelCategoriesCache[\"getModelCategoryIds\"]> {\n    return this.#elementModelCategoriesCache.getModelCategoryIds(props);\n  }\n\n  public getAllCategoriesOfElements(): ReturnType<ElementModelCategoriesCache[\"getAllCategoriesOfElements\"]> {\n    return this.#elementModelCategoriesCache.getAllCategoriesOfElements();\n  }\n\n  public getModels(props: Props<ElementModelCategoriesCache[\"getCategoryElementModels\"]>): ReturnType<ElementModelCategoriesCache[\"getCategoryElementModels\"]> {\n    return this.#elementModelCategoriesCache.getCategoryElementModels(props);\n  }\n\n  public getCategoriesOfModelsTopMostElements(\n    props: Props<ElementModelCategoriesCache[\"getCategoriesOfModelsTopMostElements\"]>,\n  ): ReturnType<ElementModelCategoriesCache[\"getCategoriesOfModelsTopMostElements\"]> {\n    return this.#elementModelCategoriesCache.getCategoriesOfModelsTopMostElements(props);\n  }\n\n  // ModelCategoryElementsCountCache methods\n\n  public getElementsCount(\n    props: Props<ModelCategoryElementsCountCache[\"getCategoryElementsCount\"]>,\n  ): ReturnType<ModelCategoryElementsCountCache[\"getCategoryElementsCount\"]> {\n    return this.#categoryElementCounts.getCategoryElementsCount(props);\n  }\n\n  // ElementChildrenCache methods\n\n  public getChildElementsTree(props: Props<ElementChildrenCache[\"getChildElementsTree\"]>): ReturnType<ElementChildrenCache[\"getChildElementsTree\"]> {\n    return this.#elementChildrenCache.getChildElementsTree(props);\n  }\n\n  public getAllChildElementsCount(\n    props: Props<ElementChildrenCache[\"getAllChildElementsCount\"]>,\n  ): ReturnType<ElementChildrenCache[\"getAllChildElementsCount\"]> {\n    return this.#elementChildrenCache.getAllChildElementsCount(props);\n  }\n\n  // SubCategoriesCache methods\n\n  public getSubCategories(props: Props<SubCategoriesCache[\"getSubCategories\"]>): ReturnType<SubCategoriesCache[\"getSubCategories\"]> {\n    return this.#subCategoriesCache.getSubCategories(props);\n  }\n\n  public getSubCategoriesInfo(): ReturnType<SubCategoriesCache[\"getSubCategoriesInfo\"]> {\n    return this.#subCategoriesCache.getSubCategoriesInfo();\n  }\n\n  // ModeledElementsCache methods\n\n  public getSubModelsUnderElement(\n    props: Props<ModeledElementsCache[\"getSubModelsUnderElement\"]>,\n  ): ReturnType<ModeledElementsCache[\"getSubModelsUnderElement\"]> {\n    return this.#modeledElementsCache.getSubModelsUnderElement(props);\n  }\n\n  public getModeledElementsInfo(): ReturnType<ModeledElementsCache[\"getModeledElementsInfo\"]> {\n    return this.#modeledElementsCache.getModeledElementsInfo();\n  }\n}\n\n/** @internal */\nexport interface BaseIdsCacheImplProps {\n  baseIdsCache: BaseIdsCache;\n}\n\n/** @internal */\nexport class BaseIdsCacheImpl implements Disposable {\n  #baseIdsCache: BaseIdsCache;\n  constructor(props: BaseIdsCacheImplProps) {\n    this.#baseIdsCache = props.baseIdsCache;\n  }\n\n  public [Symbol.dispose]() {}\n  // Implement IBaseIdsCache by re-exporting BaseIdsCache methods\n\n  public getChildElementsTree(props: Props<ElementChildrenCache[\"getChildElementsTree\"]>): ReturnType<ElementChildrenCache[\"getChildElementsTree\"]> {\n    return this.#baseIdsCache.getChildElementsTree(props);\n  }\n\n  public getAllChildElementsCount(\n    props: Props<ElementChildrenCache[\"getAllChildElementsCount\"]>,\n  ): ReturnType<ElementChildrenCache[\"getAllChildElementsCount\"]> {\n    return this.#baseIdsCache.getAllChildElementsCount(props);\n  }\n\n  public getSubCategories(props: Props<SubCategoriesCache[\"getSubCategories\"]>): ReturnType<SubCategoriesCache[\"getSubCategories\"]> {\n    return this.#baseIdsCache.getSubCategories(props);\n  }\n\n  public getSubModels(\n    props: { modelId: Id64String; categoryId?: Id64String } | { categoryId: Id64String; modelId: Id64String | undefined },\n  ): Observable<Id64Array> {\n    return this.#baseIdsCache.getSubModels(props);\n  }\n\n  public getSubModelsUnderElement(\n    props: Props<ModeledElementsCache[\"getSubModelsUnderElement\"]>,\n  ): ReturnType<ModeledElementsCache[\"getSubModelsUnderElement\"]> {\n    return this.#baseIdsCache.getSubModelsUnderElement(props);\n  }\n\n  public getElementsCount(\n    props: Props<ModelCategoryElementsCountCache[\"getCategoryElementsCount\"]>,\n  ): ReturnType<ModelCategoryElementsCountCache[\"getCategoryElementsCount\"]> {\n    return this.#baseIdsCache.getElementsCount(props);\n  }\n\n  public getCategories(props: Props<ElementModelCategoriesCache[\"getModelCategoryIds\"]>): ReturnType<ElementModelCategoriesCache[\"getModelCategoryIds\"]> {\n    return this.#baseIdsCache.getCategories(props);\n  }\n\n  public getModels(props: Props<ElementModelCategoriesCache[\"getCategoryElementModels\"]>): ReturnType<ElementModelCategoriesCache[\"getCategoryElementModels\"]> {\n    return this.#baseIdsCache.getModels(props);\n  }\n\n  public getAllCategoriesOfElements(): ReturnType<ElementModelCategoriesCache[\"getAllCategoriesOfElements\"]> {\n    return this.#baseIdsCache.getAllCategoriesOfElements();\n  }\n\n  public getCategoriesOfModelsTopMostElements(\n    props: Props<BaseIdsCache[\"getCategoriesOfModelsTopMostElements\"]>,\n  ): ReturnType<BaseIdsCache[\"getCategoriesOfModelsTopMostElements\"]> {\n    return this.#baseIdsCache.getCategoriesOfModelsTopMostElements(props);\n  }\n  public getSubCategoriesInfo(): ReturnType<SubCategoriesCache[\"getSubCategoriesInfo\"]> {\n    return this.#baseIdsCache.getSubCategoriesInfo();\n  }\n}\n"]}

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

@@ -0,0 +1,30 @@
+import type { Observable } from "rxjs";
+import type { GuidString, Id64Arg, Id64String } from "@itwin/core-bentley";
+import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+import type { ChildrenTree } from "../Utils.js";
+interface ElementChildrenCacheProps {
+    queryExecutor: LimitingECSqlQueryExecutor;
+    elementClassName: string;
+    componentId: GuidString;
+}
+/** @internal */
+export declare class ElementChildrenCache {
+    #private;
+    constructor(props: ElementChildrenCacheProps);
+    private queryChildren;
+    private getChildElementsTreeFromMap;
+    private getChildElementsCountMap;
+    /**
+     * Populates #childElementsLoadingMap with promises. When these promises resolve, they will populate #childElementsMap with values and delete entries from #childElementsLoadingMap.
+     */
+    private createChildElementsLoadingMapEntries;
+    private createChildElementsMapEntries;
+    getChildElementsTree({ elementIds }: {
+        elementIds: Id64Arg;
+    }): Observable<ChildrenTree>;
+    getAllChildElementsCount({ elementIds }: {
+        elementIds: Id64Arg;
+    }): Observable<Map<Id64String, number>>;
+}
+export {};
+//# sourceMappingURL=ElementChildrenCache.d.ts.map