@itwin/tree-widget-react 3.2.4 → 3.3.0

package/lib/esm/components/trees/models-tree/internal/AlwaysAndNeverDrawnElementInfo.js.map

@@ -1 +1 @@
-{"version":3,"file":"AlwaysAndNeverDrawnElementInfo.js","sourceRoot":"","sources":["../../../../../../src/components/trees/models-tree/internal/AlwaysAndNeverDrawnElementInfo.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EACL,eAAe,EACf,YAAY,EACZ,KAAK,EACL,MAAM,EACN,KAAK,EACL,IAAI,EACJ,gBAAgB,EAChB,GAAG,EACH,KAAK,EACL,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,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAmB/C,MAAM,CAAC,MAAM,wBAAwB,GAAG,EAAE,CAAC;AAE3C,MAAM,OAAO,8BAA8B;IAUzC,YAA6B,SAAmB;QAAnB,cAAS,GAAT,SAAS,CAAU;QANxC,oBAAe,GAAG,IAAI,OAAO,EAAQ,CAAC;QAGtC,cAAS,GAAG,IAAI,OAAO,EAAW,CAAC;QACnC,iBAAY,GAAG,IAAI,OAAO,EAAQ,CAAC;QAGzC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC;YAClD,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,oBAAoB;YAC1C,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW;YACxC,EAAE,EAAE,aAAa;SAClB,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,0BAA0B,CAAC;YACjD,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,mBAAmB;YACzC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU;YACvC,EAAE,EAAE,YAAY;SACjB,CAAC,CAAC;QACH,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;YACpB,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;YAC7B,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE;YAC5B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC3E,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE;aACrC,CAAC;SACH,CAAC;IACJ,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,WAAW,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAA0E;QACzH,MAAM,KAAK,GAAG,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;QAC1E,MAAM,WAAW,GAAG,UAAU;YAC5B,CAAC,CAAC,CAAC,KAA6B,EAAE,EAAE;gBAChC,OAAO,KAAK,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC;YAC3D,CAAC;YACH,CAAC,CAAC,CAAC,KAA6B,EAAE,EAAE;gBAChC,MAAM,UAAU,GAAG,KAAK,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;gBACvC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAc,CAAC;gBACvC,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE;oBAC5C,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;iBACvC;gBACD,OAAO,QAAQ,CAAC;YAClB,CAAC,CAAC;QAEN,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;IACtC,CAAC;IAEO,0BAA0B,CAAC,KAAgF;QACjH,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QAC1B,MAAM,aAAa,GAAG,IAAI,eAAe,CAAyB,SAAS,CAAC,CAAC;QAE7E,MAAM,GAAG,GAAG,KAAK,CACf,gBAAgB,CACd,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,EACvC,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAC3C,EACD,IAAI,CAAC,YAAY,CAClB,CAAC,IAAI;QACJ,4CAA4C;QAC5C,SAAS,CAAC,SAAS,CAAC;QACpB,oDAAoD;QACpD,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC;QAC/B,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,KAAK,CAAC,MAAM,EAAE,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;QAClF,2EAA2E;QAC3E,KAAK,CAAC;YACJ,SAAS,EAAE,GAAG,EAAE,CAAC,aAAa;YAC9B,mBAAmB,EAAE,KAAK;SAC3B,CAAC;QACF,sCAAsC;QACtC,KAAK,CAAC,CAAC,CAAC,EAAmB,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CACnC,CAAC;QACF,OAAO,GAAG,CAAC;IACb,CAAC;IAEM,OAAO;QACZ,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,GAAwB,EAAE,SAAiB;QACpF,MAAM,WAAW,GAAG,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QACnF,OAAO,WAAW,CAAC,IAAI,CACrB,MAAM,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YACpB,IAAI,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACnC,IAAI,CAAC,KAAK,EAAE;gBACV,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;gBAClB,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;aAC/B;YACD,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,OAAO,KAAK,CAAC;QACf,CAAC,EAAE,IAAI,GAAG,EAAwC,CAAC,CACpD,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,UAAqB,EAAE,SAAiB;QAC/D,MAAM,QAAQ,GAAG,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACjE,MAAM,MAAM,GAAG,QAAQ,CAAC,iBAAiB,CACvC;YACE,IAAI,EAAE;gBACJ;;;;;;;;;;;;;;;;;;;;SAoBD;aACA;YACD,KAAK,EAAE;;;;OAIR;YACC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;SACjD,EACD;YACE,YAAY,EAAE,+BAA+B,SAAS,EAAE;SACzD,CACF,CAAC;QAEF,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IAC3H,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  debounceTime,\n  EMPTY,\n  filter,\n  first,\n  from,\n  fromEventPattern,\n  map,\n  merge,\n  reduce,\n  scan,\n  share,\n  shareReplay,\n  startWith,\n  Subject,\n  switchMap,\n  take,\n  takeUntil,\n  tap,\n} from \"rxjs\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { pushToMap } from \"../../common/Utils\";\n\nimport type { Observable, Subscription } from \"rxjs\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { BeEvent, Id64Array, Id64Set, Id64String, IDisposable } from \"@itwin/core-bentley\";\n\ninterface ElementInfo {\n  elementId: Id64String;\n  modelId: Id64String;\n  categoryId: Id64String;\n}\n\ntype CacheEntry = Map<Id64String, Map<Id64String, Id64Set>>;\n\nexport interface AlwaysOrNeverDrawnElementsQueryProps {\n  modelId: Id64String;\n  categoryId?: Id64String;\n}\n\nexport const SET_CHANGE_DEBOUNCE_TIME = 20;\n\nexport class AlwaysAndNeverDrawnElementInfo implements IDisposable {\n  private _subscriptions: Subscription[];\n  private _alwaysDrawn: Observable<CacheEntry>;\n  private _neverDrawn: Observable<CacheEntry>;\n  private _disposeSubject = new Subject<void>();\n\n  private _suppressors: Observable<number>;\n  private _suppress = new Subject<boolean>();\n  private _forceUpdate = new Subject<void>();\n\n  constructor(private readonly _viewport: Viewport) {\n    this._alwaysDrawn = this.createCacheEntryObservable({\n      event: this._viewport.onAlwaysDrawnChanged,\n      getSet: () => this._viewport.alwaysDrawn,\n      id: \"alwaysDrawn\",\n    });\n    this._neverDrawn = this.createCacheEntryObservable({\n      event: this._viewport.onNeverDrawnChanged,\n      getSet: () => this._viewport.neverDrawn,\n      id: \"neverDrawn\",\n    });\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 = [\n      this._alwaysDrawn.subscribe(),\n      this._neverDrawn.subscribe(),\n      this._suppressors.pipe(filter((suppressors) => suppressors === 0)).subscribe({\n        next: () => this._forceUpdate.next(),\n      }),\n    ];\n  }\n\n  public suppressChangeEvents() {\n    this._suppress.next(true);\n  }\n\n  public resumeChangeEvents() {\n    this._suppress.next(false);\n  }\n\n  public getElements({ setType, modelId, categoryId }: { setType: \"always\" | \"never\" } & AlwaysOrNeverDrawnElementsQueryProps): Observable<Id64Set> {\n    const cache = setType === \"always\" ? this._alwaysDrawn : this._neverDrawn;\n    const getElements = categoryId\n      ? (entry: CacheEntry | undefined) => {\n          return entry?.get(modelId)?.get(categoryId) ?? new Set();\n        }\n      : (entry: CacheEntry | undefined) => {\n          const modelEntry = entry?.get(modelId);\n          const elements = new Set<Id64String>();\n          for (const set of modelEntry?.values() ?? []) {\n            set.forEach((id) => elements.add(id));\n          }\n          return elements;\n        };\n\n    return cache.pipe(map(getElements));\n  }\n\n  private createCacheEntryObservable(props: { event: BeEvent<() => void>; getSet(): Id64Set | undefined; id: string }) {\n    const event = props.event;\n    const resultSubject = new BehaviorSubject<CacheEntry | undefined>(undefined);\n\n    const obs = merge(\n      fromEventPattern(\n        (handler) => event.addListener(handler),\n        (handler) => event.removeListener(handler),\n      ),\n      this._forceUpdate,\n    ).pipe(\n      // Fire the observable once at the beginning\n      startWith(undefined),\n      // Stop listening to events when dispose() is called\n      takeUntil(this._disposeSubject),\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(props.getSet(), props.id)),\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 CacheEntry => !!x),\n    );\n    return obs;\n  }\n\n  public dispose(): void {\n    this._subscriptions.forEach((x) => x.unsubscribe());\n    this._subscriptions = [];\n    this._disposeSubject.next();\n  }\n\n  private queryAlwaysOrNeverDrawnElementInfo(set: Id64Set | undefined, requestId: string): Observable<CacheEntry> {\n    const elementInfo = set?.size ? this.queryElementInfo([...set], requestId) : EMPTY;\n    return elementInfo.pipe(\n      reduce((state, val) => {\n        let entry = state.get(val.modelId);\n        if (!entry) {\n          entry = new Map();\n          state.set(val.modelId, entry);\n        }\n        pushToMap(entry, val.categoryId, val.elementId);\n        return state;\n      }, new Map<Id64String, Map<Id64String, Id64Set>>()),\n    );\n  }\n\n  private queryElementInfo(elementIds: Id64Array, requestId: string): Observable<ElementInfo> {\n    const executor = createECSqlQueryExecutor(this._viewport.iModel);\n    const reader = executor.createQueryReader(\n      {\n        ctes: [\n          `\n        ElementInfo(elementId, modelId, categoryId, parentId) AS (\n          SELECT\n            ECInstanceId elementId,\n            Model.Id modelId,\n            Category.Id categoryId,\n            Parent.Id parentId\n          FROM bis.GeometricElement3d\n          WHERE InVirtualSet(?, ECInstanceId)\n\n          UNION ALL\n\n          SELECT\n            e.elementId,\n            e.modelId,\n            p.Category.Id categoryId,\n            p.Parent.Id parentId\n          FROM bis.GeometricElement3d p\n          JOIN ElementInfo e ON p.ECInstanceId = e.parentId\n        )\n        `,\n        ],\n        ecsql: `\n        SELECT elementId, modelId, categoryId\n        FROM ElementInfo\n        WHERE parentId IS NULL\n      `,\n        bindings: [{ type: \"idset\", value: elementIds }],\n      },\n      {\n        restartToken: `ModelsTreeVisibilityHandler/${requestId}`,\n      },\n    );\n\n    return from(reader).pipe(map((row) => ({ elementId: row.elementId, modelId: row.modelId, categoryId: row.categoryId })));\n  }\n}\n"]}
+{"version":3,"file":"AlwaysAndNeverDrawnElementInfo.js","sourceRoot":"","sources":["../../../../../../src/components/trees/models-tree/internal/AlwaysAndNeverDrawnElementInfo.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EACL,eAAe,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAC7I,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,GAChC,MAAM,MAAM,CAAC;AACd,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAmBlD,MAAM,CAAC,MAAM,wBAAwB,GAAG,EAAE,CAAC;AAE3C,MAAM,OAAO,8BAA8B;IAUzC,YAA6B,SAAmB;QAAnB,cAAS,GAAT,SAAS,CAAU;QANxC,oBAAe,GAAG,IAAI,OAAO,EAAQ,CAAC;QAGtC,cAAS,GAAG,IAAI,OAAO,EAAW,CAAC;QACnC,iBAAY,GAAG,IAAI,OAAO,EAAQ,CAAC;QAGzC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,0BAA0B,CAAC;YAClD,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,oBAAoB;YAC1C,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW;YACxC,EAAE,EAAE,aAAa;SAClB,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,0BAA0B,CAAC;YACjD,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,mBAAmB;YACzC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU;YACvC,EAAE,EAAE,YAAY;SACjB,CAAC,CAAC;QACH,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;YACpB,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;YAC7B,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE;YAC5B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC3E,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE;aACrC,CAAC;SACH,CAAC;IACJ,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,WAAW,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAA0E;QACzH,MAAM,KAAK,GAAG,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;QAC1E,MAAM,WAAW,GAAG,UAAU;YAC5B,CAAC,CAAC,CAAC,KAA6B,EAAE,EAAE;gBAChC,OAAO,KAAK,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC;YAC3D,CAAC;YACH,CAAC,CAAC,CAAC,KAA6B,EAAE,EAAE;gBAChC,MAAM,UAAU,GAAG,KAAK,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;gBACvC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAc,CAAC;gBACvC,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE;oBAC5C,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;iBACvC;gBACD,OAAO,QAAQ,CAAC;YAClB,CAAC,CAAC;QAEN,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;IACtC,CAAC;IAEO,0BAA0B,CAAC,KAAgF;QACjH,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QAC1B,MAAM,aAAa,GAAG,IAAI,eAAe,CAAyB,SAAS,CAAC,CAAC;QAE7E,MAAM,GAAG,GAAG,KAAK,CACf,gBAAgB,CACd,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,EACvC,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAC3C,EACD,IAAI,CAAC,YAAY,CAClB,CAAC,IAAI;QACJ,4CAA4C;QAC5C,SAAS,CAAC,SAAS,CAAC;QACpB,oDAAoD;QACpD,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC;QAC/B,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,KAAK,CAAC,MAAM,EAAE,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;QAClF,2EAA2E;QAC3E,KAAK,CAAC;YACJ,SAAS,EAAE,GAAG,EAAE,CAAC,aAAa;YAC9B,mBAAmB,EAAE,KAAK;SAC3B,CAAC;QACF,sCAAsC;QACtC,KAAK,CAAC,CAAC,CAAC,EAAmB,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CACnC,CAAC;QACF,OAAO,GAAG,CAAC;IACb,CAAC;IAEM,OAAO;QACZ,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,GAAwB,EAAE,SAAiB;QACpF,MAAM,WAAW,GAAG,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QACnF,OAAO,WAAW,CAAC,IAAI,CACrB,MAAM,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YACpB,IAAI,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACnC,IAAI,CAAC,KAAK,EAAE;gBACV,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;gBAClB,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;aAC/B;YACD,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,OAAO,KAAK,CAAC;QACf,CAAC,EAAE,IAAI,GAAG,EAAwC,CAAC,CACpD,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,UAAqB,EAAE,SAAiB;QAC/D,MAAM,QAAQ,GAAG,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACjE,MAAM,MAAM,GAAG,QAAQ,CAAC,iBAAiB,CACvC;YACE,IAAI,EAAE;gBACJ;;;;;;;;;;;;;;;;;;;;SAoBD;aACA;YACD,KAAK,EAAE;;;;OAIR;YACC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;SACjD,EACD;YACE,YAAY,EAAE,+BAA+B,SAAS,EAAE;SACzD,CACF,CAAC;QAEF,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IAC3H,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, debounceTime, EMPTY, filter, first, from, fromEventPattern, map, merge, reduce, scan, share, shareReplay, startWith, Subject,\n  switchMap, take, takeUntil, tap,\n} from \"rxjs\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { pushToMap } from \"../../common/Utils.js\";\n\nimport type { Observable, Subscription } from \"rxjs\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { BeEvent, Id64Array, Id64Set, Id64String, IDisposable } from \"@itwin/core-bentley\";\n\ninterface ElementInfo {\n  elementId: Id64String;\n  modelId: Id64String;\n  categoryId: Id64String;\n}\n\ntype CacheEntry = Map<Id64String, Map<Id64String, Id64Set>>;\n\nexport interface AlwaysOrNeverDrawnElementsQueryProps {\n  modelId: Id64String;\n  categoryId?: Id64String;\n}\n\nexport const SET_CHANGE_DEBOUNCE_TIME = 20;\n\nexport class AlwaysAndNeverDrawnElementInfo implements IDisposable {\n  private _subscriptions: Subscription[];\n  private _alwaysDrawn: Observable<CacheEntry>;\n  private _neverDrawn: Observable<CacheEntry>;\n  private _disposeSubject = new Subject<void>();\n\n  private _suppressors: Observable<number>;\n  private _suppress = new Subject<boolean>();\n  private _forceUpdate = new Subject<void>();\n\n  constructor(private readonly _viewport: Viewport) {\n    this._alwaysDrawn = this.createCacheEntryObservable({\n      event: this._viewport.onAlwaysDrawnChanged,\n      getSet: () => this._viewport.alwaysDrawn,\n      id: \"alwaysDrawn\",\n    });\n    this._neverDrawn = this.createCacheEntryObservable({\n      event: this._viewport.onNeverDrawnChanged,\n      getSet: () => this._viewport.neverDrawn,\n      id: \"neverDrawn\",\n    });\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 = [\n      this._alwaysDrawn.subscribe(),\n      this._neverDrawn.subscribe(),\n      this._suppressors.pipe(filter((suppressors) => suppressors === 0)).subscribe({\n        next: () => this._forceUpdate.next(),\n      }),\n    ];\n  }\n\n  public suppressChangeEvents() {\n    this._suppress.next(true);\n  }\n\n  public resumeChangeEvents() {\n    this._suppress.next(false);\n  }\n\n  public getElements({ setType, modelId, categoryId }: { setType: \"always\" | \"never\" } & AlwaysOrNeverDrawnElementsQueryProps): Observable<Id64Set> {\n    const cache = setType === \"always\" ? this._alwaysDrawn : this._neverDrawn;\n    const getElements = categoryId\n      ? (entry: CacheEntry | undefined) => {\n          return entry?.get(modelId)?.get(categoryId) ?? new Set();\n        }\n      : (entry: CacheEntry | undefined) => {\n          const modelEntry = entry?.get(modelId);\n          const elements = new Set<Id64String>();\n          for (const set of modelEntry?.values() ?? []) {\n            set.forEach((id) => elements.add(id));\n          }\n          return elements;\n        };\n\n    return cache.pipe(map(getElements));\n  }\n\n  private createCacheEntryObservable(props: { event: BeEvent<() => void>; getSet(): Id64Set | undefined; id: string }) {\n    const event = props.event;\n    const resultSubject = new BehaviorSubject<CacheEntry | undefined>(undefined);\n\n    const obs = merge(\n      fromEventPattern(\n        (handler) => event.addListener(handler),\n        (handler) => event.removeListener(handler),\n      ),\n      this._forceUpdate,\n    ).pipe(\n      // Fire the observable once at the beginning\n      startWith(undefined),\n      // Stop listening to events when dispose() is called\n      takeUntil(this._disposeSubject),\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(props.getSet(), props.id)),\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 CacheEntry => !!x),\n    );\n    return obs;\n  }\n\n  public dispose(): void {\n    this._subscriptions.forEach((x) => x.unsubscribe());\n    this._subscriptions = [];\n    this._disposeSubject.next();\n  }\n\n  private queryAlwaysOrNeverDrawnElementInfo(set: Id64Set | undefined, requestId: string): Observable<CacheEntry> {\n    const elementInfo = set?.size ? this.queryElementInfo([...set], requestId) : EMPTY;\n    return elementInfo.pipe(\n      reduce((state, val) => {\n        let entry = state.get(val.modelId);\n        if (!entry) {\n          entry = new Map();\n          state.set(val.modelId, entry);\n        }\n        pushToMap(entry, val.categoryId, val.elementId);\n        return state;\n      }, new Map<Id64String, Map<Id64String, Id64Set>>()),\n    );\n  }\n\n  private queryElementInfo(elementIds: Id64Array, requestId: string): Observable<ElementInfo> {\n    const executor = createECSqlQueryExecutor(this._viewport.iModel);\n    const reader = executor.createQueryReader(\n      {\n        ctes: [\n          `\n        ElementInfo(elementId, modelId, categoryId, parentId) AS (\n          SELECT\n            ECInstanceId elementId,\n            Model.Id modelId,\n            Category.Id categoryId,\n            Parent.Id parentId\n          FROM bis.GeometricElement3d\n          WHERE InVirtualSet(?, ECInstanceId)\n\n          UNION ALL\n\n          SELECT\n            e.elementId,\n            e.modelId,\n            p.Category.Id categoryId,\n            p.Parent.Id parentId\n          FROM bis.GeometricElement3d p\n          JOIN ElementInfo e ON p.ECInstanceId = e.parentId\n        )\n        `,\n        ],\n        ecsql: `\n        SELECT elementId, modelId, categoryId\n        FROM ElementInfo\n        WHERE parentId IS NULL\n      `,\n        bindings: [{ type: \"idset\", value: elementIds }],\n      },\n      {\n        restartToken: `ModelsTreeVisibilityHandler/${requestId}`,\n      },\n    );\n\n    return from(reader).pipe(map((row) => ({ elementId: row.elementId, modelId: row.modelId, categoryId: row.categoryId })));\n  }\n}\n"]}

package/lib/esm/components/trees/models-tree/internal/ModelsTreeIdsCache.d.ts

@@ -1,4 +1,4 @@
-import type { ModelsTreeDefinition } from "../ModelsTreeDefinition";
+import type { ModelsTreeDefinition } from "../ModelsTreeDefinition.js";
 import type { Id64Array, Id64String } from "@itwin/core-bentley";
 import type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
 type ModelsTreeHierarchyConfiguration = ConstructorParameters<typeof ModelsTreeDefinition>[0]["hierarchyConfig"];

package/lib/esm/components/trees/models-tree/internal/ModelsTreeIdsCache.js

@@ -4,7 +4,7 @@
  *--------------------------------------------------------------------------------------------*/
 import { bufferTime, filter, firstValueFrom, mergeAll, mergeMap, ReplaySubject, Subject } from "rxjs";
 import { assert } from "@itwin/core-bentley";
-import { pushToMap } from "../../common/Utils";
+import { pushToMap } from "../../common/Utils.js";
 /** @internal */
 export class ModelsTreeIdsCache {
     constructor(_queryExecutor, _hierarchyConfig) {

package/lib/esm/components/trees/models-tree/internal/ModelsTreeIdsCache.js.map

@@ -1 +1 @@
-{"version":3,"file":"ModelsTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../src/components/trees/models-tree/internal/ModelsTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAGhG,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACtG,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAqB/C,gBAAgB;AAChB,MAAM,OAAO,kBAAkB;IAS7B,YACU,cAA0C,EAC1C,gBAAkD;QADlD,mBAAc,GAAd,cAAc,CAA4B;QAC1C,qBAAgB,GAAhB,gBAAgB,CAAkC;QAE1D,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3H,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC,gBAAgB,GAAG,IAAI,GAAG,EAAE,CAAC;QAClC,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;IACrC,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,aAAa;QAC1B,MAAM,aAAa,GAAG;;;;;;;;;wCASc,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;;;;;;;KAUlF,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAChJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,iBAAiB,EAAE,GAAG,CAAC,iBAAiB,EAAE,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;SAChI;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,WAAW;QACxB,MAAM,WAAW,GAAG;;;;;;UAMd,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,6BAA6B,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,mCAAmC;KACjK,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAC9I,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;SAC9C;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,aAAa,KAAK,CAAC,KAAK,IAAI,EAAE;YACjC,MAAM,CAAC,YAAY,EAAE,uBAAuB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChE,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;oBAClD,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE;wBAChD,MAAM,WAAW,GAAgB;4BAC/B,aAAa,EAAE,OAAO,CAAC,QAAQ;4BAC/B,eAAe,EAAE,OAAO,CAAC,eAAe;4BACxC,aAAa,EAAE,IAAI,GAAG,EAAE;4BACxB,WAAW,EAAE,IAAI,GAAG,EAAE;yBACvB,CAAC;wBACF,IAAI,OAAO,CAAC,iBAAiB,EAAE;4BAC7B,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;yBACxD;wBACD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;qBACrC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA+B,CAAC;oBACtD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE;wBAC5C,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;qBAC7C;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YAEH,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE;gBACpF,IAAI,eAAe,EAAE;oBACnB,MAAM,iBAAiB,GAAG,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;oBAC5D,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC;oBAC5B,iBAAiB,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;iBAChD;aACF;YAED,KAAK,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,uBAAuB,EAAE;gBAC/D,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;oBAC/B,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChD,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;oBACtB,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC3C,CAAC,CAAC,CAAC;aACJ;YAED,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,kHAAkH;IAC3G,KAAK,CAAC,mBAAmB;QAC9B,IAAI,CAAC,iBAAiB,KAAK,CAAC,KAAK,IAAI,EAAE;YACrC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAClD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;YAC/C,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE;gBAC9C,IAAI,WAAW,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE;oBACpC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChC,IAAI,YAAY,GAAG,WAAW,CAAC,aAAa,CAAC;oBAC7C,OAAO,YAAY,EAAE;wBACnB,gBAAgB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;wBACnC,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,aAAa,CAAC;qBAC9D;iBACF;YACH,CAAC,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,gBAAgB,CAAC,CAAC;QAC/B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,kBAAkB,CAAC,gBAA8B;QAC5D,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;QAChD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE;oBACrC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACrC,OAAO,OAAO,CAAC;iBAChB;gBACD,OAAO,UAAU,CAAC;YACpB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,+GAA+G;IACxG,KAAK,CAAC,kBAAkB,CAAC,UAAqB;QACnD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,MAAM,YAAY,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,OAAO,IAAI,EAAE;YACX,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,EAAE,CAAC;YACrC,IAAI,SAAS,KAAK,SAAS,EAAE;gBAC3B,MAAM;aACP;YACD,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,WAAW,EAAE;gBAChB,SAAS;aACV;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;YACxC,YAAY,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,aAAa,CAAC,CAAC;SACjD;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,6GAA6G;IACtG,KAAK,CAAC,uBAAuB,CAAC,gBAA8B;QACjE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAElD,MAAM,gBAAgB,GAAG,IAAI,KAAK,EAAc,CAAC;QACjD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,gBAAgB,CAAC,eAAe,EAAE;oBACpC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACtC,OAAO,UAAU,CAAC;iBACnB;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,KAAK,EAAc,CAAC;QACzC,CAAC,GAAG,gBAAgB,EAAE,GAAG,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YAC/D,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,WAAW,IAAI,QAAQ,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEM,KAAK,CAAC,6BAA6B,CAAC,eAA2B;QACpE,IAAI,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QACvD,IAAI,CAAC,KAAK,EAAE;YACV,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;gBAClD,MAAM,MAAM,GAAG,IAAI,KAAK,EAAe,CAAC;gBACxC,IAAI,YAAY,GAA2B,eAAe,CAAC;gBAC3D,OAAO,YAAY,EAAE;oBACnB,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;oBAClD,IAAI,CAAC,UAAU,EAAE,eAAe,EAAE;wBAChC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;qBACjE;oBACD,YAAY,GAAG,UAAU,EAAE,aAAa,CAAC;iBAC1C;gBACD,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;SACnD;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,uBAAuB;QACpC,MAAM,KAAK,GAAG,SAAS,CAAC;;aAEf,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;KAEvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACxI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC;SAChE;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QACjC,MAAM,KAAK,GAAG,SAAS,CAAC;;aAEf,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;KAGvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACxI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAC5D;IACH,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,IAAI,CAAC,WAAW,KAAK,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,EAA6D,CAAC;YACxF,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,CAAC,KAAK,IAAI,EAAE;oBACV,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE;wBACvE,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACtC,IAAI,KAAK,EAAE;4BACT,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;yBAClC;6BAAM;4BACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;yBACjF;qBACF;gBACH,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,IAAI,CAAC,uBAAuB,EAAE,EAAE;wBAC5E,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACtC,IAAI,KAAK,EAAE;4BACT,KAAK,CAAC,YAAY,GAAG,YAAY,CAAC;yBACnC;6BAAM;4BACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;yBAClE;qBACF;gBACH,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YACH,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,OAAmB;QACjD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,UAAU,CAAC;QACvD,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAAC,OAAmB;QACnD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,OAAO,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,YAAY,IAAI,CAAC,CAAC;IACpD,CAAC;IAEM,KAAK,CAAC,2BAA2B,CAAC,OAAmB;QAC1D,IAAI,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC7C,IAAI,CAAC,KAAK,EAAE;YACV,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAG,IAAI,KAAK,EAAgC,CAAC;gBACzD,MAAM,YAAY,GAAG,CAAC,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC9D,KAAK,MAAM,CAAC,cAAc,EAAE,WAAW,CAAC,IAAI,YAAY,EAAE;oBACxD,IAAI,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;wBACxC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,6BAA6B,CAAC,cAAc,CAAC,CAAC;wBAC7E,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,WAAW,EAAE,EAAE,SAAS,EAAE,0BAA0B,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;qBACvF;iBACF;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,EAAE,CAAC;YAEL,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;SACzC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACtC,KAA6D;QAE7D,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAClD;YACE,IAAI,EAAE;gBACJ,SAAS,CAAC;;;qBAGC,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;oBAIhD,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,cAAc,OAAO,sBAAsB,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;;;;;;qBAM3G,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;WAGzD;aACF;YACD,KAAK,EAAE;;;;SAIN;SACF,EACD,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CACxD,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,KAAK,EAAkE,CAAC;QAC3F,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE;YAC9B,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,aAAa,EAAE,GAAG,CAAC,aAAa,EAAE,CAAC,CAAC;SACrG;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,KAAK,CAAC,8BAA8B,CAAC,UAAsB;QAChE,IAAI,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACnD,IAAI,CAAC,KAAK,EAAE;YACV,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAG,IAAI,GAAG,EAAc,CAAC;gBACrC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;gBAC9C,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,EAAE,OAAO,EAAE,EAAE;oBACzC,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;wBACxC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;qBACrB;gBACH,CAAC,CAAC,CAAC;gBAEH,MAAM,aAAa,GAAG,IAAI,KAAK,EAAgC,CAAC;gBAChE,KAAK,MAAM,eAAe,IAAI,CAAC,GAAG,MAAM,CAAC,EAAE;oBACzC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,eAAe,CAAC,CAAC;oBAC3E,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE;wBAClC,aAAa,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,EAAE,EAAE,SAAS,EAAE,yBAAyB,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;qBAC9F;iBACF;gBACD,OAAO,aAAa,CAAC;YACvB,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;SAC/C;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED,SAAS,mBAAmB,CAC1B,YAA0C,EAC1C,aAAuC,EACvC,EAAuF;IAEvF,MAAM,iBAAiB,GAAG,OAAO,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;IAC9G,iBAAiB;QACf,iBAAiB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YACzD,MAAM,gBAAgB,GAAG,YAAY,CAAC,GAAG,CAAC,cAAc,CAAE,CAAC;YAC3D,IAAI,EAAE,CAAC,cAAc,EAAE,gBAAgB,CAAC,KAAK,OAAO,EAAE;gBACpD,OAAO;aACR;YACD,mBAAmB,CAAC,YAAY,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,+BAA+B;IAKnC,YACU,OAE2E;QAF3E,YAAO,GAAP,OAAO,CAEoE;QAP7E,WAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;QAC5C,oBAAe,GAAG,IAAI,OAAO,EAAmD,CAAC;QAQvF,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,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EACpD,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,GAAG,UAAU,EAAE,CAAC,CAAC;gBAC3D,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9B,CAAC;SACF,CAAC,CAAC;IACP,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;IACnC,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,MAAM,QAAQ,GAAG,GAAG,OAAO,GAAG,UAAU,EAAE,CAAC;QAC3C,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,MAAM,KAAK,SAAS,EAAE;YACxB,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;SAC/B;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,cAAc,CAAC,MAAM,CAAC,CAAC;IAChC,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 type { Subscription } from \"rxjs\";\nimport { bufferTime, filter, firstValueFrom, mergeAll, mergeMap, ReplaySubject, Subject } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { pushToMap } from \"../../common/Utils\";\n\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\nimport type { ModelsTreeDefinition } from \"../ModelsTreeDefinition\";\nimport type { Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\n\ninterface SubjectInfo {\n  parentSubject: Id64String | undefined;\n  hideInHierarchy: boolean;\n  childSubjects: Id64Set;\n  childModels: Id64Set;\n}\n\ninterface ModelInfo {\n  categories: Id64Set;\n  elementCount: number;\n}\n\ntype ModelsTreeHierarchyConfiguration = ConstructorParameters<typeof ModelsTreeDefinition>[0][\"hierarchyConfig\"];\n\n/** @internal */\nexport class ModelsTreeIdsCache {\n  private readonly _categoryElementCounts: ModelCategoryElementsCountCache;\n  private _subjectInfos: Promise<Map<Id64String, SubjectInfo>> | undefined;\n  private _parentSubjectIds: Promise<Id64Array> | undefined; // the list should contain a subject id if its node should be shown as having children\n  private _modelInfos: Promise<Map<Id64String, ModelInfo>> | undefined;\n  private _modelKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath[]>>;\n  private _subjectKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath>>;\n  private _categoryKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath[]>>;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    private _hierarchyConfig: ModelsTreeHierarchyConfiguration,\n  ) {\n    this._categoryElementCounts = new ModelCategoryElementsCountCache(async (input) => this.queryCategoryElementCounts(input));\n    this._modelKeyPaths = new Map();\n    this._subjectKeyPaths = new Map();\n    this._categoryKeyPaths = new Map();\n  }\n\n  public [Symbol.dispose]() {\n    this._categoryElementCounts[Symbol.dispose]();\n  }\n\n  private async *querySubjects(): AsyncIterableIterator<{ id: Id64String; parentId?: Id64String; targetPartitionId?: Id64String; hideInHierarchy: boolean }> {\n    const subjectsQuery = `\n      SELECT\n        s.ECInstanceId id,\n        s.Parent.Id parentId,\n        (\n          SELECT m.ECInstanceId\n          FROM bis.GeometricModel3d m\n          WHERE m.ECInstanceId = HexToId(json_extract(s.JsonProperties, '$.Subject.Model.TargetPartition'))\n            AND NOT m.IsPrivate\n            AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)\n        ) targetPartitionId,\n        CASE\n          WHEN (\n            json_extract(s.JsonProperties, '$.Subject.Job.Bridge') IS NOT NULL\n            OR json_extract(s.JsonProperties, '$.Subject.Model.Type') = 'Hierarchy'\n          ) THEN 1\n          ELSE 0\n        END hideInHierarchy\n      FROM bis.Subject s\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: subjectsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId, targetPartitionId: row.targetPartitionId, hideInHierarchy: !!row.hideInHierarchy };\n    }\n  }\n\n  private async *queryModels(): AsyncIterableIterator<{ id: Id64String; parentId: Id64String }> {\n    const modelsQuery = `\n      SELECT p.ECInstanceId id, p.Parent.Id parentId\n      FROM bis.InformationPartitionElement p\n      INNER JOIN bis.GeometricModel3d m ON m.ModeledElement.Id = p.ECInstanceId\n      WHERE\n        NOT m.IsPrivate\n        ${this._hierarchyConfig.showEmptyModels ? \"\" : `AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)`}\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: modelsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId };\n    }\n  }\n\n  private async getSubjectInfos() {\n    this._subjectInfos ??= (async () => {\n      const [subjectInfos, targetPartitionSubjects] = await Promise.all([\n        (async () => {\n          const result = new Map<Id64String, SubjectInfo>();\n          for await (const subject of this.querySubjects()) {\n            const subjectInfo: SubjectInfo = {\n              parentSubject: subject.parentId,\n              hideInHierarchy: subject.hideInHierarchy,\n              childSubjects: new Set(),\n              childModels: new Set(),\n            };\n            if (subject.targetPartitionId) {\n              subjectInfo.childModels.add(subject.targetPartitionId);\n            }\n            result.set(subject.id, subjectInfo);\n          }\n          return result;\n        })(),\n        (async () => {\n          const result = new Map<Id64String, Set<Id64String>>();\n          for await (const model of this.queryModels()) {\n            pushToMap(result, model.id, model.parentId);\n          }\n          return result;\n        })(),\n      ]);\n\n      for (const [subjectId, { parentSubject: parentSubjectId }] of subjectInfos.entries()) {\n        if (parentSubjectId) {\n          const parentSubjectInfo = subjectInfos.get(parentSubjectId);\n          assert(!!parentSubjectInfo);\n          parentSubjectInfo.childSubjects.add(subjectId);\n        }\n      }\n\n      for (const [partitionId, subjectIds] of targetPartitionSubjects) {\n        subjectIds.forEach((subjectId) => {\n          const subjectInfo = subjectInfos.get(subjectId);\n          assert(!!subjectInfo);\n          subjectInfo.childModels.add(partitionId);\n        });\n      }\n\n      return subjectInfos;\n    })();\n    return this._subjectInfos;\n  }\n\n  /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */\n  public async getParentSubjectIds(): Promise<Id64String[]> {\n    this._parentSubjectIds ??= (async () => {\n      const subjectInfos = await this.getSubjectInfos();\n      const parentSubjectIds = new Set<Id64String>();\n      subjectInfos.forEach((subjectInfo, subjectId) => {\n        if (subjectInfo.childModels.size > 0) {\n          parentSubjectIds.add(subjectId);\n          let currParentId = subjectInfo.parentSubject;\n          while (currParentId) {\n            parentSubjectIds.add(currParentId);\n            currParentId = subjectInfos.get(currParentId)?.parentSubject;\n          }\n        }\n      });\n      return [...parentSubjectIds];\n    })();\n    return this._parentSubjectIds;\n  }\n\n  /**\n   * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into\n   * account `hideInHierarchy` flag.\n   */\n  public async getChildSubjectIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const childSubjectIds = new Array<Id64String>();\n    const subjectInfos = await this.getSubjectInfos();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (!childSubjectInfo.hideInHierarchy) {\n          childSubjectIds.push(childSubjectId);\n          return \"break\";\n        }\n        return \"continue\";\n      });\n    });\n    return childSubjectIds;\n  }\n\n  /** Returns ECInstanceIDs of all Models under specific parent Subjects, including their child Subjects, etc. */\n  public async getSubjectModelIds(subjectIds: Id64Array): Promise<Id64Array> {\n    const subjectInfos = await this.getSubjectInfos();\n    const subjectStack = [...subjectIds];\n    const result = new Array<Id64String>();\n    while (true) {\n      const subjectId = subjectStack.pop();\n      if (subjectId === undefined) {\n        break;\n      }\n      const subjectInfo = subjectInfos.get(subjectId);\n      if (!subjectInfo) {\n        continue;\n      }\n      result.push(...subjectInfo.childModels);\n      subjectStack.push(...subjectInfo.childSubjects);\n    }\n    return result;\n  }\n\n  /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */\n  public async getChildSubjectModelIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const subjectInfos = await this.getSubjectInfos();\n\n    const hiddenSubjectIds = new Array<Id64String>();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (childSubjectInfo.hideInHierarchy) {\n          hiddenSubjectIds.push(childSubjectId);\n          return \"continue\";\n        }\n        return \"break\";\n      });\n    });\n\n    const modelIds = new Array<Id64String>();\n    [...parentSubjectIds, ...hiddenSubjectIds].forEach((subjectId) => {\n      const subjectInfo = subjectInfos.get(subjectId);\n      subjectInfo && modelIds.push(...subjectInfo.childModels);\n    });\n    return modelIds;\n  }\n\n  public async createSubjectInstanceKeysPath(targetSubjectId: Id64String): Promise<HierarchyNodeIdentifiersPath> {\n    let entry = this._subjectKeyPaths.get(targetSubjectId);\n    if (!entry) {\n      entry = (async () => {\n        const subjectInfos = await this.getSubjectInfos();\n        const result = new Array<InstanceKey>();\n        let currParentId: Id64String | undefined = targetSubjectId;\n        while (currParentId) {\n          const parentInfo = subjectInfos.get(currParentId);\n          if (!parentInfo?.hideInHierarchy) {\n            result.push({ className: \"BisCore.Subject\", id: currParentId });\n          }\n          currParentId = parentInfo?.parentSubject;\n        }\n        return result.reverse();\n      })();\n      this._subjectKeyPaths.set(targetSubjectId, entry);\n    }\n    return entry;\n  }\n\n  private async *queryModelElementCounts() {\n    const query = /* sql */ `\n      SELECT Model.Id modelId, COUNT(*) elementCount\n      FROM ${this._hierarchyConfig.elementClassSpecification}\n      GROUP BY Model.Id\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, elementCount: row.elementCount };\n    }\n  }\n\n  private async *queryModelCategories() {\n    const query = /* sql */ `\n      SELECT Model.Id modelId, Category.Id categoryId\n      FROM ${this._hierarchyConfig.elementClassSpecification}\n      WHERE Parent.Id IS NULL\n      GROUP BY modelId, categoryId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, categoryId: row.categoryId };\n    }\n  }\n\n  private async getModelInfos() {\n    this._modelInfos ??= (async () => {\n      const modelInfos = new Map<Id64String, { categories: Id64Set; elementCount: number }>();\n      await Promise.all([\n        (async () => {\n          for await (const { modelId, categoryId } of this.queryModelCategories()) {\n            const entry = modelInfos.get(modelId);\n            if (entry) {\n              entry.categories.add(categoryId);\n            } else {\n              modelInfos.set(modelId, { categories: new Set([categoryId]), elementCount: 0 });\n            }\n          }\n        })(),\n        (async () => {\n          for await (const { modelId, elementCount } of this.queryModelElementCounts()) {\n            const entry = modelInfos.get(modelId);\n            if (entry) {\n              entry.elementCount = elementCount;\n            } else {\n              modelInfos.set(modelId, { categories: new Set(), elementCount });\n            }\n          }\n        })(),\n      ]);\n      return modelInfos;\n    })();\n    return this._modelInfos;\n  }\n\n  public async getModelCategories(modelId: Id64String): Promise<Id64Array> {\n    const modelInfos = await this.getModelInfos();\n    const categories = modelInfos.get(modelId)?.categories;\n    return categories ? [...categories] : [];\n  }\n\n  public async getModelElementCount(modelId: Id64String): Promise<number> {\n    const modelInfos = await this.getModelInfos();\n    return modelInfos.get(modelId)?.elementCount ?? 0;\n  }\n\n  public async createModelInstanceKeyPaths(modelId: Id64String): Promise<HierarchyNodeIdentifiersPath[]> {\n    let entry = this._modelKeyPaths.get(modelId);\n    if (!entry) {\n      entry = (async () => {\n        const result = new Array<HierarchyNodeIdentifiersPath>();\n        const subjectInfos = (await this.getSubjectInfos()).entries();\n        for (const [modelSubjectId, subjectInfo] of subjectInfos) {\n          if (subjectInfo.childModels.has(modelId)) {\n            const subjectPath = await this.createSubjectInstanceKeysPath(modelSubjectId);\n            result.push([...subjectPath, { className: \"BisCore.GeometricModel3d\", id: modelId }]);\n          }\n        }\n        return result;\n      })();\n\n      this._modelKeyPaths.set(modelId, entry);\n    }\n    return entry;\n  }\n\n  private async queryCategoryElementCounts(\n    input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n  ): Promise<Array<{ modelId: number; categoryId: number; elementsCount: number }>> {\n    const reader = this._queryExecutor.createQueryReader(\n      {\n        ctes: [\n          /* sql */ `\n            CategoryElements(id, modelId, categoryId) AS (\n              SELECT ECInstanceId, Model.Id, Category.Id\n              FROM ${this._hierarchyConfig.elementClassSpecification}\n              WHERE\n                Parent.Id IS NULL\n                AND (\n                  ${input.map(({ modelId, categoryId }) => `Model.Id = ${modelId} AND Category.Id = ${categoryId}`).join(\" OR \")}\n                )\n\n              UNION ALL\n\n              SELECT c.ECInstanceId, p.modelId, p.categoryId\n              FROM ${this._hierarchyConfig.elementClassSpecification} c\n              JOIN CategoryElements p ON c.Parent.Id = p.id\n            )\n          `,\n        ],\n        ecsql: `\n          SELECT modelId, categoryId, COUNT(id) elementsCount\n          FROM CategoryElements\n          GROUP BY modelId, categoryId\n        `,\n      },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" },\n    );\n\n    const result = new Array<{ modelId: number; categoryId: number; elementsCount: number }>();\n    for await (const row of reader) {\n      result.push({ modelId: row.modelId, categoryId: row.categoryId, elementsCount: row.elementsCount });\n    }\n    return result;\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  public async createCategoryInstanceKeyPaths(categoryId: Id64String): Promise<HierarchyNodeIdentifiersPath[]> {\n    let entry = this._categoryKeyPaths.get(categoryId);\n    if (!entry) {\n      entry = (async () => {\n        const result = new Set<Id64String>();\n        const modelInfos = await this.getModelInfos();\n        modelInfos?.forEach((modelInfo, modelId) => {\n          if (modelInfo.categories.has(categoryId)) {\n            result.add(modelId);\n          }\n        });\n\n        const categoryPaths = new Array<HierarchyNodeIdentifiersPath>();\n        for (const categoryModelId of [...result]) {\n          const modelPaths = await this.createModelInstanceKeyPaths(categoryModelId);\n          for (const modelPath of modelPaths) {\n            categoryPaths.push([...modelPath, { className: \"BisCore.SpatialCategory\", id: categoryId }]);\n          }\n        }\n        return categoryPaths;\n      })();\n      this._categoryKeyPaths.set(categoryId, entry);\n    }\n    return entry;\n  }\n}\n\nfunction forEachChildSubject(\n  subjectInfos: Map<Id64String, SubjectInfo>,\n  parentSubject: Id64String | SubjectInfo,\n  cb: (childSubjectId: Id64String, childSubjectInfo: SubjectInfo) => \"break\" | \"continue\",\n) {\n  const parentSubjectInfo = typeof parentSubject === \"string\" ? subjectInfos.get(parentSubject) : parentSubject;\n  parentSubjectInfo &&\n    parentSubjectInfo.childSubjects.forEach((childSubjectId) => {\n      const childSubjectInfo = subjectInfos.get(childSubjectId)!;\n      if (cb(childSubjectId, childSubjectInfo) === \"break\") {\n        return;\n      }\n      forEachChildSubject(subjectInfos, childSubjectInfo, cb);\n    });\n}\n\nclass ModelCategoryElementsCountCache {\n  private _cache = new Map<string, Subject<number>>();\n  private _requestsStream = new Subject<{ modelId: Id64String; categoryId: Id64String }>();\n  private _subscription: Subscription;\n\n  public constructor(\n    private _loader: (\n      input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n    ) => Promise<Array<{ modelId: number; categoryId: number; elementsCount: number }>>,\n  ) {\n    this._subscription = this._requestsStream\n      .pipe(\n        bufferTime(20),\n        filter((requests) => requests.length > 0),\n        mergeMap(async (requests) => this._loader(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  public [Symbol.dispose]() {\n    this._subscription.unsubscribe();\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    const cacheKey = `${modelId}${categoryId}`;\n    let result = this._cache.get(cacheKey);\n    if (result !== undefined) {\n      return firstValueFrom(result);\n    }\n\n    result = new ReplaySubject(1);\n    this._cache.set(cacheKey, result);\n    this._requestsStream.next({ modelId, categoryId });\n    return firstValueFrom(result);\n  }\n}\n"]}
+{"version":3,"file":"ModelsTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../src/components/trees/models-tree/internal/ModelsTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAGhG,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACtG,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAqBlD,gBAAgB;AAChB,MAAM,OAAO,kBAAkB;IAS7B,YACU,cAA0C,EAC1C,gBAAkD;QADlD,mBAAc,GAAd,cAAc,CAA4B;QAC1C,qBAAgB,GAAhB,gBAAgB,CAAkC;QAE1D,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3H,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC,gBAAgB,GAAG,IAAI,GAAG,EAAE,CAAC;QAClC,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;IACrC,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,aAAa;QAC1B,MAAM,aAAa,GAAG;;;;;;;;;wCASc,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;;;;;;;KAUlF,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAChJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,iBAAiB,EAAE,GAAG,CAAC,iBAAiB,EAAE,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;SAChI;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,WAAW;QACxB,MAAM,WAAW,GAAG;;;;;;UAMd,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,6BAA6B,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,mCAAmC;KACjK,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAC9I,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;SAC9C;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,aAAa,KAAK,CAAC,KAAK,IAAI,EAAE;YACjC,MAAM,CAAC,YAAY,EAAE,uBAAuB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChE,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;oBAClD,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE;wBAChD,MAAM,WAAW,GAAgB;4BAC/B,aAAa,EAAE,OAAO,CAAC,QAAQ;4BAC/B,eAAe,EAAE,OAAO,CAAC,eAAe;4BACxC,aAAa,EAAE,IAAI,GAAG,EAAE;4BACxB,WAAW,EAAE,IAAI,GAAG,EAAE;yBACvB,CAAC;wBACF,IAAI,OAAO,CAAC,iBAAiB,EAAE;4BAC7B,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;yBACxD;wBACD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;qBACrC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA+B,CAAC;oBACtD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE;wBAC5C,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;qBAC7C;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YAEH,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE;gBACpF,IAAI,eAAe,EAAE;oBACnB,MAAM,iBAAiB,GAAG,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;oBAC5D,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC;oBAC5B,iBAAiB,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;iBAChD;aACF;YAED,KAAK,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,uBAAuB,EAAE;gBAC/D,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;oBAC/B,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChD,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;oBACtB,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC3C,CAAC,CAAC,CAAC;aACJ;YAED,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,kHAAkH;IAC3G,KAAK,CAAC,mBAAmB;QAC9B,IAAI,CAAC,iBAAiB,KAAK,CAAC,KAAK,IAAI,EAAE;YACrC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAClD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;YAC/C,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE;gBAC9C,IAAI,WAAW,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE;oBACpC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChC,IAAI,YAAY,GAAG,WAAW,CAAC,aAAa,CAAC;oBAC7C,OAAO,YAAY,EAAE;wBACnB,gBAAgB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;wBACnC,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,aAAa,CAAC;qBAC9D;iBACF;YACH,CAAC,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,gBAAgB,CAAC,CAAC;QAC/B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,kBAAkB,CAAC,gBAA8B;QAC5D,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;QAChD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE;oBACrC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACrC,OAAO,OAAO,CAAC;iBAChB;gBACD,OAAO,UAAU,CAAC;YACpB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,+GAA+G;IACxG,KAAK,CAAC,kBAAkB,CAAC,UAAqB;QACnD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,MAAM,YAAY,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,OAAO,IAAI,EAAE;YACX,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,EAAE,CAAC;YACrC,IAAI,SAAS,KAAK,SAAS,EAAE;gBAC3B,MAAM;aACP;YACD,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,WAAW,EAAE;gBAChB,SAAS;aACV;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;YACxC,YAAY,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,aAAa,CAAC,CAAC;SACjD;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,6GAA6G;IACtG,KAAK,CAAC,uBAAuB,CAAC,gBAA8B;QACjE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAElD,MAAM,gBAAgB,GAAG,IAAI,KAAK,EAAc,CAAC;QACjD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,gBAAgB,CAAC,eAAe,EAAE;oBACpC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACtC,OAAO,UAAU,CAAC;iBACnB;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,KAAK,EAAc,CAAC;QACzC,CAAC,GAAG,gBAAgB,EAAE,GAAG,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YAC/D,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,WAAW,IAAI,QAAQ,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEM,KAAK,CAAC,6BAA6B,CAAC,eAA2B;QACpE,IAAI,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QACvD,IAAI,CAAC,KAAK,EAAE;YACV,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;gBAClD,MAAM,MAAM,GAAG,IAAI,KAAK,EAAe,CAAC;gBACxC,IAAI,YAAY,GAA2B,eAAe,CAAC;gBAC3D,OAAO,YAAY,EAAE;oBACnB,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;oBAClD,IAAI,CAAC,UAAU,EAAE,eAAe,EAAE;wBAChC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;qBACjE;oBACD,YAAY,GAAG,UAAU,EAAE,aAAa,CAAC;iBAC1C;gBACD,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;SACnD;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,uBAAuB;QACpC,MAAM,KAAK,GAAG,SAAS,CAAC;;aAEf,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;KAEvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACxI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC;SAChE;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QACjC,MAAM,KAAK,GAAG,SAAS,CAAC;;aAEf,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;KAGvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACxI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAC5D;IACH,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,IAAI,CAAC,WAAW,KAAK,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,EAA6D,CAAC;YACxF,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,CAAC,KAAK,IAAI,EAAE;oBACV,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE;wBACvE,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACtC,IAAI,KAAK,EAAE;4BACT,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;yBAClC;6BAAM;4BACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;yBACjF;qBACF;gBACH,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,IAAI,CAAC,uBAAuB,EAAE,EAAE;wBAC5E,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACtC,IAAI,KAAK,EAAE;4BACT,KAAK,CAAC,YAAY,GAAG,YAAY,CAAC;yBACnC;6BAAM;4BACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;yBAClE;qBACF;gBACH,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YACH,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,OAAmB;QACjD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,UAAU,CAAC;QACvD,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAAC,OAAmB;QACnD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,OAAO,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,YAAY,IAAI,CAAC,CAAC;IACpD,CAAC;IAEM,KAAK,CAAC,2BAA2B,CAAC,OAAmB;QAC1D,IAAI,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC7C,IAAI,CAAC,KAAK,EAAE;YACV,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAG,IAAI,KAAK,EAAgC,CAAC;gBACzD,MAAM,YAAY,GAAG,CAAC,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC9D,KAAK,MAAM,CAAC,cAAc,EAAE,WAAW,CAAC,IAAI,YAAY,EAAE;oBACxD,IAAI,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;wBACxC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,6BAA6B,CAAC,cAAc,CAAC,CAAC;wBAC7E,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,WAAW,EAAE,EAAE,SAAS,EAAE,0BAA0B,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;qBACvF;iBACF;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,EAAE,CAAC;YAEL,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;SACzC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACtC,KAA6D;QAE7D,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAClD;YACE,IAAI,EAAE;gBACJ,SAAS,CAAC;;;qBAGC,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;oBAIhD,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,cAAc,OAAO,sBAAsB,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;;;;;;qBAM3G,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;WAGzD;aACF;YACD,KAAK,EAAE;;;;SAIN;SACF,EACD,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CACxD,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,KAAK,EAAkE,CAAC;QAC3F,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE;YAC9B,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,aAAa,EAAE,GAAG,CAAC,aAAa,EAAE,CAAC,CAAC;SACrG;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,KAAK,CAAC,8BAA8B,CAAC,UAAsB;QAChE,IAAI,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACnD,IAAI,CAAC,KAAK,EAAE;YACV,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAG,IAAI,GAAG,EAAc,CAAC;gBACrC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;gBAC9C,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,EAAE,OAAO,EAAE,EAAE;oBACzC,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;wBACxC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;qBACrB;gBACH,CAAC,CAAC,CAAC;gBAEH,MAAM,aAAa,GAAG,IAAI,KAAK,EAAgC,CAAC;gBAChE,KAAK,MAAM,eAAe,IAAI,CAAC,GAAG,MAAM,CAAC,EAAE;oBACzC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,eAAe,CAAC,CAAC;oBAC3E,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE;wBAClC,aAAa,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,EAAE,EAAE,SAAS,EAAE,yBAAyB,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;qBAC9F;iBACF;gBACD,OAAO,aAAa,CAAC;YACvB,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;SAC/C;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED,SAAS,mBAAmB,CAC1B,YAA0C,EAC1C,aAAuC,EACvC,EAAuF;IAEvF,MAAM,iBAAiB,GAAG,OAAO,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;IAC9G,iBAAiB;QACf,iBAAiB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YACzD,MAAM,gBAAgB,GAAG,YAAY,CAAC,GAAG,CAAC,cAAc,CAAE,CAAC;YAC3D,IAAI,EAAE,CAAC,cAAc,EAAE,gBAAgB,CAAC,KAAK,OAAO,EAAE;gBACpD,OAAO;aACR;YACD,mBAAmB,CAAC,YAAY,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,+BAA+B;IAKnC,YACU,OAE2E;QAF3E,YAAO,GAAP,OAAO,CAEoE;QAP7E,WAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;QAC5C,oBAAe,GAAG,IAAI,OAAO,EAAmD,CAAC;QAQvF,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,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EACpD,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,GAAG,UAAU,EAAE,CAAC,CAAC;gBAC3D,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9B,CAAC;SACF,CAAC,CAAC;IACP,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;IACnC,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,MAAM,QAAQ,GAAG,GAAG,OAAO,GAAG,UAAU,EAAE,CAAC;QAC3C,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,MAAM,KAAK,SAAS,EAAE;YACxB,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;SAC/B;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,cAAc,CAAC,MAAM,CAAC,CAAC;IAChC,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 type { Subscription } from \"rxjs\";\nimport { bufferTime, filter, firstValueFrom, mergeAll, mergeMap, ReplaySubject, Subject } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { pushToMap } from \"../../common/Utils.js\";\n\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\nimport type { ModelsTreeDefinition } from \"../ModelsTreeDefinition.js\";\nimport type { Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\n\ninterface SubjectInfo {\n  parentSubject: Id64String | undefined;\n  hideInHierarchy: boolean;\n  childSubjects: Id64Set;\n  childModels: Id64Set;\n}\n\ninterface ModelInfo {\n  categories: Id64Set;\n  elementCount: number;\n}\n\ntype ModelsTreeHierarchyConfiguration = ConstructorParameters<typeof ModelsTreeDefinition>[0][\"hierarchyConfig\"];\n\n/** @internal */\nexport class ModelsTreeIdsCache {\n  private readonly _categoryElementCounts: ModelCategoryElementsCountCache;\n  private _subjectInfos: Promise<Map<Id64String, SubjectInfo>> | undefined;\n  private _parentSubjectIds: Promise<Id64Array> | undefined; // the list should contain a subject id if its node should be shown as having children\n  private _modelInfos: Promise<Map<Id64String, ModelInfo>> | undefined;\n  private _modelKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath[]>>;\n  private _subjectKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath>>;\n  private _categoryKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath[]>>;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    private _hierarchyConfig: ModelsTreeHierarchyConfiguration,\n  ) {\n    this._categoryElementCounts = new ModelCategoryElementsCountCache(async (input) => this.queryCategoryElementCounts(input));\n    this._modelKeyPaths = new Map();\n    this._subjectKeyPaths = new Map();\n    this._categoryKeyPaths = new Map();\n  }\n\n  public [Symbol.dispose]() {\n    this._categoryElementCounts[Symbol.dispose]();\n  }\n\n  private async *querySubjects(): AsyncIterableIterator<{ id: Id64String; parentId?: Id64String; targetPartitionId?: Id64String; hideInHierarchy: boolean }> {\n    const subjectsQuery = `\n      SELECT\n        s.ECInstanceId id,\n        s.Parent.Id parentId,\n        (\n          SELECT m.ECInstanceId\n          FROM bis.GeometricModel3d m\n          WHERE m.ECInstanceId = HexToId(json_extract(s.JsonProperties, '$.Subject.Model.TargetPartition'))\n            AND NOT m.IsPrivate\n            AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)\n        ) targetPartitionId,\n        CASE\n          WHEN (\n            json_extract(s.JsonProperties, '$.Subject.Job.Bridge') IS NOT NULL\n            OR json_extract(s.JsonProperties, '$.Subject.Model.Type') = 'Hierarchy'\n          ) THEN 1\n          ELSE 0\n        END hideInHierarchy\n      FROM bis.Subject s\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: subjectsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId, targetPartitionId: row.targetPartitionId, hideInHierarchy: !!row.hideInHierarchy };\n    }\n  }\n\n  private async *queryModels(): AsyncIterableIterator<{ id: Id64String; parentId: Id64String }> {\n    const modelsQuery = `\n      SELECT p.ECInstanceId id, p.Parent.Id parentId\n      FROM bis.InformationPartitionElement p\n      INNER JOIN bis.GeometricModel3d m ON m.ModeledElement.Id = p.ECInstanceId\n      WHERE\n        NOT m.IsPrivate\n        ${this._hierarchyConfig.showEmptyModels ? \"\" : `AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)`}\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: modelsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId };\n    }\n  }\n\n  private async getSubjectInfos() {\n    this._subjectInfos ??= (async () => {\n      const [subjectInfos, targetPartitionSubjects] = await Promise.all([\n        (async () => {\n          const result = new Map<Id64String, SubjectInfo>();\n          for await (const subject of this.querySubjects()) {\n            const subjectInfo: SubjectInfo = {\n              parentSubject: subject.parentId,\n              hideInHierarchy: subject.hideInHierarchy,\n              childSubjects: new Set(),\n              childModels: new Set(),\n            };\n            if (subject.targetPartitionId) {\n              subjectInfo.childModels.add(subject.targetPartitionId);\n            }\n            result.set(subject.id, subjectInfo);\n          }\n          return result;\n        })(),\n        (async () => {\n          const result = new Map<Id64String, Set<Id64String>>();\n          for await (const model of this.queryModels()) {\n            pushToMap(result, model.id, model.parentId);\n          }\n          return result;\n        })(),\n      ]);\n\n      for (const [subjectId, { parentSubject: parentSubjectId }] of subjectInfos.entries()) {\n        if (parentSubjectId) {\n          const parentSubjectInfo = subjectInfos.get(parentSubjectId);\n          assert(!!parentSubjectInfo);\n          parentSubjectInfo.childSubjects.add(subjectId);\n        }\n      }\n\n      for (const [partitionId, subjectIds] of targetPartitionSubjects) {\n        subjectIds.forEach((subjectId) => {\n          const subjectInfo = subjectInfos.get(subjectId);\n          assert(!!subjectInfo);\n          subjectInfo.childModels.add(partitionId);\n        });\n      }\n\n      return subjectInfos;\n    })();\n    return this._subjectInfos;\n  }\n\n  /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */\n  public async getParentSubjectIds(): Promise<Id64String[]> {\n    this._parentSubjectIds ??= (async () => {\n      const subjectInfos = await this.getSubjectInfos();\n      const parentSubjectIds = new Set<Id64String>();\n      subjectInfos.forEach((subjectInfo, subjectId) => {\n        if (subjectInfo.childModels.size > 0) {\n          parentSubjectIds.add(subjectId);\n          let currParentId = subjectInfo.parentSubject;\n          while (currParentId) {\n            parentSubjectIds.add(currParentId);\n            currParentId = subjectInfos.get(currParentId)?.parentSubject;\n          }\n        }\n      });\n      return [...parentSubjectIds];\n    })();\n    return this._parentSubjectIds;\n  }\n\n  /**\n   * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into\n   * account `hideInHierarchy` flag.\n   */\n  public async getChildSubjectIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const childSubjectIds = new Array<Id64String>();\n    const subjectInfos = await this.getSubjectInfos();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (!childSubjectInfo.hideInHierarchy) {\n          childSubjectIds.push(childSubjectId);\n          return \"break\";\n        }\n        return \"continue\";\n      });\n    });\n    return childSubjectIds;\n  }\n\n  /** Returns ECInstanceIDs of all Models under specific parent Subjects, including their child Subjects, etc. */\n  public async getSubjectModelIds(subjectIds: Id64Array): Promise<Id64Array> {\n    const subjectInfos = await this.getSubjectInfos();\n    const subjectStack = [...subjectIds];\n    const result = new Array<Id64String>();\n    while (true) {\n      const subjectId = subjectStack.pop();\n      if (subjectId === undefined) {\n        break;\n      }\n      const subjectInfo = subjectInfos.get(subjectId);\n      if (!subjectInfo) {\n        continue;\n      }\n      result.push(...subjectInfo.childModels);\n      subjectStack.push(...subjectInfo.childSubjects);\n    }\n    return result;\n  }\n\n  /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */\n  public async getChildSubjectModelIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const subjectInfos = await this.getSubjectInfos();\n\n    const hiddenSubjectIds = new Array<Id64String>();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (childSubjectInfo.hideInHierarchy) {\n          hiddenSubjectIds.push(childSubjectId);\n          return \"continue\";\n        }\n        return \"break\";\n      });\n    });\n\n    const modelIds = new Array<Id64String>();\n    [...parentSubjectIds, ...hiddenSubjectIds].forEach((subjectId) => {\n      const subjectInfo = subjectInfos.get(subjectId);\n      subjectInfo && modelIds.push(...subjectInfo.childModels);\n    });\n    return modelIds;\n  }\n\n  public async createSubjectInstanceKeysPath(targetSubjectId: Id64String): Promise<HierarchyNodeIdentifiersPath> {\n    let entry = this._subjectKeyPaths.get(targetSubjectId);\n    if (!entry) {\n      entry = (async () => {\n        const subjectInfos = await this.getSubjectInfos();\n        const result = new Array<InstanceKey>();\n        let currParentId: Id64String | undefined = targetSubjectId;\n        while (currParentId) {\n          const parentInfo = subjectInfos.get(currParentId);\n          if (!parentInfo?.hideInHierarchy) {\n            result.push({ className: \"BisCore.Subject\", id: currParentId });\n          }\n          currParentId = parentInfo?.parentSubject;\n        }\n        return result.reverse();\n      })();\n      this._subjectKeyPaths.set(targetSubjectId, entry);\n    }\n    return entry;\n  }\n\n  private async *queryModelElementCounts() {\n    const query = /* sql */ `\n      SELECT Model.Id modelId, COUNT(*) elementCount\n      FROM ${this._hierarchyConfig.elementClassSpecification}\n      GROUP BY Model.Id\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, elementCount: row.elementCount };\n    }\n  }\n\n  private async *queryModelCategories() {\n    const query = /* sql */ `\n      SELECT Model.Id modelId, Category.Id categoryId\n      FROM ${this._hierarchyConfig.elementClassSpecification}\n      WHERE Parent.Id IS NULL\n      GROUP BY modelId, categoryId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, categoryId: row.categoryId };\n    }\n  }\n\n  private async getModelInfos() {\n    this._modelInfos ??= (async () => {\n      const modelInfos = new Map<Id64String, { categories: Id64Set; elementCount: number }>();\n      await Promise.all([\n        (async () => {\n          for await (const { modelId, categoryId } of this.queryModelCategories()) {\n            const entry = modelInfos.get(modelId);\n            if (entry) {\n              entry.categories.add(categoryId);\n            } else {\n              modelInfos.set(modelId, { categories: new Set([categoryId]), elementCount: 0 });\n            }\n          }\n        })(),\n        (async () => {\n          for await (const { modelId, elementCount } of this.queryModelElementCounts()) {\n            const entry = modelInfos.get(modelId);\n            if (entry) {\n              entry.elementCount = elementCount;\n            } else {\n              modelInfos.set(modelId, { categories: new Set(), elementCount });\n            }\n          }\n        })(),\n      ]);\n      return modelInfos;\n    })();\n    return this._modelInfos;\n  }\n\n  public async getModelCategories(modelId: Id64String): Promise<Id64Array> {\n    const modelInfos = await this.getModelInfos();\n    const categories = modelInfos.get(modelId)?.categories;\n    return categories ? [...categories] : [];\n  }\n\n  public async getModelElementCount(modelId: Id64String): Promise<number> {\n    const modelInfos = await this.getModelInfos();\n    return modelInfos.get(modelId)?.elementCount ?? 0;\n  }\n\n  public async createModelInstanceKeyPaths(modelId: Id64String): Promise<HierarchyNodeIdentifiersPath[]> {\n    let entry = this._modelKeyPaths.get(modelId);\n    if (!entry) {\n      entry = (async () => {\n        const result = new Array<HierarchyNodeIdentifiersPath>();\n        const subjectInfos = (await this.getSubjectInfos()).entries();\n        for (const [modelSubjectId, subjectInfo] of subjectInfos) {\n          if (subjectInfo.childModels.has(modelId)) {\n            const subjectPath = await this.createSubjectInstanceKeysPath(modelSubjectId);\n            result.push([...subjectPath, { className: \"BisCore.GeometricModel3d\", id: modelId }]);\n          }\n        }\n        return result;\n      })();\n\n      this._modelKeyPaths.set(modelId, entry);\n    }\n    return entry;\n  }\n\n  private async queryCategoryElementCounts(\n    input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n  ): Promise<Array<{ modelId: number; categoryId: number; elementsCount: number }>> {\n    const reader = this._queryExecutor.createQueryReader(\n      {\n        ctes: [\n          /* sql */ `\n            CategoryElements(id, modelId, categoryId) AS (\n              SELECT ECInstanceId, Model.Id, Category.Id\n              FROM ${this._hierarchyConfig.elementClassSpecification}\n              WHERE\n                Parent.Id IS NULL\n                AND (\n                  ${input.map(({ modelId, categoryId }) => `Model.Id = ${modelId} AND Category.Id = ${categoryId}`).join(\" OR \")}\n                )\n\n              UNION ALL\n\n              SELECT c.ECInstanceId, p.modelId, p.categoryId\n              FROM ${this._hierarchyConfig.elementClassSpecification} c\n              JOIN CategoryElements p ON c.Parent.Id = p.id\n            )\n          `,\n        ],\n        ecsql: `\n          SELECT modelId, categoryId, COUNT(id) elementsCount\n          FROM CategoryElements\n          GROUP BY modelId, categoryId\n        `,\n      },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" },\n    );\n\n    const result = new Array<{ modelId: number; categoryId: number; elementsCount: number }>();\n    for await (const row of reader) {\n      result.push({ modelId: row.modelId, categoryId: row.categoryId, elementsCount: row.elementsCount });\n    }\n    return result;\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  public async createCategoryInstanceKeyPaths(categoryId: Id64String): Promise<HierarchyNodeIdentifiersPath[]> {\n    let entry = this._categoryKeyPaths.get(categoryId);\n    if (!entry) {\n      entry = (async () => {\n        const result = new Set<Id64String>();\n        const modelInfos = await this.getModelInfos();\n        modelInfos?.forEach((modelInfo, modelId) => {\n          if (modelInfo.categories.has(categoryId)) {\n            result.add(modelId);\n          }\n        });\n\n        const categoryPaths = new Array<HierarchyNodeIdentifiersPath>();\n        for (const categoryModelId of [...result]) {\n          const modelPaths = await this.createModelInstanceKeyPaths(categoryModelId);\n          for (const modelPath of modelPaths) {\n            categoryPaths.push([...modelPath, { className: \"BisCore.SpatialCategory\", id: categoryId }]);\n          }\n        }\n        return categoryPaths;\n      })();\n      this._categoryKeyPaths.set(categoryId, entry);\n    }\n    return entry;\n  }\n}\n\nfunction forEachChildSubject(\n  subjectInfos: Map<Id64String, SubjectInfo>,\n  parentSubject: Id64String | SubjectInfo,\n  cb: (childSubjectId: Id64String, childSubjectInfo: SubjectInfo) => \"break\" | \"continue\",\n) {\n  const parentSubjectInfo = typeof parentSubject === \"string\" ? subjectInfos.get(parentSubject) : parentSubject;\n  parentSubjectInfo &&\n    parentSubjectInfo.childSubjects.forEach((childSubjectId) => {\n      const childSubjectInfo = subjectInfos.get(childSubjectId)!;\n      if (cb(childSubjectId, childSubjectInfo) === \"break\") {\n        return;\n      }\n      forEachChildSubject(subjectInfos, childSubjectInfo, cb);\n    });\n}\n\nclass ModelCategoryElementsCountCache {\n  private _cache = new Map<string, Subject<number>>();\n  private _requestsStream = new Subject<{ modelId: Id64String; categoryId: Id64String }>();\n  private _subscription: Subscription;\n\n  public constructor(\n    private _loader: (\n      input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n    ) => Promise<Array<{ modelId: number; categoryId: number; elementsCount: number }>>,\n  ) {\n    this._subscription = this._requestsStream\n      .pipe(\n        bufferTime(20),\n        filter((requests) => requests.length > 0),\n        mergeMap(async (requests) => this._loader(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  public [Symbol.dispose]() {\n    this._subscription.unsubscribe();\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    const cacheKey = `${modelId}${categoryId}`;\n    let result = this._cache.get(cacheKey);\n    if (result !== undefined) {\n      return firstValueFrom(result);\n    }\n\n    result = new ReplaySubject(1);\n    this._cache.set(cacheKey, result);\n    this._requestsStream.next({ modelId, categoryId });\n    return firstValueFrom(result);\n  }\n}\n"]}

package/lib/esm/components/trees/models-tree/internal/ModelsTreeVisibilityHandler.d.ts

@@ -1,6 +1,6 @@
 import type { GroupingHierarchyNode, HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
-import type { HierarchyVisibilityHandler, HierarchyVisibilityHandlerOverridableMethod, VisibilityStatus } from "../../common/UseHierarchyVisibility";
-import type { ModelsTreeIdsCache } from "./ModelsTreeIdsCache";
+import type { HierarchyVisibilityHandler, HierarchyVisibilityHandlerOverridableMethod, VisibilityStatus } from "../../common/UseHierarchyVisibility.js";
+import type { ModelsTreeIdsCache } from "./ModelsTreeIdsCache.js";
 import type { Id64Arg, Id64Array, Id64Set, Id64String } from "@itwin/core-bentley";
 import type { Viewport } from "@itwin/core-frontend";
 import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";

package/lib/esm/components/trees/models-tree/internal/ModelsTreeVisibilityHandler.js

@@ -6,15 +6,15 @@ import { concat, concatAll, defer, distinct, EMPTY, filter, firstValueFrom, fork
 import { assert } from "@itwin/core-bentley";
 import { PerModelCategoryVisibility } from "@itwin/core-frontend";
 import { HierarchyNode, HierarchyNodeKey } from "@itwin/presentation-hierarchies";
-import { toggleAllCategories } from "../../common/CategoriesVisibilityUtils";
-import { reduceWhile, toVoidPromise } from "../../common/Rxjs";
-import { createVisibilityStatus } from "../../common/Tooltip";
-import { createVisibilityHandlerResult } from "../../common/UseHierarchyVisibility";
-import { releaseMainThreadOnItemsCount } from "../Utils";
-import { AlwaysAndNeverDrawnElementInfo } from "./AlwaysAndNeverDrawnElementInfo";
-import { createFilteredTree, parseCategoryKey } from "./FilteredTree";
-import { ModelsTreeNode } from "./ModelsTreeNode";
-import { createVisibilityChangeEventListener } from "./VisibilityChangeEventListener";
+import { toggleAllCategories } from "../../common/CategoriesVisibilityUtils.js";
+import { reduceWhile, toVoidPromise } from "../../common/Rxjs.js";
+import { createVisibilityStatus } from "../../common/Tooltip.js";
+import { createVisibilityHandlerResult } from "../../common/UseHierarchyVisibility.js";
+import { releaseMainThreadOnItemsCount } from "../Utils.js";
+import { AlwaysAndNeverDrawnElementInfo } from "./AlwaysAndNeverDrawnElementInfo.js";
+import { createFilteredTree, parseCategoryKey } from "./FilteredTree.js";
+import { ModelsTreeNode } from "./ModelsTreeNode.js";
+import { createVisibilityChangeEventListener } from "./VisibilityChangeEventListener.js";
 /**
  * Creates an instance if `ModelsTreeVisibilityHandler`.
  * @internal
@@ -34,7 +34,6 @@ class ModelsTreeVisibilityHandlerImpl {
         this._filteredTree = _props.filteredPaths ? createFilteredTree(this._props.imodelAccess, _props.filteredPaths) : undefined;
         this._subscriptions.push(this._elementChangeQueue.pipe(concatAll()).subscribe());
     }
-    // istanbul ignore next
     get onVisibilityChange() {
         return this._eventListener.onVisibilityChange;
     }
@@ -261,7 +260,6 @@ class ModelsTreeVisibilityHandlerImpl {
         if (HierarchyNode.isClassGroupingNode(node)) {
             return this.changeElementGroupingNodeState(node, on);
         }
-        // istanbul ignore if
         if (!HierarchyNode.isInstancesNode(node)) {
             return EMPTY;
         }
@@ -272,7 +270,6 @@ class ModelsTreeVisibilityHandlerImpl {
             return this.changeModelState({ ids: node.key.instanceKeys[0].id, on });
         }
         const modelId = ModelsTreeNode.getModelId(node);
-        // istanbul ignore if
         if (!modelId) {
             return EMPTY;
         }
@@ -284,9 +281,7 @@ class ModelsTreeVisibilityHandlerImpl {
             });
         }
         const categoryId = ModelsTreeNode.getCategoryId(node);
-        // istanbul ignore if
         if (!categoryId) {
-            // istanbul ignore next
             return EMPTY;
         }
         return this.changeElementsState({
@@ -326,7 +321,6 @@ class ModelsTreeVisibilityHandlerImpl {
     }
     changeSubjectNodeState(ids, on) {
         const result = defer(() => {
-            // istanbul ignore if
             if (!this._props.viewport.view.isSpatialView()) {
                 return EMPTY;
             }
@@ -337,7 +331,6 @@ class ModelsTreeVisibilityHandlerImpl {
     changeModelState(props) {
         const result = defer(() => {
             const viewport = this._props.viewport;
-            // istanbul ignore if
             if (!viewport.view.isSpatialView()) {
                 return EMPTY;
             }
@@ -645,7 +638,6 @@ export async function invertAllModels(models, viewport) {
  * @public
  */
 export async function toggleModels(models, enable, viewport) {
-    // istanbul ignore if
     if (!models) {
         return;
     }