@itwin/frontend-tiles 4.4.0-dev.20 → 4.4.0-dev.24

package/lib/esm/BatchedSpatialTileTreeRefs.js.map

@@ -1 +1 @@
-{"version":3,"file":"BatchedSpatialTileTreeRefs.js","sourceRoot":"","sources":["../../src/BatchedSpatialTileTreeRefs.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAErD,OAAO,EACiB,+BAA+B,EACrD,kBAAkB,EAAiB,iBAAiB,GACrD,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,gCAAgC,EAAE,+BAA+B,EAAE,MAAM,4BAA4B,CAAC;AAC/G,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AACpE,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAElD,sDAAsD;AACtD,MAAM,gCAAgC;IAUpC,YAAmB,IAAwB,EAAE,IAAsB;QALlD,kBAAa,GAAuC,EAAE,CAAC;QAMtE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,OAAO,GAAG,IAAI,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAE5D,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC;QAChD,IAAI,CAAC,cAAc,GAAG,MAAM,EAAE,gBAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;QAEpE,IAAI,CAAC,aAAa,GAAG,+BAA+B,CAAC,IAAI,EAAE,IAAI,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAEhG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAE7B,MAAM,CAAC,SAAS,KAAK,IAAI,CAAC,WAAW,CAAC,CAAC;IACzC,CAAC;IAEO,IAAI,CAAC,IAAwB,EAAE,MAAwB;QAC7D,MAAM,SAAS,GAAG,uBAAuB,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;QACzF,IAAI,CAAC,WAAW,GAAG,IAAI,+BAA+B,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAEhF,IAAI,CAAC,0BAA0B,CAAC,SAAS,CAAC,CAAC;QAE3C,MAAM,eAAe,GAAG,CAAC,SAA4C,EAAE,EAAE;YACvE,IAAI,CAAC,SAAS,EAAE,gBAAgB;gBAC9B,SAAS,GAAG,SAAS,CAAC;YAExB,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;YAC1C,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAEhC,IAAI,SAAS,KAAK,aAAa;gBAC7B,IAAI,CAAC,SAAS,IAAI,CAAC,aAAa,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC;oBAClE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC9B,CAAC,CAAC;QAEF,IAAI,2BAA2B,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC;QACzI,IAAI,CAAC,KAAK,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,EAAE;YACxD,2BAA2B,EAAE,CAAC;YAC9B,eAAe,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;YACzC,2BAA2B,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC;QACvI,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;QACvB,MAAM,IAAI,CAAC,WAAW,CAAC;QAEvB,KAAK,MAAM,WAAW,IAAI,IAAI,CAAC,aAAa;YAC1C,MAAM,WAAW,CAAC;QAEpB,KAAK,MAAM,WAAW,IAAI,IAAI,CAAC,aAAa;YAC1C,MAAM,WAAW,CAAC;IACtB,CAAC;IAEO,0BAA0B,CAAC,SAAwB;QACzD,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC;QAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC;QACnC,IAAI,CAAC,MAAM;YACT,OAAO;QAET,MAAM,mBAAmB,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC;QACpG,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,cAAc,EAAE;YAC5C,MAAM,OAAO,GAAG,QAAQ,CAAC,iBAAiB,CAAC;YAC3C,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE;gBAC5B,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,gCAAgC,CAAC,SAAS,EAAE;oBACtE,QAAQ;oBACR,MAAM;oBACN,mBAAmB;iBACpB,CAAC,CAAC,CAAC;aACL;SACF;IACH,CAAC;IAEM,MAAM;QACX,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;QAC5B,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,uBAAuB;YAC9B,IAAI,CAAC,uBAAuB,EAAE,CAAC;IACnC,CAAC;IAEM,gBAAgB,CAAC,IAA0B;QAChD,IAAI,CAAC,uBAAuB,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,CAAC;QACzE,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC;IAEM,kBAAkB;QACvB,IAAI,CAAC,uBAAuB,GAAG,SAAS,CAAC;QACzC,IAAI,CAAC,aAAa,CAAC,kBAAkB,EAAE,CAAC;IAC1C,CAAC;IAEM,cAAc;QACnB,0CAA0C;IAC5C,CAAC;CACF;AAED,iGAAiG;AACjG,4DAA4D;AAC5D,8GAA8G;AAC9G,MAAM,sBAAuB,SAAQ,iBAAiB;IAGpD,YAAmB,MAAwB;QACzC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,UAAU,GAAG;YAChB,MAAM;YACN,QAAQ,EAAE,SAAS;YACnB,UAAU,EAAE,kBAAkB,CAAC,SAAS;YACxC,IAAI,EAAE,GAAG,EAAE,CAAC,SAAS;YACrB,OAAO,EAAE,GAAG,EAAE,GAAG,CAAC;YAClB,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC;SACjD,CAAC;IACJ,CAAC;IAED,IAAoB,SAAS;QAC3B,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,gEAAgE;IAChE,IAAoB,kBAAkB;QACpC,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED,+HAA+H;AAC/H,gCAAgC;AAChC,MAAM,8BAA8B;IAOlC,YAAmB,IAAsB,EAAE,OAA2C;QACpF,IAAI,CAAC,SAAS,GAAG,IAAI,sBAAsB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzD,OAAO,CAAC,IAAI,CAAC,CAAC,IAA+B,EAAE,EAAE;YAC/C,IAAI,IAAI,EAAE;gBACR,IAAI,CAAC,WAAW,CAAC,IAAI,gCAAgC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;aACpE;iBAAM;gBACL,IAAI,CAAC,WAAW,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC,CAAC;aACzD;QACH,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YACZ,IAAI,CAAC,WAAW,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,WAAW,CAAC,IAA+B;QACjD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,IAAI,CAAC,WAAW,EAAE;YACpB,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC9C,IAAI,CAAC,WAAW,CAAC,4BAA4B,EAAE,CAAC;YAChD,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;SAC9B;IACH,CAAC;IAEM,MAAM;QACX,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;IACvB,CAAC;IAEM,gBAAgB,CAAC,IAA0B;QAChD,IAAI,IAAI,CAAC,KAAK;YACZ,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;;YAElC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC5B,CAAC;IAEM,kBAAkB;QACvB,IAAI,IAAI,CAAC,KAAK;YACZ,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;;YAEhC,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;IACjC,CAAC;IAEM,cAAc,KAAW,CAAC;IAE1B,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;QACvB,IAAI,IAAI,CAAC,KAAK,EAAE;YACd,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,KAAK;gBAC1B,MAAM,GAAG,CAAC;SACb;aAAM;YACL,MAAM,IAAI,CAAC,SAAS,CAAC;SACtB;IACH,CAAC;CACF;AAED,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAoF,CAAC;AAExH,KAAK,UAAU,gBAAgB,CAAC,MAAwB,EAAE,cAA4C;IACpG,IAAI;QACF,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,SAAS,KAAK,OAAO;YACvB,OAAO,IAAI,CAAC;QAEd,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAC7C,GAAG,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC5B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,OAAO,kBAAkB,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;KACjD;IAAC,OAAO,GAAG,EAAE;QACZ,MAAM,CAAC,YAAY,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC;KACb;AACH,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,sCAAsC,CAAC,IAAsB,EAAE,cAA4C;IACzH,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAC3B,IAAI,KAAK,GAAG,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC5C,IAAI,SAAS,KAAK,KAAK,EAAE;QACvB,MAAM,OAAO,GAAG,KAAK,GAAG,gBAAgB,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QACjE,mBAAmB,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAEjE,OAAO,CAAC,IAAI,CAAC,CAAC,IAA+B,EAAE,EAAE;YAC/C,IAAI,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC;gBACjC,mBAAmB,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YACZ,IAAI,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC;gBACjC,mBAAmB,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;KACJ;IAED,IAAI,IAAI,KAAK,KAAK,EAAE;QAClB,sFAAsF;QACtF,OAAO,+BAA+B,CAAC,IAAI,CAAC,CAAC;KAC9C;IAED,IAAI,KAAK,YAAY,OAAO;QAC1B,OAAO,IAAI,8BAA8B,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAEzD,OAAO,IAAI,gCAAgC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;AAC3D,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { assert, Logger } from \"@itwin/core-bentley\";\r\nimport { RenderSchedule } from \"@itwin/core-common\";\r\nimport {\r\n  AttachToViewportArgs, createSpatialTileTreeReferences, IModelConnection, SpatialTileTreeReferences, SpatialViewState,\r\n  TileTreeLoadStatus, TileTreeOwner, TileTreeReference,\r\n} from \"@itwin/core-frontend\";\r\nimport { AnimatedBatchedTileTreeReference, PrimaryBatchedTileTreeReference } from \"./BatchedTileTreeReference\";\r\nimport { getBatchedTileTreeOwner } from \"./BatchedTileTreeSupplier\";\r\nimport { BatchedModels } from \"./BatchedModels\";\r\nimport { ComputeSpatialTilesetBaseUrl } from \"./FrontendTiles\";\r\nimport { BatchedTilesetSpec } from \"./BatchedTilesetReader\";\r\nimport { loggerCategory } from \"./LoggerCategory\";\r\n\r\n// Obtains tiles pre-published by mesh export service.\r\nclass BatchedSpatialTileTreeReferences implements SpatialTileTreeReferences {\r\n  private readonly _view: SpatialViewState;\r\n  private readonly _models: BatchedModels;\r\n  private _currentScript?: RenderSchedule.Script;\r\n  private _primaryRef!: PrimaryBatchedTileTreeReference;\r\n  private readonly _animatedRefs: AnimatedBatchedTileTreeReference[] = [];\r\n  private _onModelSelectorChanged?: () => void;\r\n  /** Provides tile trees for models that are not included in the batched tile set. */\r\n  private readonly _excludedRefs: SpatialTileTreeReferences;\r\n\r\n  public constructor(spec: BatchedTilesetSpec, view: SpatialViewState) {\r\n    this._view = view;\r\n    this._models = new BatchedModels(view, spec.includedModels);\r\n\r\n    const script = view.displayStyle.scheduleScript;\r\n    this._currentScript = script?.requiresBatching ? script : undefined;\r\n\r\n    this._excludedRefs = createSpatialTileTreeReferences(view, new Set(spec.includedModels.keys()));\r\n\r\n    this.load(spec, view.iModel);\r\n\r\n    assert(undefined !== this._primaryRef);\r\n  }\r\n\r\n  private load(spec: BatchedTilesetSpec, iModel: IModelConnection): void {\r\n    const treeOwner = getBatchedTileTreeOwner(iModel, { spec, script: this._currentScript });\r\n    this._primaryRef = new PrimaryBatchedTileTreeReference(treeOwner, this._models);\r\n\r\n    this.populateAnimatedReferences(treeOwner);\r\n\r\n    const onScriptChanged = (newScript: RenderSchedule.Script | undefined) => {\r\n      if (!newScript?.requiresBatching)\r\n        newScript = undefined;\r\n\r\n      const currentScript = this._currentScript;\r\n      this._currentScript = newScript;\r\n\r\n      if (newScript !== currentScript)\r\n        if (!newScript || !currentScript || !newScript.equals(currentScript))\r\n          this.load(spec, iModel);\r\n    };\r\n\r\n    let removeScriptChangedListener = this._view.displayStyle.onScheduleScriptChanged.addListener((newScript) => onScriptChanged(newScript));\r\n    this._view.onDisplayStyleChanged.addListener((newStyle) => {\r\n      removeScriptChangedListener();\r\n      onScriptChanged(newStyle.scheduleScript);\r\n      removeScriptChangedListener = this._view.displayStyle.onScheduleScriptChanged.addListener((newScript) => onScriptChanged(newScript));\r\n    });\r\n  }\r\n\r\n  public *[Symbol.iterator](): Iterator<TileTreeReference> {\r\n    yield this._primaryRef;\r\n\r\n    for (const animatedRef of this._animatedRefs)\r\n      yield animatedRef;\r\n\r\n    for (const excludedRef of this._excludedRefs)\r\n      yield excludedRef;\r\n  }\r\n\r\n  private populateAnimatedReferences(treeOwner: TileTreeOwner): void {\r\n    this._animatedRefs.length = 0;\r\n    const script = this._currentScript;\r\n    if (!script)\r\n      return;\r\n\r\n    const getCurrentTimePoint = () => this._view.displayStyle.settings.timePoint ?? script.duration.low;\r\n    for (const timeline of script.modelTimelines) {\r\n      const nodeIds = timeline.transformBatchIds;\r\n      for (const nodeId of nodeIds) {\r\n        this._animatedRefs.push(new AnimatedBatchedTileTreeReference(treeOwner, {\r\n          timeline,\r\n          nodeId,\r\n          getCurrentTimePoint,\r\n        }));\r\n      }\r\n    }\r\n  }\r\n\r\n  public update(): void {\r\n    this._excludedRefs.update();\r\n    this._models.setViewedModels(this._view.modelSelector.models);\r\n    if (this._onModelSelectorChanged)\r\n      this._onModelSelectorChanged();\r\n  }\r\n\r\n  public attachToViewport(args: AttachToViewportArgs): void {\r\n    this._onModelSelectorChanged = () => args.invalidateSymbologyOverrides();\r\n    this._excludedRefs.attachToViewport(args);\r\n  }\r\n\r\n  public detachFromViewport(): void {\r\n    this._onModelSelectorChanged = undefined;\r\n    this._excludedRefs.detachFromViewport();\r\n  }\r\n\r\n  public setDeactivated(): void {\r\n    // Used for debugging. Unimplemented here.\r\n  }\r\n}\r\n\r\n// A placeholder used by [[ProxySpatialTileTreeReferences]] until asynchronous loading completes.\r\n// It provides a TileTreeOwner that never loads a tile tree.\r\n// This ensures that [ViewState.areAllTileTreesLoaded]($frontend) will not return `true` while we are loading.\r\nclass ProxyTileTreeReference extends TileTreeReference {\r\n  private readonly _treeOwner: TileTreeOwner;\r\n\r\n  public constructor(iModel: IModelConnection) {\r\n    super();\r\n    this._treeOwner = {\r\n      iModel,\r\n      tileTree: undefined,\r\n      loadStatus: TileTreeLoadStatus.NotLoaded,\r\n      load: () => undefined,\r\n      dispose: () => { },\r\n      loadTree: async () => Promise.resolve(undefined),\r\n    };\r\n  }\r\n\r\n  public override get treeOwner() {\r\n    return this._treeOwner;\r\n  }\r\n\r\n  // eslint-disable-next-line @typescript-eslint/naming-convention\r\n  public override get _isLoadingComplete() {\r\n    return false;\r\n  }\r\n}\r\n\r\n// Serves as a placeholder while we asynchronously obtain the base URL for a pre-published tileset (or asynchronously determine\r\n// that no such tileset exists).\r\nclass ProxySpatialTileTreeReferences implements SpatialTileTreeReferences {\r\n  // Once async loading completes, all methods will be forwarded to this implementation.\r\n  private _impl?: SpatialTileTreeReferences;\r\n  private readonly _proxyRef: ProxyTileTreeReference;\r\n  // Retained if attachToViewport is called while we are still loading; and reset if detachFromViewport is called while loading.\r\n  private _attachArgs?: AttachToViewportArgs;\r\n\r\n  public constructor(view: SpatialViewState, getSpec: Promise<BatchedTilesetSpec | null>) {\r\n    this._proxyRef = new ProxyTileTreeReference(view.iModel);\r\n    getSpec.then((spec: BatchedTilesetSpec | null) => {\r\n      if (spec) {\r\n        this.setTreeRefs(new BatchedSpatialTileTreeReferences(spec, view));\r\n      } else {\r\n        this.setTreeRefs(createSpatialTileTreeReferences(view));\r\n      }\r\n    }).catch(() => {\r\n      this.setTreeRefs(createSpatialTileTreeReferences(view));\r\n    });\r\n  }\r\n\r\n  private setTreeRefs(refs: SpatialTileTreeReferences): void {\r\n    this._impl = refs;\r\n    if (this._attachArgs) {\r\n      this._impl.attachToViewport(this._attachArgs);\r\n      this._attachArgs.invalidateSymbologyOverrides();\r\n      this._attachArgs = undefined;\r\n    }\r\n  }\r\n\r\n  public update(): void {\r\n    this._impl?.update();\r\n  }\r\n\r\n  public attachToViewport(args: AttachToViewportArgs): void {\r\n    if (this._impl)\r\n      this._impl.attachToViewport(args);\r\n    else\r\n      this._attachArgs = args;\r\n  }\r\n\r\n  public detachFromViewport(): void {\r\n    if (this._impl)\r\n      this._impl.detachFromViewport();\r\n    else\r\n      this._attachArgs = undefined;\r\n  }\r\n\r\n  public setDeactivated(): void { }\r\n\r\n  public *[Symbol.iterator](): Iterator<TileTreeReference> {\r\n    if (this._impl) {\r\n      for (const ref of this._impl)\r\n        yield ref;\r\n    } else {\r\n      yield this._proxyRef;\r\n    }\r\n  }\r\n}\r\n\r\nconst iModelToTilesetSpec = new Map<IModelConnection, BatchedTilesetSpec | null | Promise<BatchedTilesetSpec | null>>();\r\n\r\nasync function fetchTilesetSpec(iModel: IModelConnection, computeBaseUrl: ComputeSpatialTilesetBaseUrl): Promise<BatchedTilesetSpec | null> {\r\n  try {\r\n    const baseUrl = await computeBaseUrl(iModel);\r\n    if (undefined === baseUrl)\r\n      return null;\r\n\r\n    const url = new URL(\"tileset.json\", baseUrl);\r\n    url.search = baseUrl.search;\r\n    const response = await fetch(url.toString());\r\n    const json = await response.json();\r\n    return BatchedTilesetSpec.create(baseUrl, json);\r\n  } catch (err) {\r\n    Logger.logException(loggerCategory, err);\r\n    return null;\r\n  }\r\n}\r\n\r\n/** @internal */\r\nexport function createBatchedSpatialTileTreeReferences(view: SpatialViewState, computeBaseUrl: ComputeSpatialTilesetBaseUrl): SpatialTileTreeReferences {\r\n  const iModel = view.iModel;\r\n  let entry = iModelToTilesetSpec.get(iModel);\r\n  if (undefined === entry) {\r\n    const promise = entry = fetchTilesetSpec(iModel, computeBaseUrl);\r\n    iModelToTilesetSpec.set(iModel, entry);\r\n    iModel.onClose.addOnce(() => iModelToTilesetSpec.delete(iModel));\r\n\r\n    promise.then((spec: BatchedTilesetSpec | null) => {\r\n      if (iModelToTilesetSpec.has(iModel))\r\n        iModelToTilesetSpec.set(iModel, spec);\r\n    }).catch(() => {\r\n      if (iModelToTilesetSpec.has(iModel))\r\n        iModelToTilesetSpec.set(iModel, null);\r\n    });\r\n  }\r\n\r\n  if (null === entry) {\r\n    // No tileset could be obtained for this iModel - use default tile generation instead.\r\n    return createSpatialTileTreeReferences(view);\r\n  }\r\n\r\n  if (entry instanceof Promise)\r\n    return new ProxySpatialTileTreeReferences(view, entry);\r\n\r\n  return new BatchedSpatialTileTreeReferences(entry, view);\r\n}\r\n\r\n"]}
+{"version":3,"file":"BatchedSpatialTileTreeRefs.js","sourceRoot":"","sources":["../../src/BatchedSpatialTileTreeRefs.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAE7C,OAAO,EACL,eAAe,EACO,+BAA+B,EACrD,kBAAkB,EAAiB,iBAAiB,GACrD,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAG,wBAAwB,EAAiC,MAAM,4BAA4B,CAAC;AACtG,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AACpE,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE1D,sDAAsD;AACtD,MAAM,gCAAgC;IAapC,YAAmB,IAAwB,EAAE,IAAsB;QAP3D,UAAK,GAA+B,EAAE,CAAC;QAQ7C,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,OAAO,GAAG,IAAI,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACpD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAElB,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC;QAChD,IAAI,CAAC,cAAc,GAAG,MAAM,EAAE,gBAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;QAEpE,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,+BAA+B,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;QAE3E,IAAI,CAAC,OAAO,GAAG,IAAI,kBAAkB,CAAC,IAAI,EAAE,IAAI,CAAC,cAAc,EAAE,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAC9F,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEhB,IAAI,CAAC,qBAAqB,EAAE,CAAC;IAC/B,CAAC;IAEO,YAAY;QAClB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;YACxB,OAAO;QAET,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;IAClB,CAAC;IAEO,YAAY;QAClB,OAAO,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE;YAChD,IAAI,EAAE,IAAI,CAAC,KAAK;YAChB,MAAM,EAAE,IAAI,CAAC,cAAc;YAC3B,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;SAC/B,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB;QAC3B,MAAM,eAAe,GAAG,CAAC,SAA4C,EAAE,EAAE;YACvE,IAAI,CAAC,SAAS,EAAE,gBAAgB;gBAC9B,SAAS,GAAG,SAAS,CAAC;YAExB,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;YAC1C,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAEhC,IAAI,SAAS,KAAK,aAAa;gBAC7B,IAAI,CAAC,SAAS,IAAI,CAAC,aAAa,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC;oBAClE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC,CAAC;QAEF,IAAI,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC;QACxH,IAAI,CAAC,KAAK,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,EAAE;YACxD,UAAU,EAAE,CAAC;YACb,eAAe,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;YACzC,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC;QACtH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;QACtB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;QACnC,MAAM,IAAI,GAAiC;YACzC,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,MAAM;YACN,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,mBAAmB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;SACtI,CAAC;QAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACtC,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;YACpC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,wBAAwB,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;YAC7G,IAAI,QAAQ,EAAE;gBACZ,KAAK,MAAM,MAAM,IAAI,QAAQ,CAAC,iBAAiB;oBAC7C,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,wBAAwB,CAAC,IAAI,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;aAClE;SACF;IACH,CAAC;IAEM,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;QACvB,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,KAAK;YAC1B,MAAM,GAAG,CAAC;QAEZ,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,aAAa;YAClC,MAAM,GAAG,CAAC;IACd,CAAC;IAEM,MAAM;QACX,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;QAC5B,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,uBAAuB;YAC9B,IAAI,CAAC,uBAAuB,EAAE,CAAC;IACnC,CAAC;IAEM,gBAAgB,CAAC,IAA0B;QAChD,IAAI,CAAC,uBAAuB,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,CAAC;QACzE,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,gCAAgC,GAAG,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC,CAAC;IACzH,CAAC;IAEM,kBAAkB;QACvB,IAAI,CAAC,uBAAuB,GAAG,SAAS,CAAC;QACzC,IAAI,CAAC,aAAa,CAAC,kBAAkB,EAAE,CAAC;QAExC,IAAI,IAAI,CAAC,gCAAgC,EAAE;YACzC,IAAI,CAAC,gCAAgC,EAAE,CAAC;YACxC,IAAI,CAAC,gCAAgC,GAAG,SAAS,CAAC;SACnD;IACH,CAAC;IAEM,cAAc;QACnB,0CAA0C;IAC5C,CAAC;CACF;AAED,iGAAiG;AACjG,4DAA4D;AAC5D,8GAA8G;AAC9G,MAAM,sBAAuB,SAAQ,iBAAiB;IAGpD,YAAmB,MAAwB;QACzC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,UAAU,GAAG;YAChB,MAAM;YACN,QAAQ,EAAE,SAAS;YACnB,UAAU,EAAE,kBAAkB,CAAC,SAAS;YACxC,IAAI,EAAE,GAAG,EAAE,CAAC,SAAS;YACrB,OAAO,EAAE,GAAG,EAAE,GAAG,CAAC;YAClB,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC;SACjD,CAAC;IACJ,CAAC;IAED,IAAoB,SAAS;QAC3B,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,gEAAgE;IAChE,IAAoB,kBAAkB;QACpC,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED,+HAA+H;AAC/H,gCAAgC;AAChC,MAAM,8BAA8B;IAOlC,YAAmB,IAAsB,EAAE,OAA2C;QACpF,IAAI,CAAC,SAAS,GAAG,IAAI,sBAAsB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzD,OAAO,CAAC,IAAI,CAAC,CAAC,IAA+B,EAAE,EAAE;YAC/C,IAAI,IAAI,EAAE;gBACR,IAAI,CAAC,WAAW,CAAC,IAAI,gCAAgC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;aACpE;iBAAM;gBACL,IAAI,CAAC,WAAW,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC,CAAC;aACzD;QACH,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YACZ,IAAI,CAAC,WAAW,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,WAAW,CAAC,IAA+B;QACjD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,IAAI,CAAC,WAAW,EAAE;YACpB,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC9C,IAAI,CAAC,WAAW,CAAC,4BAA4B,EAAE,CAAC;YAChD,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;SAC9B;IACH,CAAC;IAEM,MAAM;QACX,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;IACvB,CAAC;IAEM,gBAAgB,CAAC,IAA0B;QAChD,IAAI,IAAI,CAAC,KAAK;YACZ,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;;YAElC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC5B,CAAC;IAEM,kBAAkB;QACvB,IAAI,IAAI,CAAC,KAAK;YACZ,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;;YAEhC,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;IACjC,CAAC;IAEM,cAAc,KAAW,CAAC;IAE1B,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;QACvB,IAAI,IAAI,CAAC,KAAK,EAAE;YACd,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,KAAK;gBAC1B,MAAM,GAAG,CAAC;SACb;aAAM;YACL,MAAM,IAAI,CAAC,SAAS,CAAC;SACtB;IACH,CAAC;CACF;AAED,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAoF,CAAC;AAExH,KAAK,UAAU,gBAAgB,CAAC,MAAwB,EAAE,cAA4C;IACpG,IAAI;QACF,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,SAAS,KAAK,OAAO;YACvB,OAAO,IAAI,CAAC;QAEd,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAC7C,GAAG,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC5B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,OAAO,kBAAkB,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;KACjD;IAAC,OAAO,GAAG,EAAE;QACZ,MAAM,CAAC,YAAY,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC;KACb;AACH,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,sCAAsC,CAAC,IAAsB,EAAE,cAA4C;IACzH,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAC3B,IAAI,KAAK,GAAG,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC5C,IAAI,SAAS,KAAK,KAAK,EAAE;QACvB,MAAM,OAAO,GAAG,KAAK,GAAG,gBAAgB,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QACjE,mBAAmB,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAEjE,OAAO,CAAC,IAAI,CAAC,CAAC,IAA+B,EAAE,EAAE;YAC/C,IAAI,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC;gBACjC,mBAAmB,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YACZ,IAAI,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC;gBACjC,mBAAmB,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;KACJ;IAED,IAAI,IAAI,KAAK,KAAK,EAAE;QAClB,sFAAsF;QACtF,OAAO,+BAA+B,CAAC,IAAI,CAAC,CAAC;KAC9C;IAED,IAAI,KAAK,YAAY,OAAO;QAC1B,OAAO,IAAI,8BAA8B,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAEzD,OAAO,IAAI,gCAAgC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;AAC3D,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { Logger } from \"@itwin/core-bentley\";\r\nimport { RenderSchedule } from \"@itwin/core-common\";\r\nimport {\r\n  AnimationNodeId,\r\n  AttachToViewportArgs, createSpatialTileTreeReferences, IModelConnection, SpatialTileTreeReferences, SpatialViewState,\r\n  TileTreeLoadStatus, TileTreeOwner, TileTreeReference,\r\n} from \"@itwin/core-frontend\";\r\nimport {  BatchedTileTreeReference, BatchedTileTreeReferenceArgs  } from \"./BatchedTileTreeReference\";\r\nimport { getBatchedTileTreeOwner } from \"./BatchedTileTreeSupplier\";\r\nimport { BatchedModels } from \"./BatchedModels\";\r\nimport { ComputeSpatialTilesetBaseUrl } from \"./FrontendTiles\";\r\nimport { BatchedTilesetSpec } from \"./BatchedTilesetReader\";\r\nimport { loggerCategory } from \"./LoggerCategory\";\r\nimport { BatchedModelGroups } from \"./BatchedModelGroups\";\r\n\r\n// Obtains tiles pre-published by mesh export service.\r\nclass BatchedSpatialTileTreeReferences implements SpatialTileTreeReferences {\r\n  private readonly _view: SpatialViewState;\r\n  private readonly _models: BatchedModels;\r\n  private readonly _groups: BatchedModelGroups;\r\n  private readonly _spec: BatchedTilesetSpec;\r\n  private _treeOwner: TileTreeOwner;\r\n  private _refs: BatchedTileTreeReference[] = [];\r\n  private _currentScript?: RenderSchedule.Script;\r\n  private _onModelSelectorChanged?: () => void;\r\n  /** Provides tile trees for models that are not included in the batched tile set. */\r\n  private readonly _excludedRefs: SpatialTileTreeReferences;\r\n  private _removeSceneInvalidationListener?: () => void;\r\n\r\n  public constructor(spec: BatchedTilesetSpec, view: SpatialViewState) {\r\n    this._view = view;\r\n    this._models = new BatchedModels(view, spec.models);\r\n    this._spec = spec;\r\n\r\n    const script = view.displayStyle.scheduleScript;\r\n    this._currentScript = script?.requiresBatching ? script : undefined;\r\n\r\n    const includedModels = new Set(spec.models.keys());\r\n    this._excludedRefs = createSpatialTileTreeReferences(view, includedModels);\r\n\r\n    this._groups = new BatchedModelGroups(view, this._currentScript, includedModels, spec.models);\r\n    this._treeOwner = this.getTreeOwner();\r\n    this.loadRefs();\r\n\r\n    this.listenForScriptChange();\r\n  }\r\n\r\n  private ensureLoaded(): void {\r\n    if (!this._groups.update())\r\n      return;\r\n\r\n    this._treeOwner = this.getTreeOwner();\r\n    this.loadRefs();\r\n  }\r\n\r\n  private getTreeOwner(): TileTreeOwner {\r\n    return getBatchedTileTreeOwner(this._view.iModel, {\r\n      spec: this._spec,\r\n      script: this._currentScript,\r\n      modelGroups: this._groups.guid,\r\n    });\r\n  }\r\n\r\n  private listenForScriptChange(): void {\r\n    const onScriptChanged = (newScript: RenderSchedule.Script | undefined) => {\r\n      if (!newScript?.requiresBatching)\r\n        newScript = undefined;\r\n\r\n      const currentScript = this._currentScript;\r\n      this._currentScript = newScript;\r\n\r\n      if (newScript !== currentScript)\r\n        if (!newScript || !currentScript || !newScript.equals(currentScript))\r\n          this._groups.setScript(newScript);\r\n    };\r\n\r\n    let rmListener = this._view.displayStyle.onScheduleScriptChanged.addListener((newScript) => onScriptChanged(newScript));\r\n    this._view.onDisplayStyleChanged.addListener((newStyle) => {\r\n      rmListener();\r\n      onScriptChanged(newStyle.scheduleScript);\r\n      rmListener = this._view.displayStyle.onScheduleScriptChanged.addListener((newScript) => onScriptChanged(newScript));\r\n    });\r\n  }\r\n\r\n  private loadRefs(): void {\r\n    this._refs.length = 0;\r\n    const groups = this._groups.groups;\r\n    const args: BatchedTileTreeReferenceArgs = {\r\n      models: this._models,\r\n      groups,\r\n      treeOwner: this._treeOwner,\r\n      getCurrentTimePoint: () => this._currentScript ? (this._view.displayStyle.settings.timePoint ?? this._currentScript.duration.low) : 0,\r\n    };\r\n\r\n    for (let i = 0; i < groups.length; i++) {\r\n      const timeline = groups[i].timeline;\r\n      this._refs.push(new BatchedTileTreeReference(args, i, timeline ? AnimationNodeId.Untransformed : undefined));\r\n      if (timeline) {\r\n        for (const nodeId of timeline.transformBatchIds)\r\n          this._refs.push(new BatchedTileTreeReference(args, i, nodeId));\r\n      }\r\n    }\r\n  }\r\n\r\n  public *[Symbol.iterator](): Iterator<TileTreeReference> {\r\n    this.ensureLoaded();\r\n    for (const ref of this._refs)\r\n      yield ref;\r\n\r\n    for (const ref of this._excludedRefs)\r\n      yield ref;\r\n  }\r\n\r\n  public update(): void {\r\n    this._excludedRefs.update();\r\n    this._models.setViewedModels(this._view.modelSelector.models);\r\n    if (this._onModelSelectorChanged)\r\n      this._onModelSelectorChanged();\r\n  }\r\n\r\n  public attachToViewport(args: AttachToViewportArgs): void {\r\n    this._onModelSelectorChanged = () => args.invalidateSymbologyOverrides();\r\n    this._excludedRefs.attachToViewport(args);\r\n    this._removeSceneInvalidationListener = args.onSceneInvalidated.addListener(() => this._groups.invalidateTransforms());\r\n  }\r\n\r\n  public detachFromViewport(): void {\r\n    this._onModelSelectorChanged = undefined;\r\n    this._excludedRefs.detachFromViewport();\r\n\r\n    if (this._removeSceneInvalidationListener) {\r\n      this._removeSceneInvalidationListener();\r\n      this._removeSceneInvalidationListener = undefined;\r\n    }\r\n  }\r\n\r\n  public setDeactivated(): void {\r\n    // Used for debugging. Unimplemented here.\r\n  }\r\n}\r\n\r\n// A placeholder used by [[ProxySpatialTileTreeReferences]] until asynchronous loading completes.\r\n// It provides a TileTreeOwner that never loads a tile tree.\r\n// This ensures that [ViewState.areAllTileTreesLoaded]($frontend) will not return `true` while we are loading.\r\nclass ProxyTileTreeReference extends TileTreeReference {\r\n  private readonly _treeOwner: TileTreeOwner;\r\n\r\n  public constructor(iModel: IModelConnection) {\r\n    super();\r\n    this._treeOwner = {\r\n      iModel,\r\n      tileTree: undefined,\r\n      loadStatus: TileTreeLoadStatus.NotLoaded,\r\n      load: () => undefined,\r\n      dispose: () => { },\r\n      loadTree: async () => Promise.resolve(undefined),\r\n    };\r\n  }\r\n\r\n  public override get treeOwner() {\r\n    return this._treeOwner;\r\n  }\r\n\r\n  // eslint-disable-next-line @typescript-eslint/naming-convention\r\n  public override get _isLoadingComplete() {\r\n    return false;\r\n  }\r\n}\r\n\r\n// Serves as a placeholder while we asynchronously obtain the base URL for a pre-published tileset (or asynchronously determine\r\n// that no such tileset exists).\r\nclass ProxySpatialTileTreeReferences implements SpatialTileTreeReferences {\r\n  // Once async loading completes, all methods will be forwarded to this implementation.\r\n  private _impl?: SpatialTileTreeReferences;\r\n  private readonly _proxyRef: ProxyTileTreeReference;\r\n  // Retained if attachToViewport is called while we are still loading; and reset if detachFromViewport is called while loading.\r\n  private _attachArgs?: AttachToViewportArgs;\r\n\r\n  public constructor(view: SpatialViewState, getSpec: Promise<BatchedTilesetSpec | null>) {\r\n    this._proxyRef = new ProxyTileTreeReference(view.iModel);\r\n    getSpec.then((spec: BatchedTilesetSpec | null) => {\r\n      if (spec) {\r\n        this.setTreeRefs(new BatchedSpatialTileTreeReferences(spec, view));\r\n      } else {\r\n        this.setTreeRefs(createSpatialTileTreeReferences(view));\r\n      }\r\n    }).catch(() => {\r\n      this.setTreeRefs(createSpatialTileTreeReferences(view));\r\n    });\r\n  }\r\n\r\n  private setTreeRefs(refs: SpatialTileTreeReferences): void {\r\n    this._impl = refs;\r\n    if (this._attachArgs) {\r\n      this._impl.attachToViewport(this._attachArgs);\r\n      this._attachArgs.invalidateSymbologyOverrides();\r\n      this._attachArgs = undefined;\r\n    }\r\n  }\r\n\r\n  public update(): void {\r\n    this._impl?.update();\r\n  }\r\n\r\n  public attachToViewport(args: AttachToViewportArgs): void {\r\n    if (this._impl)\r\n      this._impl.attachToViewport(args);\r\n    else\r\n      this._attachArgs = args;\r\n  }\r\n\r\n  public detachFromViewport(): void {\r\n    if (this._impl)\r\n      this._impl.detachFromViewport();\r\n    else\r\n      this._attachArgs = undefined;\r\n  }\r\n\r\n  public setDeactivated(): void { }\r\n\r\n  public *[Symbol.iterator](): Iterator<TileTreeReference> {\r\n    if (this._impl) {\r\n      for (const ref of this._impl)\r\n        yield ref;\r\n    } else {\r\n      yield this._proxyRef;\r\n    }\r\n  }\r\n}\r\n\r\nconst iModelToTilesetSpec = new Map<IModelConnection, BatchedTilesetSpec | null | Promise<BatchedTilesetSpec | null>>();\r\n\r\nasync function fetchTilesetSpec(iModel: IModelConnection, computeBaseUrl: ComputeSpatialTilesetBaseUrl): Promise<BatchedTilesetSpec | null> {\r\n  try {\r\n    const baseUrl = await computeBaseUrl(iModel);\r\n    if (undefined === baseUrl)\r\n      return null;\r\n\r\n    const url = new URL(\"tileset.json\", baseUrl);\r\n    url.search = baseUrl.search;\r\n    const response = await fetch(url.toString());\r\n    const json = await response.json();\r\n    return BatchedTilesetSpec.create(baseUrl, json);\r\n  } catch (err) {\r\n    Logger.logException(loggerCategory, err);\r\n    return null;\r\n  }\r\n}\r\n\r\n/** @internal */\r\nexport function createBatchedSpatialTileTreeReferences(view: SpatialViewState, computeBaseUrl: ComputeSpatialTilesetBaseUrl): SpatialTileTreeReferences {\r\n  const iModel = view.iModel;\r\n  let entry = iModelToTilesetSpec.get(iModel);\r\n  if (undefined === entry) {\r\n    const promise = entry = fetchTilesetSpec(iModel, computeBaseUrl);\r\n    iModelToTilesetSpec.set(iModel, entry);\r\n    iModel.onClose.addOnce(() => iModelToTilesetSpec.delete(iModel));\r\n\r\n    promise.then((spec: BatchedTilesetSpec | null) => {\r\n      if (iModelToTilesetSpec.has(iModel))\r\n        iModelToTilesetSpec.set(iModel, spec);\r\n    }).catch(() => {\r\n      if (iModelToTilesetSpec.has(iModel))\r\n        iModelToTilesetSpec.set(iModel, null);\r\n    });\r\n  }\r\n\r\n  if (null === entry) {\r\n    // No tileset could be obtained for this iModel - use default tile generation instead.\r\n    return createSpatialTileTreeReferences(view);\r\n  }\r\n\r\n  if (entry instanceof Promise)\r\n    return new ProxySpatialTileTreeReferences(view, entry);\r\n\r\n  return new BatchedSpatialTileTreeReferences(entry, view);\r\n}\r\n\r\n"]}

package/lib/esm/BatchedTile.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"BatchedTile.d.ts","sourceRoot":"","sources":["../../src/BatchedTile.ts"],"names":[],"mappings":"AAKA,OAAO,EAAU,WAAW,EAAsB,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAY,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EACU,cAAc,EAAgC,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,WAAW,EAC/G,YAAY,EAAE,UAAU,EAAE,WAAW,EAAE,kBAAkB,EAAsB,QAAQ,EAAkB,QAAQ,EAClH,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAGpD,gBAAgB;AAChB,MAAM,WAAW,iBAAkB,SAAQ,UAAU;IACnD,aAAa,EAAE,eAAe,CAAC,IAAI,EAAE,GAAG,SAAS,CAAC;IAClD,uCAAuC;IACvC,eAAe,EAAE,SAAS,GAAG,SAAS,CAAC;CACxC;AAID,gBAAgB;AAChB,qBAAa,WAAY,SAAQ,IAAI;IACnC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAyB;IACzD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAU;IACvC,gFAAgF;IAChF,SAAgB,eAAe,CAAC,EAAE,SAAS,CAAC;IAE5C,IAAW,WAAW,IAAI,eAAe,CAExC;gBAEkB,MAAM,EAAE,iBAAiB,EAAE,IAAI,EAAE,eAAe;IAyBnE,OAAO,KAAK,gBAAgB,GAE3B;IAEe,mBAAmB,CAAC,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,QAAQ,CAAC,GAAG,MAAM;IAK/F,WAAW,CAAC,QAAQ,EAAE,GAAG,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,0BAA0B,EAAE,WAAW,GAAG,SAAS,GAAG,IAAI;cAoC1G,aAAa,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,GAAG,SAAS,KAAK,IAAI,EAAE,MAAM,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,GAAG,IAAI;IAqBvH,IAAoB,OAAO,IAAI,kBAAkB,CAOhD;IAEqB,cAAc,CAAC,WAAW,EAAE,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC;IAOzE,WAAW,CAAC,IAAI,EAAE,WAAW,CAAC,YAAY,EAAE,MAAM,EAAE,YAAY,EAAE,UAAU,CAAC,EAAE,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC;cA+BtH,eAAe,CAAC,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,iBAAiB,GAAG,IAAI;IAsBnF,KAAK,CAAC,SAAS,EAAE,WAAW,GAAG,IAAI;CAY3C"}
+{"version":3,"file":"BatchedTile.d.ts","sourceRoot":"","sources":["../../src/BatchedTile.ts"],"names":[],"mappings":"AAKA,OAAO,EAAU,WAAW,EAAsB,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAY,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EACU,cAAc,EAAgC,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,WAAW,EAC/G,YAAY,EAAE,UAAU,EAAE,WAAW,EAAE,kBAAkB,EAAsB,QAAQ,EAAkB,QAAQ,EAClH,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAGpD,gBAAgB;AAChB,MAAM,WAAW,iBAAkB,SAAQ,UAAU;IACnD,aAAa,EAAE,eAAe,CAAC,IAAI,EAAE,GAAG,SAAS,CAAC;IAClD,uCAAuC;IACvC,eAAe,EAAE,SAAS,GAAG,SAAS,CAAC;CACxC;AAID,gBAAgB;AAChB,qBAAa,WAAY,SAAQ,IAAI;IACnC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAyB;IACzD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAU;IACvC,gFAAgF;IAChF,SAAgB,eAAe,CAAC,EAAE,SAAS,CAAC;IAE5C,IAAW,WAAW,IAAI,eAAe,CAExC;gBAEkB,MAAM,EAAE,iBAAiB,EAAE,IAAI,EAAE,eAAe;IAyBnE,OAAO,KAAK,gBAAgB,GAE3B;IAEe,mBAAmB,CAAC,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,QAAQ,CAAC,GAAG,MAAM;IAK/F,WAAW,CAAC,QAAQ,EAAE,GAAG,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,0BAA0B,EAAE,WAAW,GAAG,SAAS,GAAG,IAAI;cAoC1G,aAAa,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,GAAG,SAAS,KAAK,IAAI,EAAE,MAAM,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,GAAG,IAAI;IAqBvH,IAAoB,OAAO,IAAI,kBAAkB,CAOhD;IAEqB,cAAc,CAAC,WAAW,EAAE,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC;IAOzE,WAAW,CAAC,IAAI,EAAE,WAAW,CAAC,YAAY,EAAE,MAAM,EAAE,YAAY,EAAE,UAAU,CAAC,EAAE,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC;cAmCtH,eAAe,CAAC,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,iBAAiB,GAAG,IAAI;IAsBnF,KAAK,CAAC,SAAS,EAAE,WAAW,GAAG,IAAI;CAY3C"}

package/lib/esm/BatchedTile.js

@@ -106,12 +106,15 @@ export class BatchedTile extends Tile {
         if (!(data instanceof Uint8Array))
             return {};
         try {
+            const modelGroups = this.batchedTree.modelGroups;
             const content = await this.batchedTree.decoder.decode({
                 stream: ByteStream.fromUint8Array(data),
                 options: { tileId: this.contentId },
                 system,
                 isCanceled,
                 isLeaf: this.isLeaf,
+                // Don't waste time attempting to split based on model groupings if all models are in the same group.
+                modelGroups: modelGroups && modelGroups.length > 1 ? modelGroups : undefined,
             });
             if (this.transformToRoot) {
                 if (content.graphic) {

package/lib/esm/BatchedTile.js.map

@@ -1 +1 @@
-{"version":3,"file":"BatchedTile.js","sourceRoot":"","sources":["../../src/BatchedTile.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,MAAM,EAAe,UAAU,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAE9E,OAAO,EAAE,QAAQ,EAAmB,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EACL,aAAa,EAAkB,SAAS,EAAE,iBAAiB,EAAgB,IAAI,EAAE,iBAAiB,EAC3D,kBAAkB,EAAE,kBAAkB,EAAY,cAAc,GACxG,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAElD,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AASvD,IAAI,OAAuC,CAAC;AAE5C,gBAAgB;AAChB,MAAM,OAAO,WAAY,SAAQ,IAAI;IAMnC,IAAW,WAAW;QACpB,OAAO,IAAI,CAAC,IAAuB,CAAC;IACtC,CAAC;IAED,YAAmB,MAAyB,EAAE,IAAqB;QACjE,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAEpB,kFAAkF;QAClF,IAAI,CAAC,YAAY,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,GAAG,oBAAoB,CAAC,eAAe,CAAC,CAAC;QAE9E,IAAI,MAAM,CAAC,aAAa,EAAE,MAAM;YAC9B,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;QAE7C,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YACnB,IAAI,CAAC,UAAU,EAAE,CAAC;YAClB,uBAAuB;YACvB,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;SACvB;QAED,IAAI,CAAC,MAAM,CAAC,eAAe;YACzB,OAAO;QAET,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QAC9C,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3E,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3D,IAAI,IAAI,CAAC,aAAa;YACpB,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;IAC/E,CAAC;IAED,IAAY,gBAAgB;QAC1B,OAAO,IAAI,CAAC,QAAqC,CAAC;IACpD,CAAC;IAEe,mBAAmB,CAAC,SAA6B,EAAE,MAA0B;QAC3F,2FAA2F;QAC3F,OAAO,iBAAiB,CAAC,2BAA2B,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACnG,CAAC;IAEM,WAAW,CAAC,QAA0B,EAAE,IAAkB,EAAE,0BAAmD;QACpH,MAAM,GAAG,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,cAAc,CAAC,cAAc,KAAK,GAAG;YACvC,OAAO;QAET,IAAI,IAAI,CAAC,YAAY,EAAE;YACrB,0EAA0E;YAC1E,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC5B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;SACrB;QAED,0BAA0B,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,0BAA0B,CAAC;QAClF,IAAI,cAAc,CAAC,SAAS,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE;YAC5E,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACpB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACrB,MAAM,kBAAkB,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;YAC/C,IAAI,kBAAkB,CAAC,OAAO,KAAK,kBAAkB;gBACnD,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAE7B,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC;YACvC,IAAI,QAAQ,EAAE;gBACZ,KAAK,MAAM,KAAK,IAAI,QAAQ;oBAC1B,KAAK,CAAC,WAAW,CAAC,QAAQ,EAAE,IAAI,EAAE,0BAA0B,CAAC,CAAC;gBAEhE,OAAO;aACR;SACF;QAED,2EAA2E;QAC3E,IAAI,CAAC,cAAc,CAAC,OAAO,KAAK,GAAG,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO;YACxE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QAE3B,IAAI,0BAA0B;YAC5B,QAAQ,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;IAC7C,CAAC;IAEkB,aAAa,CAAC,OAA+C,EAAE,MAA8B;QAC9G,IAAI,QAAmC,CAAC;QACxC,IAAI,IAAI,CAAC,cAAc,EAAE;YACvB,IAAI;gBACF,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,cAAc,EAAE;oBAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,cAAc,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;oBACxE,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;oBACxD,QAAQ,GAAG,QAAQ,IAAI,EAAE,CAAC;oBAC1B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;iBACtB;aACF;YAAC,OAAO,GAAG,EAAE;gBACZ,MAAM,CAAC,YAAY,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;gBACzC,QAAQ,GAAG,SAAS,CAAC;gBACrB,IAAI,GAAG,YAAY,KAAK;oBACtB,MAAM,CAAC,GAAG,CAAC,CAAC;aACf;SACF;QAED,OAAO,CAAC,QAAQ,CAAC,CAAC;IACpB,CAAC;IAED,IAAoB,OAAO;QACzB,IAAI,CAAC,OAAO,EAAE;YACZ,OAAO,GAAG,IAAI,kBAAkB,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC;YAC/D,SAAS,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;SAC3C;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEe,KAAK,CAAC,cAAc,CAAC,WAA0B;QAC7D,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACrE,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QACpD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC7C,OAAO,QAAQ,CAAC,WAAW,EAAE,CAAC;IAChC,CAAC;IAEe,KAAK,CAAC,WAAW,CAAC,IAA8B,EAAE,MAAoB,EAAE,UAA0B;QAChH,MAAM,CAAC,IAAI,YAAY,UAAU,CAAC,CAAC;QACnC,IAAI,CAAC,CAAC,IAAI,YAAY,UAAU,CAAC;YAC/B,OAAO,EAAG,CAAC;QAEb,IAAI;YACF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;gBACpD,MAAM,EAAE,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC;gBACvC,OAAO,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE;gBACnC,MAAM;gBACN,UAAU;gBACV,MAAM,EAAE,IAAI,CAAC,MAAM;aACpB,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,eAAe,EAAE;gBACxB,IAAI,OAAO,CAAC,OAAO,EAAE;oBACnB,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC;oBACvC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBAC5B,OAAO,CAAC,OAAO,GAAG,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;iBACrE;gBAED,IAAI,OAAO,CAAC,YAAY;oBACtB,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;aACnF;YAED,OAAO,OAAO,CAAC;SAChB;QAAC,MAAM;YACN,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;SACzB;IACH,CAAC;IAEkB,eAAe,CAAC,OAAuB,EAAE,IAAuB;QACjF,IAAI,iBAAiB,CAAC,YAAY,KAAK,IAAI,EAAE;YAC3C,KAAK,CAAC,eAAe,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YACrC,OAAO;SACR;QAED,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACxD,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEhC,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,QAAQ;YACX,OAAO;QAET,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7E,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE;YAC5B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;YAC1B,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAC3B,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;SAClC;IACH,CAAC;IAEM,KAAK,CAAC,SAAsB;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC;QACvC,IAAI,CAAC,QAAQ;YACX,OAAO;QAET,IAAI,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE;YACzC,IAAI,CAAC,eAAe,EAAE,CAAC;SACxB;aAAM;YACL,KAAK,MAAM,KAAK,IAAI,QAAQ;gBAC1B,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;SAC1B;IACH,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { assert, BeTimePoint, ByteStream, Logger } from \"@itwin/core-bentley\";\r\nimport { Transform } from \"@itwin/core-geometry\";\r\nimport { ColorDef, Tileset3dSchema } from \"@itwin/core-common\";\r\nimport {\r\n  GraphicBranch, GraphicBuilder, IModelApp, RealityTileLoader, RenderSystem, Tile, TileBoundingBoxes, TileContent,\r\n  TileDrawArgs, TileParams, TileRequest, TileRequestChannel, TileTreeLoadStatus, TileUser, TileVisibility, Viewport,\r\n} from \"@itwin/core-frontend\";\r\nimport { loggerCategory } from \"./LoggerCategory\";\r\nimport { BatchedTileTree } from \"./BatchedTileTree\";\r\nimport { frontendTilesOptions } from \"./FrontendTiles\";\r\n\r\n/** @internal */\r\nexport interface BatchedTileParams extends TileParams {\r\n  childrenProps: Tileset3dSchema.Tile[] | undefined;\r\n  /** See BatchedTile.transformToRoot. */\r\n  transformToRoot: Transform | undefined;\r\n}\r\n\r\nlet channel: TileRequestChannel | undefined;\r\n\r\n/** @internal */\r\nexport class BatchedTile extends Tile {\r\n  private readonly _childrenProps?: Tileset3dSchema.Tile[];\r\n  private readonly _unskippable: boolean;\r\n  /** Transform from the tile's local coordinate system to that of the tileset. */\r\n  public readonly transformToRoot?: Transform;\r\n\r\n  public get batchedTree(): BatchedTileTree {\r\n    return this.tree as BatchedTileTree;\r\n  }\r\n\r\n  public constructor(params: BatchedTileParams, tree: BatchedTileTree) {\r\n    super(params, tree);\r\n\r\n    // The root tile never has content, so it doesn't count toward max levels to skip.\r\n    this._unskippable = 0 === (this.depth % frontendTilesOptions.maxLevelsToSkip);\r\n\r\n    if (params.childrenProps?.length)\r\n      this._childrenProps = params.childrenProps;\r\n\r\n    if (!this.contentId) {\r\n      this.setIsReady();\r\n      // mark \"undisplayable\"\r\n      this._maximumSize = 0;\r\n    }\r\n\r\n    if (!params.transformToRoot)\r\n      return;\r\n\r\n    this.transformToRoot = params.transformToRoot;\r\n    this.boundingSphere.transformBy(this.transformToRoot, this.boundingSphere);\r\n    this.transformToRoot.multiplyRange(this.range, this.range);\r\n    if (this._contentRange)\r\n      this.transformToRoot.multiplyRange(this._contentRange, this._contentRange);\r\n  }\r\n\r\n  private get _batchedChildren(): BatchedTile[] | undefined {\r\n    return this.children as BatchedTile[] | undefined;\r\n  }\r\n\r\n  public override computeLoadPriority(viewports: Iterable<Viewport>, _users: Iterable<TileUser>): number {\r\n    // Prioritize tiles closer to camera and center of attention (zoom point or screen center).\r\n    return RealityTileLoader.computeTileLocationPriority(this, viewports, this.tree.iModelTransform);\r\n  }\r\n\r\n  public selectTiles(selected: Set<BatchedTile>, args: TileDrawArgs, closestDisplayableAncestor: BatchedTile | undefined): void {\r\n    const vis = this.computeVisibility(args);\r\n    if (TileVisibility.OutsideFrustum === vis)\r\n      return;\r\n\r\n    if (this._unskippable) {\r\n      // Prevent this tile's content from being unloaded due to memory pressure.\r\n      args.touchedTiles.add(this);\r\n      args.markUsed(this);\r\n    }\r\n\r\n    closestDisplayableAncestor = this.hasGraphics ? this : closestDisplayableAncestor;\r\n    if (TileVisibility.TooCoarse === vis && (this.isReady || !this._unskippable)) {\r\n      args.markUsed(this);\r\n      args.markReady(this);\r\n      const childrenLoadStatus = this.loadChildren();\r\n      if (TileTreeLoadStatus.Loading === childrenLoadStatus)\r\n        args.markChildrenLoading();\r\n\r\n      const children = this._batchedChildren;\r\n      if (children) {\r\n        for (const child of children)\r\n          child.selectTiles(selected, args, closestDisplayableAncestor);\r\n\r\n        return;\r\n      }\r\n    }\r\n\r\n    // We want to display this tile. Request its content if not already loaded.\r\n    if ((TileVisibility.Visible === vis || this._unskippable) && !this.isReady)\r\n      args.insertMissing(this);\r\n\r\n    if (closestDisplayableAncestor)\r\n      selected.add(closestDisplayableAncestor);\r\n  }\r\n\r\n  protected override _loadChildren(resolve: (children: Tile[] | undefined) => void, reject: (error: Error) => void): void {\r\n    let children: BatchedTile[] | undefined;\r\n    if (this._childrenProps) {\r\n      try {\r\n        for (const childProps of this._childrenProps) {\r\n          const params = this.batchedTree.reader.readTileParams(childProps, this);\r\n          const child = new BatchedTile(params, this.batchedTree);\r\n          children = children ?? [];\r\n          children.push(child);\r\n        }\r\n      } catch (err) {\r\n        Logger.logException(loggerCategory, err);\r\n        children = undefined;\r\n        if (err instanceof Error)\r\n          reject(err);\r\n      }\r\n    }\r\n\r\n    resolve(children);\r\n  }\r\n\r\n  public override get channel(): TileRequestChannel {\r\n    if (!channel) {\r\n      channel = new TileRequestChannel(\"itwinjs-batched-models\", 20);\r\n      IModelApp.tileAdmin.channels.add(channel);\r\n    }\r\n\r\n    return channel;\r\n  }\r\n\r\n  public override async requestContent(_isCanceled: () => boolean): Promise<TileRequest.Response> {\r\n    const url = new URL(this.contentId, this.batchedTree.reader.baseUrl);\r\n    url.search = this.batchedTree.reader.baseUrl.search;\r\n    const response = await fetch(url.toString());\r\n    return response.arrayBuffer();\r\n  }\r\n\r\n  public override async readContent(data: TileRequest.ResponseData, system: RenderSystem, isCanceled?: () => boolean): Promise<TileContent> {\r\n    assert(data instanceof Uint8Array);\r\n    if (!(data instanceof Uint8Array))\r\n      return { };\r\n\r\n    try {\r\n      const content = await this.batchedTree.decoder.decode({\r\n        stream: ByteStream.fromUint8Array(data),\r\n        options: { tileId: this.contentId },\r\n        system,\r\n        isCanceled,\r\n        isLeaf: this.isLeaf,\r\n      });\r\n\r\n      if (this.transformToRoot) {\r\n        if (content.graphic) {\r\n          const branch = new GraphicBranch(true);\r\n          branch.add(content.graphic);\r\n          content.graphic = system.createBranch(branch, this.transformToRoot);\r\n        }\r\n\r\n        if (content.contentRange)\r\n          content.contentRange = this.transformToRoot.multiplyRange(content.contentRange);\r\n      }\r\n\r\n      return content;\r\n    } catch {\r\n      return { isLeaf: true };\r\n    }\r\n  }\r\n\r\n  protected override addRangeGraphic(builder: GraphicBuilder, type: TileBoundingBoxes): void {\r\n    if (TileBoundingBoxes.ChildVolumes !== type) {\r\n      super.addRangeGraphic(builder, type);\r\n      return;\r\n    }\r\n\r\n    builder.setSymbology(ColorDef.green, ColorDef.green, 2);\r\n    builder.addRangeBox(this.range);\r\n\r\n    this.loadChildren();\r\n    const children = this.children;\r\n    if (!children)\r\n      return;\r\n\r\n    builder.setSymbology(ColorDef.blue, ColorDef.blue.withTransparency(0xdf), 1);\r\n    for (const child of children) {\r\n      const range = child.range;\r\n      builder.addRangeBox(range);\r\n      builder.addRangeBox(range, true);\r\n    }\r\n  }\r\n\r\n  public prune(olderThan: BeTimePoint): void {\r\n    const children = this._batchedChildren;\r\n    if (!children)\r\n      return;\r\n\r\n    if (this.usageMarker.isExpired(olderThan)) {\r\n      this.disposeChildren();\r\n    } else {\r\n      for (const child of children)\r\n        child.prune(olderThan);\r\n    }\r\n  }\r\n}\r\n"]}
+{"version":3,"file":"BatchedTile.js","sourceRoot":"","sources":["../../src/BatchedTile.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,MAAM,EAAe,UAAU,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAE9E,OAAO,EAAE,QAAQ,EAAmB,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EACL,aAAa,EAAkB,SAAS,EAAE,iBAAiB,EAAgB,IAAI,EAAE,iBAAiB,EAC3D,kBAAkB,EAAE,kBAAkB,EAAY,cAAc,GACxG,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAElD,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AASvD,IAAI,OAAuC,CAAC;AAE5C,gBAAgB;AAChB,MAAM,OAAO,WAAY,SAAQ,IAAI;IAMnC,IAAW,WAAW;QACpB,OAAO,IAAI,CAAC,IAAuB,CAAC;IACtC,CAAC;IAED,YAAmB,MAAyB,EAAE,IAAqB;QACjE,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAEpB,kFAAkF;QAClF,IAAI,CAAC,YAAY,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,GAAG,oBAAoB,CAAC,eAAe,CAAC,CAAC;QAE9E,IAAI,MAAM,CAAC,aAAa,EAAE,MAAM;YAC9B,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;QAE7C,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YACnB,IAAI,CAAC,UAAU,EAAE,CAAC;YAClB,uBAAuB;YACvB,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;SACvB;QAED,IAAI,CAAC,MAAM,CAAC,eAAe;YACzB,OAAO;QAET,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QAC9C,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3E,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3D,IAAI,IAAI,CAAC,aAAa;YACpB,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;IAC/E,CAAC;IAED,IAAY,gBAAgB;QAC1B,OAAO,IAAI,CAAC,QAAqC,CAAC;IACpD,CAAC;IAEe,mBAAmB,CAAC,SAA6B,EAAE,MAA0B;QAC3F,2FAA2F;QAC3F,OAAO,iBAAiB,CAAC,2BAA2B,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACnG,CAAC;IAEM,WAAW,CAAC,QAA0B,EAAE,IAAkB,EAAE,0BAAmD;QACpH,MAAM,GAAG,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,cAAc,CAAC,cAAc,KAAK,GAAG;YACvC,OAAO;QAET,IAAI,IAAI,CAAC,YAAY,EAAE;YACrB,0EAA0E;YAC1E,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC5B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;SACrB;QAED,0BAA0B,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,0BAA0B,CAAC;QAClF,IAAI,cAAc,CAAC,SAAS,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE;YAC5E,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACpB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACrB,MAAM,kBAAkB,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;YAC/C,IAAI,kBAAkB,CAAC,OAAO,KAAK,kBAAkB;gBACnD,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAE7B,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC;YACvC,IAAI,QAAQ,EAAE;gBACZ,KAAK,MAAM,KAAK,IAAI,QAAQ;oBAC1B,KAAK,CAAC,WAAW,CAAC,QAAQ,EAAE,IAAI,EAAE,0BAA0B,CAAC,CAAC;gBAEhE,OAAO;aACR;SACF;QAED,2EAA2E;QAC3E,IAAI,CAAC,cAAc,CAAC,OAAO,KAAK,GAAG,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO;YACxE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QAE3B,IAAI,0BAA0B;YAC5B,QAAQ,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;IAC7C,CAAC;IAEkB,aAAa,CAAC,OAA+C,EAAE,MAA8B;QAC9G,IAAI,QAAmC,CAAC;QACxC,IAAI,IAAI,CAAC,cAAc,EAAE;YACvB,IAAI;gBACF,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,cAAc,EAAE;oBAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,cAAc,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;oBACxE,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;oBACxD,QAAQ,GAAG,QAAQ,IAAI,EAAE,CAAC;oBAC1B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;iBACtB;aACF;YAAC,OAAO,GAAG,EAAE;gBACZ,MAAM,CAAC,YAAY,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;gBACzC,QAAQ,GAAG,SAAS,CAAC;gBACrB,IAAI,GAAG,YAAY,KAAK;oBACtB,MAAM,CAAC,GAAG,CAAC,CAAC;aACf;SACF;QAED,OAAO,CAAC,QAAQ,CAAC,CAAC;IACpB,CAAC;IAED,IAAoB,OAAO;QACzB,IAAI,CAAC,OAAO,EAAE;YACZ,OAAO,GAAG,IAAI,kBAAkB,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC;YAC/D,SAAS,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;SAC3C;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEe,KAAK,CAAC,cAAc,CAAC,WAA0B;QAC7D,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACrE,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QACpD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC7C,OAAO,QAAQ,CAAC,WAAW,EAAE,CAAC;IAChC,CAAC;IAEe,KAAK,CAAC,WAAW,CAAC,IAA8B,EAAE,MAAoB,EAAE,UAA0B;QAChH,MAAM,CAAC,IAAI,YAAY,UAAU,CAAC,CAAC;QACnC,IAAI,CAAC,CAAC,IAAI,YAAY,UAAU,CAAC;YAC/B,OAAO,EAAG,CAAC;QAEb,IAAI;YACF,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC;YAEjD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;gBACpD,MAAM,EAAE,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC;gBACvC,OAAO,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE;gBACnC,MAAM;gBACN,UAAU;gBACV,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,qGAAqG;gBACrG,WAAW,EAAE,WAAW,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;aAC7E,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,eAAe,EAAE;gBACxB,IAAI,OAAO,CAAC,OAAO,EAAE;oBACnB,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC;oBACvC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBAC5B,OAAO,CAAC,OAAO,GAAG,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;iBACrE;gBAED,IAAI,OAAO,CAAC,YAAY;oBACtB,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;aACnF;YAED,OAAO,OAAO,CAAC;SAChB;QAAC,MAAM;YACN,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;SACzB;IACH,CAAC;IAEkB,eAAe,CAAC,OAAuB,EAAE,IAAuB;QACjF,IAAI,iBAAiB,CAAC,YAAY,KAAK,IAAI,EAAE;YAC3C,KAAK,CAAC,eAAe,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YACrC,OAAO;SACR;QAED,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACxD,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEhC,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,QAAQ;YACX,OAAO;QAET,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7E,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE;YAC5B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;YAC1B,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAC3B,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;SAClC;IACH,CAAC;IAEM,KAAK,CAAC,SAAsB;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC;QACvC,IAAI,CAAC,QAAQ;YACX,OAAO;QAET,IAAI,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE;YACzC,IAAI,CAAC,eAAe,EAAE,CAAC;SACxB;aAAM;YACL,KAAK,MAAM,KAAK,IAAI,QAAQ;gBAC1B,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;SAC1B;IACH,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { assert, BeTimePoint, ByteStream, Logger } from \"@itwin/core-bentley\";\r\nimport { Transform } from \"@itwin/core-geometry\";\r\nimport { ColorDef, Tileset3dSchema } from \"@itwin/core-common\";\r\nimport {\r\n  GraphicBranch, GraphicBuilder, IModelApp, RealityTileLoader, RenderSystem, Tile, TileBoundingBoxes, TileContent,\r\n  TileDrawArgs, TileParams, TileRequest, TileRequestChannel, TileTreeLoadStatus, TileUser, TileVisibility, Viewport,\r\n} from \"@itwin/core-frontend\";\r\nimport { loggerCategory } from \"./LoggerCategory\";\r\nimport { BatchedTileTree } from \"./BatchedTileTree\";\r\nimport { frontendTilesOptions } from \"./FrontendTiles\";\r\n\r\n/** @internal */\r\nexport interface BatchedTileParams extends TileParams {\r\n  childrenProps: Tileset3dSchema.Tile[] | undefined;\r\n  /** See BatchedTile.transformToRoot. */\r\n  transformToRoot: Transform | undefined;\r\n}\r\n\r\nlet channel: TileRequestChannel | undefined;\r\n\r\n/** @internal */\r\nexport class BatchedTile extends Tile {\r\n  private readonly _childrenProps?: Tileset3dSchema.Tile[];\r\n  private readonly _unskippable: boolean;\r\n  /** Transform from the tile's local coordinate system to that of the tileset. */\r\n  public readonly transformToRoot?: Transform;\r\n\r\n  public get batchedTree(): BatchedTileTree {\r\n    return this.tree as BatchedTileTree;\r\n  }\r\n\r\n  public constructor(params: BatchedTileParams, tree: BatchedTileTree) {\r\n    super(params, tree);\r\n\r\n    // The root tile never has content, so it doesn't count toward max levels to skip.\r\n    this._unskippable = 0 === (this.depth % frontendTilesOptions.maxLevelsToSkip);\r\n\r\n    if (params.childrenProps?.length)\r\n      this._childrenProps = params.childrenProps;\r\n\r\n    if (!this.contentId) {\r\n      this.setIsReady();\r\n      // mark \"undisplayable\"\r\n      this._maximumSize = 0;\r\n    }\r\n\r\n    if (!params.transformToRoot)\r\n      return;\r\n\r\n    this.transformToRoot = params.transformToRoot;\r\n    this.boundingSphere.transformBy(this.transformToRoot, this.boundingSphere);\r\n    this.transformToRoot.multiplyRange(this.range, this.range);\r\n    if (this._contentRange)\r\n      this.transformToRoot.multiplyRange(this._contentRange, this._contentRange);\r\n  }\r\n\r\n  private get _batchedChildren(): BatchedTile[] | undefined {\r\n    return this.children as BatchedTile[] | undefined;\r\n  }\r\n\r\n  public override computeLoadPriority(viewports: Iterable<Viewport>, _users: Iterable<TileUser>): number {\r\n    // Prioritize tiles closer to camera and center of attention (zoom point or screen center).\r\n    return RealityTileLoader.computeTileLocationPriority(this, viewports, this.tree.iModelTransform);\r\n  }\r\n\r\n  public selectTiles(selected: Set<BatchedTile>, args: TileDrawArgs, closestDisplayableAncestor: BatchedTile | undefined): void {\r\n    const vis = this.computeVisibility(args);\r\n    if (TileVisibility.OutsideFrustum === vis)\r\n      return;\r\n\r\n    if (this._unskippable) {\r\n      // Prevent this tile's content from being unloaded due to memory pressure.\r\n      args.touchedTiles.add(this);\r\n      args.markUsed(this);\r\n    }\r\n\r\n    closestDisplayableAncestor = this.hasGraphics ? this : closestDisplayableAncestor;\r\n    if (TileVisibility.TooCoarse === vis && (this.isReady || !this._unskippable)) {\r\n      args.markUsed(this);\r\n      args.markReady(this);\r\n      const childrenLoadStatus = this.loadChildren();\r\n      if (TileTreeLoadStatus.Loading === childrenLoadStatus)\r\n        args.markChildrenLoading();\r\n\r\n      const children = this._batchedChildren;\r\n      if (children) {\r\n        for (const child of children)\r\n          child.selectTiles(selected, args, closestDisplayableAncestor);\r\n\r\n        return;\r\n      }\r\n    }\r\n\r\n    // We want to display this tile. Request its content if not already loaded.\r\n    if ((TileVisibility.Visible === vis || this._unskippable) && !this.isReady)\r\n      args.insertMissing(this);\r\n\r\n    if (closestDisplayableAncestor)\r\n      selected.add(closestDisplayableAncestor);\r\n  }\r\n\r\n  protected override _loadChildren(resolve: (children: Tile[] | undefined) => void, reject: (error: Error) => void): void {\r\n    let children: BatchedTile[] | undefined;\r\n    if (this._childrenProps) {\r\n      try {\r\n        for (const childProps of this._childrenProps) {\r\n          const params = this.batchedTree.reader.readTileParams(childProps, this);\r\n          const child = new BatchedTile(params, this.batchedTree);\r\n          children = children ?? [];\r\n          children.push(child);\r\n        }\r\n      } catch (err) {\r\n        Logger.logException(loggerCategory, err);\r\n        children = undefined;\r\n        if (err instanceof Error)\r\n          reject(err);\r\n      }\r\n    }\r\n\r\n    resolve(children);\r\n  }\r\n\r\n  public override get channel(): TileRequestChannel {\r\n    if (!channel) {\r\n      channel = new TileRequestChannel(\"itwinjs-batched-models\", 20);\r\n      IModelApp.tileAdmin.channels.add(channel);\r\n    }\r\n\r\n    return channel;\r\n  }\r\n\r\n  public override async requestContent(_isCanceled: () => boolean): Promise<TileRequest.Response> {\r\n    const url = new URL(this.contentId, this.batchedTree.reader.baseUrl);\r\n    url.search = this.batchedTree.reader.baseUrl.search;\r\n    const response = await fetch(url.toString());\r\n    return response.arrayBuffer();\r\n  }\r\n\r\n  public override async readContent(data: TileRequest.ResponseData, system: RenderSystem, isCanceled?: () => boolean): Promise<TileContent> {\r\n    assert(data instanceof Uint8Array);\r\n    if (!(data instanceof Uint8Array))\r\n      return { };\r\n\r\n    try {\r\n      const modelGroups = this.batchedTree.modelGroups;\r\n\r\n      const content = await this.batchedTree.decoder.decode({\r\n        stream: ByteStream.fromUint8Array(data),\r\n        options: { tileId: this.contentId },\r\n        system,\r\n        isCanceled,\r\n        isLeaf: this.isLeaf,\r\n        // Don't waste time attempting to split based on model groupings if all models are in the same group.\r\n        modelGroups: modelGroups && modelGroups.length > 1 ? modelGroups : undefined,\r\n      });\r\n\r\n      if (this.transformToRoot) {\r\n        if (content.graphic) {\r\n          const branch = new GraphicBranch(true);\r\n          branch.add(content.graphic);\r\n          content.graphic = system.createBranch(branch, this.transformToRoot);\r\n        }\r\n\r\n        if (content.contentRange)\r\n          content.contentRange = this.transformToRoot.multiplyRange(content.contentRange);\r\n      }\r\n\r\n      return content;\r\n    } catch {\r\n      return { isLeaf: true };\r\n    }\r\n  }\r\n\r\n  protected override addRangeGraphic(builder: GraphicBuilder, type: TileBoundingBoxes): void {\r\n    if (TileBoundingBoxes.ChildVolumes !== type) {\r\n      super.addRangeGraphic(builder, type);\r\n      return;\r\n    }\r\n\r\n    builder.setSymbology(ColorDef.green, ColorDef.green, 2);\r\n    builder.addRangeBox(this.range);\r\n\r\n    this.loadChildren();\r\n    const children = this.children;\r\n    if (!children)\r\n      return;\r\n\r\n    builder.setSymbology(ColorDef.blue, ColorDef.blue.withTransparency(0xdf), 1);\r\n    for (const child of children) {\r\n      const range = child.range;\r\n      builder.addRangeBox(range);\r\n      builder.addRangeBox(range, true);\r\n    }\r\n  }\r\n\r\n  public prune(olderThan: BeTimePoint): void {\r\n    const children = this._batchedChildren;\r\n    if (!children)\r\n      return;\r\n\r\n    if (this.usageMarker.isExpired(olderThan)) {\r\n      this.disposeChildren();\r\n    } else {\r\n      for (const child of children)\r\n        child.prune(olderThan);\r\n    }\r\n  }\r\n}\r\n"]}

package/lib/esm/BatchedTileTree.d.ts

@@ -1,15 +1,15 @@
-import { Id64String } from "@itwin/core-bentley";
-import { Range3d } from "@itwin/core-geometry";
+import { Id64Set, Id64String } from "@itwin/core-bentley";
 import { RenderSchedule, ViewFlagOverrides } from "@itwin/core-common";
 import { ImdlDecoder, Tile, TileDrawArgs, TileTree, TileTreeParams } from "@itwin/core-frontend";
 import { BatchedTile, BatchedTileParams } from "./BatchedTile";
-import { BatchedTilesetReader } from "./BatchedTilesetReader";
+import { BatchedTilesetReader, ModelMetadata } from "./BatchedTilesetReader";
 /** @internal */
 export interface BatchedTileTreeParams extends TileTreeParams {
     rootTile: BatchedTileParams;
     reader: BatchedTilesetReader;
     script?: RenderSchedule.Script;
-    includedModels?: Map<Id64String, Range3d>;
+    models: Map<Id64String, ModelMetadata>;
+    modelGroups: Id64Set[] | undefined;
 }
 /** @internal */
 export declare class BatchedTileTree extends TileTree {
@@ -17,6 +17,7 @@ export declare class BatchedTileTree extends TileTree {
     readonly reader: BatchedTilesetReader;
     readonly scheduleScript?: RenderSchedule.Script;
     readonly decoder: ImdlDecoder;
+    readonly modelGroups: Id64Set[] | undefined;
     constructor(params: BatchedTileTreeParams);
     dispose(): void;
     get rootTile(): BatchedTile;

package/lib/esm/BatchedTileTree.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"BatchedTileTree.d.ts","sourceRoot":"","sources":["../../src/BatchedTileTree.ts"],"names":[],"mappings":"AAKA,OAAO,EAAe,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAC9D,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAyB,cAAc,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAC9F,OAAO,EACe,WAAW,EAAa,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,cAAc,EACzF,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAC/D,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAQ9D,gBAAgB;AAChB,MAAM,WAAW,qBAAsB,SAAQ,cAAc;IAC3D,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,MAAM,EAAE,oBAAoB,CAAC;IAC7B,MAAM,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;IAC/B,cAAc,CAAC,EAAE,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;CAC3C;AAED,gBAAgB;AAChB,qBAAa,eAAgB,SAAQ,QAAQ;IAC3C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAc;IACxC,SAAgB,MAAM,EAAE,oBAAoB,CAAC;IAC7C,SAAgB,cAAc,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;IACvD,SAAgB,OAAO,EAAE,WAAW,CAAC;gBAElB,MAAM,EAAE,qBAAqB;IAiBhC,OAAO,IAAI,IAAI;IAK/B,IAAoB,QAAQ,IAAI,WAAW,CAE1C;IAED,IAAoB,IAAI,IAAI,OAAO,CAElC;IAED,IAAoB,QAAQ,IAAI,MAAM,GAAG,SAAS,CAEjD;IAED,IAAoB,iBAAiB,IAAI,iBAAiB,CAEzD;IAGe,YAAY,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,EAAE;IAMxC,IAAI,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI;IAQ9B,KAAK,IAAI,IAAI;CAI9B"}
+{"version":3,"file":"BatchedTileTree.d.ts","sourceRoot":"","sources":["../../src/BatchedTileTree.ts"],"names":[],"mappings":"AAKA,OAAO,EAAe,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAAyB,cAAc,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAC9F,OAAO,EACe,WAAW,EAAa,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,cAAc,EACzF,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAC/D,OAAO,EAAE,oBAAoB,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAQ7E,gBAAgB;AAChB,MAAM,WAAW,qBAAsB,SAAQ,cAAc;IAC3D,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,MAAM,EAAE,oBAAoB,CAAC;IAC7B,MAAM,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;IAC/B,MAAM,EAAE,GAAG,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;IACvC,WAAW,EAAE,OAAO,EAAE,GAAG,SAAS,CAAC;CACpC;AAED,gBAAgB;AAChB,qBAAa,eAAgB,SAAQ,QAAQ;IAC3C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAc;IACxC,SAAgB,MAAM,EAAE,oBAAoB,CAAC;IAC7C,SAAgB,cAAc,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;IACvD,SAAgB,OAAO,EAAE,WAAW,CAAC;IACrC,SAAgB,WAAW,EAAE,OAAO,EAAE,GAAG,SAAS,CAAC;gBAEhC,MAAM,EAAE,qBAAqB;IAkBhC,OAAO,IAAI,IAAI;IAK/B,IAAoB,QAAQ,IAAI,WAAW,CAE1C;IAED,IAAoB,IAAI,IAAI,OAAO,CAElC;IAED,IAAoB,QAAQ,IAAI,MAAM,GAAG,SAAS,CAEjD;IAED,IAAoB,iBAAiB,IAAI,iBAAiB,CAEzD;IAGe,YAAY,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,EAAE;IAMxC,IAAI,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI;IAQ9B,KAAK,IAAI,IAAI;CAI9B"}

package/lib/esm/BatchedTileTree.js

@@ -18,6 +18,7 @@ export class BatchedTileTree extends TileTree {
         this._rootTile = new BatchedTile(params.rootTile, this);
         this.reader = params.reader;
         this.scheduleScript = params.script;
+        this.modelGroups = params.modelGroups;
         this.decoder = acquireImdlDecoder({
             type: BatchType.Primary,
             timeline: this.scheduleScript,

package/lib/esm/BatchedTileTree.js.map

@@ -1 +1 @@
-{"version":3,"file":"BatchedTileTree.js","sourceRoot":"","sources":["../../src/BatchedTileTree.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,WAAW,EAAc,MAAM,qBAAqB,CAAC;AAE9D,OAAO,EAAE,SAAS,EAAE,UAAU,EAAqC,MAAM,oBAAoB,CAAC;AAC9F,OAAO,EACL,kBAAkB,EAAe,SAAS,EAAsB,QAAQ,GACzE,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAqB,MAAM,eAAe,CAAC;AAE/D,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAEvD,MAAM,gBAAgB,GAAsB;IAC1C,UAAU,EAAE,UAAU,CAAC,WAAW;IAClC,YAAY,EAAE,KAAK;CACpB,CAAC;AAUF,gBAAgB;AAChB,MAAM,OAAO,eAAgB,SAAQ,QAAQ;IAM3C,YAAmB,MAA6B;QAC9C,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,SAAS,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACxD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC;QAEpC,IAAI,CAAC,OAAO,GAAG,kBAAkB,CAAC;YAChC,IAAI,EAAE,SAAS,CAAC,OAAO;YACvB,QAAQ,EAAE,IAAI,CAAC,cAAc;YAC7B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,YAAY,EAAE,IAAI,CAAC,OAAO;YAC1B,IAAI,EAAE,IAAI;YACV,sBAAsB,EAAE,KAAK;YAC7B,QAAQ,EAAE,CAAC,SAAS,CAAC,SAAS,CAAC,kBAAkB;SAClD,CAAC,CAAC;IACL,CAAC;IAEe,OAAO;QACrB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACvB,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;IAED,IAAoB,QAAQ;QAC1B,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAoB,IAAI;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAoB,QAAQ;QAC1B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAoB,iBAAiB;QACnC,OAAO,oBAAoB,CAAC,WAAW,CAAC,CAAC,CAAA,EAAG,CAAC,CAAC,CAAC,gBAAgB,CAAC;IAClE,CAAC;IAED,gEAAgE;IAChD,YAAY,CAAC,IAAkB;QAC7C,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAe,CAAC;QACxC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QACrD,OAAO,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC;IAEe,IAAI,CAAC,IAAkB;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACrC,KAAK,MAAM,IAAI,IAAI,KAAK;YACtB,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAE1B,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAEe,KAAK;QACnB,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { BeTimePoint, Id64String } from \"@itwin/core-bentley\";\r\nimport { Range3d } from \"@itwin/core-geometry\";\r\nimport { BatchType, RenderMode, RenderSchedule, ViewFlagOverrides } from \"@itwin/core-common\";\r\nimport {\r\n  acquireImdlDecoder, ImdlDecoder, IModelApp, Tile, TileDrawArgs, TileTree, TileTreeParams,\r\n} from \"@itwin/core-frontend\";\r\nimport { BatchedTile, BatchedTileParams } from \"./BatchedTile\";\r\nimport { BatchedTilesetReader } from \"./BatchedTilesetReader\";\r\nimport { frontendTilesOptions } from \"./FrontendTiles\";\r\n\r\nconst defaultViewFlags: ViewFlagOverrides = {\r\n  renderMode: RenderMode.SmoothShade,\r\n  visibleEdges: false,\r\n};\r\n\r\n/** @internal */\r\nexport interface BatchedTileTreeParams extends TileTreeParams {\r\n  rootTile: BatchedTileParams;\r\n  reader: BatchedTilesetReader;\r\n  script?: RenderSchedule.Script;\r\n  includedModels?: Map<Id64String, Range3d>;\r\n}\r\n\r\n/** @internal */\r\nexport class BatchedTileTree extends TileTree {\r\n  private readonly _rootTile: BatchedTile;\r\n  public readonly reader: BatchedTilesetReader;\r\n  public readonly scheduleScript?: RenderSchedule.Script;\r\n  public readonly decoder: ImdlDecoder;\r\n\r\n  public constructor(params: BatchedTileTreeParams) {\r\n    super(params);\r\n    this._rootTile = new BatchedTile(params.rootTile, this);\r\n    this.reader = params.reader;\r\n    this.scheduleScript = params.script;\r\n\r\n    this.decoder = acquireImdlDecoder({\r\n      type: BatchType.Primary,\r\n      timeline: this.scheduleScript,\r\n      iModel: this.iModel,\r\n      batchModelId: this.modelId,\r\n      is3d: true,\r\n      containsTransformNodes: false,\r\n      noWorker: !IModelApp.tileAdmin.decodeImdlInWorker,\r\n    });\r\n  }\r\n\r\n  public override dispose(): void {\r\n    this.decoder.release();\r\n    super.dispose();\r\n  }\r\n\r\n  public override get rootTile(): BatchedTile {\r\n    return this._rootTile;\r\n  }\r\n\r\n  public override get is3d(): boolean {\r\n    return true;\r\n  }\r\n\r\n  public override get maxDepth(): number | undefined {\r\n    return undefined;\r\n  }\r\n\r\n  public override get viewFlagOverrides(): ViewFlagOverrides {\r\n    return frontendTilesOptions.enableEdges ?{ } : defaultViewFlags;\r\n  }\r\n\r\n  // eslint-disable-next-line @typescript-eslint/naming-convention\r\n  public override _selectTiles(args: TileDrawArgs): Tile[] {\r\n    const selected = new Set<BatchedTile>();\r\n    this.rootTile.selectTiles(selected, args, undefined);\r\n    return Array.from(selected);\r\n  }\r\n\r\n  public override draw(args: TileDrawArgs): void {\r\n    const tiles = this.selectTiles(args);\r\n    for (const tile of tiles)\r\n      tile.drawGraphics(args);\r\n\r\n    args.drawGraphics();\r\n  }\r\n\r\n  public override prune(): void {\r\n    const olderThan = BeTimePoint.now().minus(this.expirationTime);\r\n    this.rootTile.prune(olderThan);\r\n  }\r\n}\r\n"]}
+{"version":3,"file":"BatchedTileTree.js","sourceRoot":"","sources":["../../src/BatchedTileTree.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,WAAW,EAAuB,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAAE,SAAS,EAAE,UAAU,EAAqC,MAAM,oBAAoB,CAAC;AAC9F,OAAO,EACL,kBAAkB,EAAe,SAAS,EAAsB,QAAQ,GACzE,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAqB,MAAM,eAAe,CAAC;AAE/D,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAEvD,MAAM,gBAAgB,GAAsB;IAC1C,UAAU,EAAE,UAAU,CAAC,WAAW;IAClC,YAAY,EAAE,KAAK;CACpB,CAAC;AAWF,gBAAgB;AAChB,MAAM,OAAO,eAAgB,SAAQ,QAAQ;IAO3C,YAAmB,MAA6B;QAC9C,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,SAAS,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACxD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC;QACpC,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QAEtC,IAAI,CAAC,OAAO,GAAG,kBAAkB,CAAC;YAChC,IAAI,EAAE,SAAS,CAAC,OAAO;YACvB,QAAQ,EAAE,IAAI,CAAC,cAAc;YAC7B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,YAAY,EAAE,IAAI,CAAC,OAAO;YAC1B,IAAI,EAAE,IAAI;YACV,sBAAsB,EAAE,KAAK;YAC7B,QAAQ,EAAE,CAAC,SAAS,CAAC,SAAS,CAAC,kBAAkB;SAClD,CAAC,CAAC;IACL,CAAC;IAEe,OAAO;QACrB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACvB,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;IAED,IAAoB,QAAQ;QAC1B,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAoB,IAAI;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAoB,QAAQ;QAC1B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAoB,iBAAiB;QACnC,OAAO,oBAAoB,CAAC,WAAW,CAAC,CAAC,CAAA,EAAG,CAAC,CAAC,CAAC,gBAAgB,CAAC;IAClE,CAAC;IAED,gEAAgE;IAChD,YAAY,CAAC,IAAkB;QAC7C,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAe,CAAC;QACxC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QACrD,OAAO,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC;IAEe,IAAI,CAAC,IAAkB;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACrC,KAAK,MAAM,IAAI,IAAI,KAAK;YACtB,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAE1B,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAEe,KAAK;QACnB,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { BeTimePoint, Id64Set, Id64String } from \"@itwin/core-bentley\";\r\nimport { BatchType, RenderMode, RenderSchedule, ViewFlagOverrides } from \"@itwin/core-common\";\r\nimport {\r\n  acquireImdlDecoder, ImdlDecoder, IModelApp, Tile, TileDrawArgs, TileTree, TileTreeParams,\r\n} from \"@itwin/core-frontend\";\r\nimport { BatchedTile, BatchedTileParams } from \"./BatchedTile\";\r\nimport { BatchedTilesetReader, ModelMetadata } from \"./BatchedTilesetReader\";\r\nimport { frontendTilesOptions } from \"./FrontendTiles\";\r\n\r\nconst defaultViewFlags: ViewFlagOverrides = {\r\n  renderMode: RenderMode.SmoothShade,\r\n  visibleEdges: false,\r\n};\r\n\r\n/** @internal */\r\nexport interface BatchedTileTreeParams extends TileTreeParams {\r\n  rootTile: BatchedTileParams;\r\n  reader: BatchedTilesetReader;\r\n  script?: RenderSchedule.Script;\r\n  models: Map<Id64String, ModelMetadata>;\r\n  modelGroups: Id64Set[] | undefined;\r\n}\r\n\r\n/** @internal */\r\nexport class BatchedTileTree extends TileTree {\r\n  private readonly _rootTile: BatchedTile;\r\n  public readonly reader: BatchedTilesetReader;\r\n  public readonly scheduleScript?: RenderSchedule.Script;\r\n  public readonly decoder: ImdlDecoder;\r\n  public readonly modelGroups: Id64Set[] | undefined;\r\n\r\n  public constructor(params: BatchedTileTreeParams) {\r\n    super(params);\r\n    this._rootTile = new BatchedTile(params.rootTile, this);\r\n    this.reader = params.reader;\r\n    this.scheduleScript = params.script;\r\n    this.modelGroups = params.modelGroups;\r\n\r\n    this.decoder = acquireImdlDecoder({\r\n      type: BatchType.Primary,\r\n      timeline: this.scheduleScript,\r\n      iModel: this.iModel,\r\n      batchModelId: this.modelId,\r\n      is3d: true,\r\n      containsTransformNodes: false,\r\n      noWorker: !IModelApp.tileAdmin.decodeImdlInWorker,\r\n    });\r\n  }\r\n\r\n  public override dispose(): void {\r\n    this.decoder.release();\r\n    super.dispose();\r\n  }\r\n\r\n  public override get rootTile(): BatchedTile {\r\n    return this._rootTile;\r\n  }\r\n\r\n  public override get is3d(): boolean {\r\n    return true;\r\n  }\r\n\r\n  public override get maxDepth(): number | undefined {\r\n    return undefined;\r\n  }\r\n\r\n  public override get viewFlagOverrides(): ViewFlagOverrides {\r\n    return frontendTilesOptions.enableEdges ?{ } : defaultViewFlags;\r\n  }\r\n\r\n  // eslint-disable-next-line @typescript-eslint/naming-convention\r\n  public override _selectTiles(args: TileDrawArgs): Tile[] {\r\n    const selected = new Set<BatchedTile>();\r\n    this.rootTile.selectTiles(selected, args, undefined);\r\n    return Array.from(selected);\r\n  }\r\n\r\n  public override draw(args: TileDrawArgs): void {\r\n    const tiles = this.selectTiles(args);\r\n    for (const tile of tiles)\r\n      tile.drawGraphics(args);\r\n\r\n    args.drawGraphics();\r\n  }\r\n\r\n  public override prune(): void {\r\n    const olderThan = BeTimePoint.now().minus(this.expirationTime);\r\n    this.rootTile.prune(olderThan);\r\n  }\r\n}\r\n"]}

package/lib/esm/BatchedTileTreeReference.d.ts

@@ -1,34 +1,32 @@
 import { Range3d, Transform } from "@itwin/core-geometry";
-import { BatchType, FeatureAppearance, FeatureAppearanceProvider, FeatureAppearanceSource, GeometryClass, RenderSchedule } from "@itwin/core-common";
-import { AnimationNodeId, SceneContext, TileDrawArgs, TileTree, TileTreeOwner, TileTreeReference } from "@itwin/core-frontend";
+import { BatchType, FeatureAppearance, FeatureAppearanceProvider, FeatureAppearanceSource, GeometryClass, ViewFlagOverrides } from "@itwin/core-common";
+import { RenderClipVolume, SceneContext, TileDrawArgs, TileTree, TileTreeOwner, TileTreeReference } from "@itwin/core-frontend";
 import { BatchedModels } from "./BatchedModels";
-/** @internal */
-export declare abstract class BatchedTileTreeReference extends TileTreeReference {
-    protected readonly _treeOwner: TileTreeOwner;
-    protected constructor(treeOwner: TileTreeOwner);
-    get treeOwner(): TileTreeOwner;
-    protected computeBaseTransform(tree: TileTree): Transform;
-    protected computeTransform(tree: TileTree): Transform;
+import { ModelGroupInfo } from "./ModelGroup";
+export interface BatchedTileTreeReferenceArgs {
+    readonly models: BatchedModels;
+    readonly groups: ReadonlyArray<ModelGroupInfo>;
+    readonly treeOwner: TileTreeOwner;
+    readonly getCurrentTimePoint: () => number;
 }
-export declare class PrimaryBatchedTileTreeReference extends BatchedTileTreeReference implements FeatureAppearanceProvider {
-    private readonly _models;
-    constructor(treeOwner: TileTreeOwner, models: BatchedModels);
-    unionFitRange(range: Range3d): void;
-    getAppearanceProvider(): FeatureAppearanceProvider | undefined;
+export declare class BatchedTileTreeReference extends TileTreeReference implements FeatureAppearanceProvider {
+    private readonly _args;
+    private readonly _groupIndex;
+    private readonly _animationNodeId?;
+    private readonly _branchId?;
+    constructor(args: BatchedTileTreeReferenceArgs, groupIndex: number, animationNodeId: number | undefined);
+    private get _groupInfo();
+    get treeOwner(): TileTreeOwner;
+    getAppearanceProvider(): FeatureAppearanceProvider;
+    protected getClipVolume(): RenderClipVolume | undefined;
     getFeatureAppearance(source: FeatureAppearanceSource, elemLo: number, elemHi: number, subcatLo: number, subcatHi: number, geomClass: GeometryClass, modelLo: number, modelHi: number, type: BatchType, animationNodeId: number): FeatureAppearance | undefined;
-    getAnimationTransformNodeId(): AnimationNodeId;
-}
-export interface AnimationNode {
-    readonly timeline: RenderSchedule.ModelTimeline;
-    readonly nodeId: number;
-    getCurrentTimePoint(): number;
-}
-export declare class AnimatedBatchedTileTreeReference extends BatchedTileTreeReference {
-    private readonly _node;
-    private readonly _branchId;
-    constructor(treeOwner: TileTreeOwner, node: AnimationNode);
-    getAnimationTransformNodeId(): number;
-    computeBaseTransform(tree: TileTree): Transform;
+    unionFitRange(range: Range3d): void;
+    get castsShadows(): boolean;
+    getViewFlagOverrides(): ViewFlagOverrides;
+    draw(args: TileDrawArgs): void;
+    protected computeTransform(tree: TileTree): Transform;
+    protected getAnimationTransformNodeId(): number | undefined;
+    protected getGroupNodeId(): number | undefined;
     createDrawArgs(context: SceneContext): TileDrawArgs | undefined;
 }
 //# sourceMappingURL=BatchedTileTreeReference.d.ts.map

package/lib/esm/BatchedTileTreeReference.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"BatchedTileTreeReference.d.ts","sourceRoot":"","sources":["../../src/BatchedTileTreeReference.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EACL,SAAS,EAAE,iBAAiB,EAAE,yBAAyB,EAAE,uBAAuB,EAAE,aAAa,EAAE,cAAc,EAChH,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,eAAe,EAA2B,YAAY,EAAE,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE,iBAAiB,EACjH,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,gBAAgB;AAChB,8BAAsB,wBAAyB,SAAQ,iBAAiB;IACtE,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,aAAa,CAAC;IAE7C,SAAS,aAAa,SAAS,EAAE,aAAa;IAK9C,IAAoB,SAAS,IAAI,aAAa,CAE7C;IAED,SAAS,CAAC,oBAAoB,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS;cAItC,gBAAgB,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS;CAK/D;AAED,qBAAa,+BAAgC,SAAQ,wBAAyB,YAAW,yBAAyB;IAChH,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAgB;gBAErB,SAAS,EAAE,aAAa,EAAE,MAAM,EAAE,aAAa;IAKlD,aAAa,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI;IAInC,qBAAqB,IAAI,yBAAyB,GAAG,SAAS;IAIvE,oBAAoB,CACzB,MAAM,EAAE,uBAAuB,EAC/B,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAC9B,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAClC,SAAS,EAAE,aAAa,EACxB,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAChC,IAAI,EAAE,SAAS,EACf,eAAe,EAAE,MAAM,GACtB,iBAAiB,GAAG,SAAS;IAOhB,2BAA2B;CAG5C;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAC,aAAa,CAAC;IAChD,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,mBAAmB,IAAI,MAAM,CAAC;CAC/B;AAED,qBAAa,gCAAiC,SAAQ,wBAAwB;IAC5E,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAgB;IACtC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;gBAEhB,SAAS,EAAE,aAAa,EAAE,IAAI,EAAE,aAAa;IAMhD,2BAA2B,IAAI,MAAM;IAIrC,oBAAoB,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS;IAS/C,cAAc,CAAC,OAAO,EAAE,YAAY,GAAG,YAAY,GAAG,SAAS;CAShF"}
+{"version":3,"file":"BatchedTileTreeReference.d.ts","sourceRoot":"","sources":["../../src/BatchedTileTreeReference.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EACL,SAAS,EAAE,iBAAiB,EAAE,yBAAyB,EAAE,uBAAuB,EAAE,aAAa,EAAE,iBAAiB,EACnH,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACoB,gBAAgB,EAAE,YAAY,EAAE,YAAY,EAAmB,QAAQ,EAAE,aAAa,EAAE,iBAAiB,EACnI,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE9C,MAAM,WAAW,4BAA4B;IAC3C,QAAQ,CAAC,MAAM,EAAE,aAAa,CAAC;IAC/B,QAAQ,CAAC,MAAM,EAAE,aAAa,CAAC,cAAc,CAAC,CAAC;IAC/C,QAAQ,CAAC,SAAS,EAAE,aAAa,CAAC;IAClC,QAAQ,CAAC,mBAAmB,EAAE,MAAM,MAAM,CAAC;CAC5C;AAED,qBAAa,wBAAyB,SAAQ,iBAAkB,YAAW,yBAAyB;IAClG,OAAO,CAAC,QAAQ,CAAC,KAAK,CAA+B;IACrD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAS;IAC3C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAS;gBAEjB,IAAI,EAAE,4BAA4B,EAAE,UAAU,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,GAAG,SAAS;IAW9G,OAAO,KAAK,UAAU,GAGrB;IAED,IAAoB,SAAS,IAAI,aAAa,CAE7C;IAEe,qBAAqB,IAAI,yBAAyB;cAI/C,aAAa,IAAI,gBAAgB,GAAG,SAAS;IAIzD,oBAAoB,CACzB,MAAM,EAAE,uBAAuB,EAC/B,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAC9B,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAClC,SAAS,EAAE,aAAa,EACxB,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAChC,IAAI,EAAE,SAAS,EACf,eAAe,EAAE,MAAM,GACtB,iBAAiB,GAAG,SAAS;IAOhB,aAAa,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI;IAInD,IAAoB,YAAY,IAAI,OAAO,CAK1C;IAEe,oBAAoB,IAAI,iBAAiB;IAIzC,IAAI,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI;cAO3B,gBAAgB,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS;cAuB3C,2BAA2B;cAI3B,cAAc;IAIjB,cAAc,CAAC,OAAO,EAAE,YAAY,GAAG,YAAY,GAAG,SAAS;CAgBhF"}

package/lib/esm/BatchedTileTreeReference.js

@@ -2,67 +2,89 @@
 * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
 * See LICENSE.md in the project root for license terms and full copyright notice.
 *--------------------------------------------------------------------------------------------*/
-import { AnimationNodeId, formatAnimationBranchId, TileTreeReference, } from "@itwin/core-frontend";
-/** @internal */
+import { assert } from "@itwin/core-bentley";
+import { formatAnimationBranchId, TileGraphicType, TileTreeReference, } from "@itwin/core-frontend";
 export class BatchedTileTreeReference extends TileTreeReference {
-    constructor(treeOwner) {
+    constructor(args, groupIndex, animationNodeId) {
         super();
-        this._treeOwner = treeOwner;
+        this._args = args;
+        this._groupIndex = groupIndex;
+        this._animationNodeId = animationNodeId;
+        if (undefined !== animationNodeId) {
+            assert(undefined !== this._groupInfo.timeline);
+            this._branchId = formatAnimationBranchId(this._groupInfo.timeline.modelId, animationNodeId);
+        }
     }
-    get treeOwner() {
-        return this._treeOwner;
-    }
-    computeBaseTransform(tree) {
-        return super.computeTransform(tree);
+    get _groupInfo() {
+        assert(this._groupIndex < this._args.groups.length);
+        return this._args.groups[this._groupIndex];
     }
-    computeTransform(tree) {
-        const baseTf = this.computeBaseTransform(tree);
-        // ###TODO this.view.modelDisplayTransformProvider?.getModelDisplayTransform(modelId...)
-        return baseTf;
-    }
-}
-export class PrimaryBatchedTileTreeReference extends BatchedTileTreeReference {
-    constructor(treeOwner, models) {
-        super(treeOwner);
-        this._models = models;
-    }
-    unionFitRange(range) {
-        this._models.unionRange(range);
+    get treeOwner() {
+        return this._args.treeOwner;
     }
     getAppearanceProvider() {
         return this;
     }
+    getClipVolume() {
+        return this._groupInfo.clip;
+    }
     getFeatureAppearance(source, elemLo, elemHi, subcatLo, subcatHi, geomClass, modelLo, modelHi, type, animationNodeId) {
-        if (!this._models.isViewed(modelLo, modelHi))
+        if (!this._args.models.isViewed(modelLo, modelHi))
             return undefined;
         return source.getAppearance(elemLo, elemHi, subcatLo, subcatHi, geomClass, modelLo, modelHi, type, animationNodeId);
     }
-    getAnimationTransformNodeId() {
-        return AnimationNodeId.Untransformed;
+    unionFitRange(range) {
+        this._args.models.unionRange(range);
     }
-}
-export class AnimatedBatchedTileTreeReference extends BatchedTileTreeReference {
-    constructor(treeOwner, node) {
-        super(treeOwner);
-        this._node = node;
-        this._branchId = formatAnimationBranchId(node.timeline.modelId, node.nodeId);
+    get castsShadows() {
+        if (this._groupInfo.planProjection)
+            return false;
+        return super.castsShadows;
+    }
+    getViewFlagOverrides() {
+        return this._groupInfo.viewFlags;
+    }
+    draw(args) {
+        if (this._groupInfo.planProjection?.overlay)
+            args.context.withGraphicType(TileGraphicType.Overlay, () => args.tree.draw(args));
+        else
+            super.draw(args);
+    }
+    computeTransform(tree) {
+        const group = this._groupInfo;
+        let baseTf = super.computeTransform(tree);
+        if (group.planProjection) {
+            baseTf = baseTf.clone();
+            baseTf.origin.z += group.planProjection.elevation;
+        }
+        if (group.timeline) {
+            assert(undefined !== this._animationNodeId);
+            const animTf = group.timeline.getTransform(this._animationNodeId, this._args.getCurrentTimePoint());
+            if (animTf)
+                animTf.multiplyTransformTransform(baseTf, baseTf);
+        }
+        const displayTf = group.displayTransform;
+        if (!displayTf)
+            return baseTf;
+        return displayTf.premultiply ? displayTf.transform.multiplyTransformTransform(baseTf) : baseTf.multiplyTransformTransform(displayTf.transform);
     }
     getAnimationTransformNodeId() {
-        return this._node.nodeId;
+        return this._animationNodeId;
     }
-    computeBaseTransform(tree) {
-        const tf = super.computeBaseTransform(tree);
-        const animTf = this._node.timeline.getTransform(this._node.nodeId, this._node.getCurrentTimePoint());
-        if (animTf)
-            animTf.multiplyTransformTransform(tf, tf);
-        return tf;
+    getGroupNodeId() {
+        return this._args.groups.length > 1 ? this._groupIndex : undefined;
     }
     createDrawArgs(context) {
-        const animBranch = context.viewport.target.animationBranches?.branchStates.get(this._branchId);
-        if (animBranch && animBranch.omit)
-            return undefined;
+        if (this._branchId) {
+            const branch = context.viewport.target.animationBranches?.branchStates.get(this._branchId);
+            if (branch?.omit) {
+                // This branch is not supposed to be drawn
+                return undefined;
+            }
+        }
         const args = super.createDrawArgs(context);
         // ###TODO args.boundingRange = args.tree.getTransformNodeRange(this._animationTransformNodeId);
+        // ###TODO if PlanProjectionSettings.enforceDisplayPriority, createGraphicLayerContainer.
         return args;
     }
 }

package/lib/esm/BatchedTileTreeReference.js.map

@@ -1 +1 @@
-{"version":3,"file":"BatchedTileTreeReference.js","sourceRoot":"","sources":["../../src/BatchedTileTreeReference.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAM/F,OAAO,EACL,eAAe,EAAE,uBAAuB,EAAuD,iBAAiB,GACjH,MAAM,sBAAsB,CAAC;AAG9B,gBAAgB;AAChB,MAAM,OAAgB,wBAAyB,SAAQ,iBAAiB;IAGtE,YAAsB,SAAwB;QAC5C,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;IAC9B,CAAC;IAED,IAAoB,SAAS;QAC3B,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAES,oBAAoB,CAAC,IAAc;QAC3C,OAAO,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC;IAEkB,gBAAgB,CAAC,IAAc;QAChD,MAAM,MAAM,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC/C,wFAAwF;QACxF,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED,MAAM,OAAO,+BAAgC,SAAQ,wBAAwB;IAG3E,YAAmB,SAAwB,EAAE,MAAqB;QAChE,KAAK,CAAC,SAAS,CAAC,CAAC;QACjB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAEe,aAAa,CAAC,KAAc;QAC1C,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;IAEe,qBAAqB;QACnC,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,oBAAoB,CACzB,MAA+B,EAC/B,MAAc,EAAE,MAAc,EAC9B,QAAgB,EAAE,QAAgB,EAClC,SAAwB,EACxB,OAAe,EAAE,OAAe,EAChC,IAAe,EACf,eAAuB;QAEvB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;YAC1C,OAAO,SAAS,CAAC;QAEnB,OAAO,MAAM,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,eAAe,CAAC,CAAC;IACtH,CAAC;IAEe,2BAA2B;QACzC,OAAO,eAAe,CAAC,aAAa,CAAC;IACvC,CAAC;CACF;AAQD,MAAM,OAAO,gCAAiC,SAAQ,wBAAwB;IAI5E,YAAmB,SAAwB,EAAE,IAAmB;QAC9D,KAAK,CAAC,SAAS,CAAC,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,SAAS,GAAG,uBAAuB,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IAC/E,CAAC;IAEe,2BAA2B;QACzC,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;IAC3B,CAAC;IAEe,oBAAoB,CAAC,IAAc;QACjD,MAAM,EAAE,GAAG,KAAK,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,CAAC;QACrG,IAAI,MAAM;YACR,MAAM,CAAC,0BAA0B,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAE5C,OAAO,EAAE,CAAC;IACZ,CAAC;IAEe,cAAc,CAAC,OAAqB;QAClD,MAAM,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,EAAE,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/F,IAAI,UAAU,IAAI,UAAU,CAAC,IAAI;YAC/B,OAAO,SAAS,CAAC;QAEnB,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAC3C,gGAAgG;QAChG,OAAO,IAAI,CAAC;IACd,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { Range3d, Transform } from \"@itwin/core-geometry\";\r\nimport {\r\n  BatchType, FeatureAppearance, FeatureAppearanceProvider, FeatureAppearanceSource, GeometryClass, RenderSchedule,\r\n} from \"@itwin/core-common\";\r\nimport {\r\n  AnimationNodeId, formatAnimationBranchId, SceneContext, TileDrawArgs, TileTree, TileTreeOwner, TileTreeReference,\r\n} from \"@itwin/core-frontend\";\r\nimport { BatchedModels } from \"./BatchedModels\";\r\n\r\n/** @internal */\r\nexport abstract class BatchedTileTreeReference extends TileTreeReference {\r\n  protected readonly _treeOwner: TileTreeOwner;\r\n\r\n  protected constructor(treeOwner: TileTreeOwner) {\r\n    super();\r\n    this._treeOwner = treeOwner;\r\n  }\r\n\r\n  public override get treeOwner(): TileTreeOwner {\r\n    return this._treeOwner;\r\n  }\r\n\r\n  protected computeBaseTransform(tree: TileTree): Transform {\r\n    return super.computeTransform(tree);\r\n  }\r\n\r\n  protected override computeTransform(tree: TileTree): Transform {\r\n    const baseTf = this.computeBaseTransform(tree);\r\n    // ###TODO this.view.modelDisplayTransformProvider?.getModelDisplayTransform(modelId...)\r\n    return baseTf;\r\n  }\r\n}\r\n\r\nexport class PrimaryBatchedTileTreeReference extends BatchedTileTreeReference implements FeatureAppearanceProvider {\r\n  private readonly _models: BatchedModels;\r\n\r\n  public constructor(treeOwner: TileTreeOwner, models: BatchedModels) {\r\n    super(treeOwner);\r\n    this._models = models;\r\n  }\r\n\r\n  public override unionFitRange(range: Range3d): void {\r\n    this._models.unionRange(range);\r\n  }\r\n\r\n  public override getAppearanceProvider(): FeatureAppearanceProvider | undefined {\r\n    return this;\r\n  }\r\n\r\n  public getFeatureAppearance(\r\n    source: FeatureAppearanceSource,\r\n    elemLo: number, elemHi: number,\r\n    subcatLo: number, subcatHi: number,\r\n    geomClass: GeometryClass,\r\n    modelLo: number, modelHi: number,\r\n    type: BatchType,\r\n    animationNodeId: number,\r\n  ): FeatureAppearance | undefined {\r\n    if (!this._models.isViewed(modelLo, modelHi))\r\n      return undefined;\r\n\r\n    return source.getAppearance(elemLo, elemHi, subcatLo, subcatHi, geomClass, modelLo, modelHi, type, animationNodeId);\r\n  }\r\n\r\n  public override getAnimationTransformNodeId() {\r\n    return AnimationNodeId.Untransformed;\r\n  }\r\n}\r\n\r\nexport interface AnimationNode {\r\n  readonly timeline: RenderSchedule.ModelTimeline;\r\n  readonly nodeId: number;\r\n  getCurrentTimePoint(): number;\r\n}\r\n\r\nexport class AnimatedBatchedTileTreeReference extends BatchedTileTreeReference {\r\n  private readonly _node: AnimationNode;\r\n  private readonly _branchId: string;\r\n\r\n  public constructor(treeOwner: TileTreeOwner, node: AnimationNode) {\r\n    super(treeOwner);\r\n    this._node = node;\r\n    this._branchId = formatAnimationBranchId(node.timeline.modelId, node.nodeId);\r\n  }\r\n\r\n  public override getAnimationTransformNodeId(): number {\r\n    return this._node.nodeId;\r\n  }\r\n\r\n  public override computeBaseTransform(tree: TileTree): Transform {\r\n    const tf = super.computeBaseTransform(tree);\r\n    const animTf = this._node.timeline.getTransform(this._node.nodeId, this._node.getCurrentTimePoint());\r\n    if (animTf)\r\n      animTf.multiplyTransformTransform(tf, tf);\r\n\r\n    return tf;\r\n  }\r\n\r\n  public override createDrawArgs(context: SceneContext): TileDrawArgs | undefined {\r\n    const animBranch = context.viewport.target.animationBranches?.branchStates.get(this._branchId);\r\n    if (animBranch && animBranch.omit)\r\n      return undefined;\r\n\r\n    const args = super.createDrawArgs(context);\r\n    // ###TODO args.boundingRange = args.tree.getTransformNodeRange(this._animationTransformNodeId);\r\n    return args;\r\n  }\r\n}\r\n"]}
+{"version":3,"file":"BatchedTileTreeReference.js","sourceRoot":"","sources":["../../src/BatchedTileTreeReference.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAK7C,OAAO,EACL,uBAAuB,EAAgD,eAAe,EAA2B,iBAAiB,GACnI,MAAM,sBAAsB,CAAC;AAW9B,MAAM,OAAO,wBAAyB,SAAQ,iBAAiB;IAM7D,YAAmB,IAAkC,EAAE,UAAkB,EAAE,eAAmC;QAC5G,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;QACxC,IAAI,SAAS,KAAK,eAAe,EAAE;YACjC,MAAM,CAAC,SAAS,KAAK,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC/C,IAAI,CAAC,SAAS,GAAG,uBAAuB,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;SAC7F;IACH,CAAC;IAED,IAAY,UAAU;QACpB,MAAM,CAAC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACpD,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC7C,CAAC;IAED,IAAoB,SAAS;QAC3B,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;IAC9B,CAAC;IAEe,qBAAqB;QACnC,OAAO,IAAI,CAAC;IACd,CAAC;IAEkB,aAAa;QAC9B,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;IAC9B,CAAC;IAEM,oBAAoB,CACzB,MAA+B,EAC/B,MAAc,EAAE,MAAc,EAC9B,QAAgB,EAAE,QAAgB,EAClC,SAAwB,EACxB,OAAe,EAAE,OAAe,EAChC,IAAe,EACf,eAAuB;QAEvB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;YAC/C,OAAO,SAAS,CAAC;QAEnB,OAAO,MAAM,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,eAAe,CAAC,CAAC;IACtH,CAAC;IAEe,aAAa,CAAC,KAAc;QAC1C,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC;IAED,IAAoB,YAAY;QAC9B,IAAI,IAAI,CAAC,UAAU,CAAC,cAAc;YAChC,OAAO,KAAK,CAAC;QAEf,OAAO,KAAK,CAAC,YAAY,CAAC;IAC5B,CAAC;IAEe,oBAAoB;QAClC,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC;IACnC,CAAC;IAEe,IAAI,CAAC,IAAkB;QACrC,IAAI,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,OAAO;YACzC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,eAAe,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;;YAElF,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrB,CAAC;IAEkB,gBAAgB,CAAC,IAAc;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC;QAC9B,IAAI,MAAM,GAAG,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAE1C,IAAI,KAAK,CAAC,cAAc,EAAE;YACxB,MAAM,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;YACxB,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,CAAC;SACnD;QAED,IAAI,KAAK,CAAC,QAAQ,EAAE;YAClB,MAAM,CAAC,SAAS,KAAK,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC5C,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,CAAC;YACpG,IAAI,MAAM;gBACR,MAAM,CAAC,0BAA0B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;SACrD;QAED,MAAM,SAAS,GAAG,KAAK,CAAC,gBAAgB,CAAC;QACzC,IAAI,CAAC,SAAS;YACZ,OAAO,MAAM,CAAC;QAEhB,OAAO,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,0BAA0B,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,0BAA0B,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACjJ,CAAC;IAEkB,2BAA2B;QAC5C,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC/B,CAAC;IAEkB,cAAc;QAC/B,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;IACrE,CAAC;IAEe,cAAc,CAAC,OAAqB;QAClD,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,EAAE,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3F,IAAI,MAAM,EAAE,IAAI,EAAE;gBAChB,0CAA0C;gBAC1C,OAAO,SAAS,CAAC;aAClB;SACF;QAED,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAE3C,gGAAgG;QAChG,yFAAyF;QAEzF,OAAO,IAAI,CAAC;IACd,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { assert } from \"@itwin/core-bentley\";\r\nimport { Range3d, Transform } from \"@itwin/core-geometry\";\r\nimport {\r\n  BatchType, FeatureAppearance, FeatureAppearanceProvider, FeatureAppearanceSource, GeometryClass, ViewFlagOverrides,\r\n} from \"@itwin/core-common\";\r\nimport {\r\n  formatAnimationBranchId, RenderClipVolume, SceneContext, TileDrawArgs, TileGraphicType, TileTree, TileTreeOwner, TileTreeReference,\r\n} from \"@itwin/core-frontend\";\r\nimport { BatchedModels } from \"./BatchedModels\";\r\nimport { ModelGroupInfo } from \"./ModelGroup\";\r\n\r\nexport interface BatchedTileTreeReferenceArgs {\r\n  readonly models: BatchedModels;\r\n  readonly groups: ReadonlyArray<ModelGroupInfo>;\r\n  readonly treeOwner: TileTreeOwner;\r\n  readonly getCurrentTimePoint: () => number;\r\n}\r\n\r\nexport class BatchedTileTreeReference extends TileTreeReference implements FeatureAppearanceProvider {\r\n  private readonly _args: BatchedTileTreeReferenceArgs;\r\n  private readonly _groupIndex: number;\r\n  private readonly _animationNodeId?: number;\r\n  private readonly _branchId?: string;\r\n\r\n  public constructor(args: BatchedTileTreeReferenceArgs, groupIndex: number, animationNodeId: number | undefined) {\r\n    super();\r\n    this._args = args;\r\n    this._groupIndex = groupIndex;\r\n    this._animationNodeId = animationNodeId;\r\n    if (undefined !== animationNodeId) {\r\n      assert(undefined !== this._groupInfo.timeline);\r\n      this._branchId = formatAnimationBranchId(this._groupInfo.timeline.modelId, animationNodeId);\r\n    }\r\n  }\r\n\r\n  private get _groupInfo(): ModelGroupInfo {\r\n    assert(this._groupIndex < this._args.groups.length);\r\n    return this._args.groups[this._groupIndex];\r\n  }\r\n\r\n  public override get treeOwner(): TileTreeOwner {\r\n    return this._args.treeOwner;\r\n  }\r\n\r\n  public override getAppearanceProvider(): FeatureAppearanceProvider {\r\n    return this;\r\n  }\r\n\r\n  protected override getClipVolume(): RenderClipVolume | undefined {\r\n    return this._groupInfo.clip;\r\n  }\r\n\r\n  public getFeatureAppearance(\r\n    source: FeatureAppearanceSource,\r\n    elemLo: number, elemHi: number,\r\n    subcatLo: number, subcatHi: number,\r\n    geomClass: GeometryClass,\r\n    modelLo: number, modelHi: number,\r\n    type: BatchType,\r\n    animationNodeId: number,\r\n  ): FeatureAppearance | undefined {\r\n    if (!this._args.models.isViewed(modelLo, modelHi))\r\n      return undefined;\r\n\r\n    return source.getAppearance(elemLo, elemHi, subcatLo, subcatHi, geomClass, modelLo, modelHi, type, animationNodeId);\r\n  }\r\n\r\n  public override unionFitRange(range: Range3d): void {\r\n    this._args.models.unionRange(range);\r\n  }\r\n\r\n  public override get castsShadows(): boolean {\r\n    if (this._groupInfo.planProjection)\r\n      return false;\r\n\r\n    return super.castsShadows;\r\n  }\r\n\r\n  public override getViewFlagOverrides(): ViewFlagOverrides {\r\n    return this._groupInfo.viewFlags;\r\n  }\r\n\r\n  public override draw(args: TileDrawArgs): void {\r\n    if (this._groupInfo.planProjection?.overlay)\r\n      args.context.withGraphicType(TileGraphicType.Overlay, () => args.tree.draw(args));\r\n    else\r\n      super.draw(args);\r\n  }\r\n\r\n  protected override computeTransform(tree: TileTree): Transform {\r\n    const group = this._groupInfo;\r\n    let baseTf = super.computeTransform(tree);\r\n\r\n    if (group.planProjection) {\r\n      baseTf = baseTf.clone();\r\n      baseTf.origin.z += group.planProjection.elevation;\r\n    }\r\n\r\n    if (group.timeline) {\r\n      assert(undefined !== this._animationNodeId);\r\n      const animTf = group.timeline.getTransform(this._animationNodeId, this._args.getCurrentTimePoint());\r\n      if (animTf)\r\n        animTf.multiplyTransformTransform(baseTf, baseTf);\r\n    }\r\n\r\n    const displayTf = group.displayTransform;\r\n    if (!displayTf)\r\n      return baseTf;\r\n\r\n    return displayTf.premultiply ? displayTf.transform.multiplyTransformTransform(baseTf) : baseTf.multiplyTransformTransform(displayTf.transform);\r\n  }\r\n\r\n  protected override getAnimationTransformNodeId() {\r\n    return this._animationNodeId;\r\n  }\r\n\r\n  protected override getGroupNodeId() {\r\n    return this._args.groups.length > 1 ? this._groupIndex : undefined;\r\n  }\r\n\r\n  public override createDrawArgs(context: SceneContext): TileDrawArgs | undefined {\r\n    if (this._branchId) {\r\n      const branch = context.viewport.target.animationBranches?.branchStates.get(this._branchId);\r\n      if (branch?.omit) {\r\n        // This branch is not supposed to be drawn\r\n        return undefined;\r\n      }\r\n    }\r\n\r\n    const args = super.createDrawArgs(context);\r\n\r\n    // ###TODO args.boundingRange = args.tree.getTransformNodeRange(this._animationTransformNodeId);\r\n    // ###TODO if PlanProjectionSettings.enforceDisplayPriority, createGraphicLayerContainer.\r\n\r\n    return args;\r\n  }\r\n}\r\n"]}

package/lib/esm/BatchedTileTreeSupplier.d.ts

@@ -5,6 +5,11 @@ import { BatchedTilesetSpec } from "./BatchedTilesetReader";
 export interface BatchedTileTreeId {
     spec: BatchedTilesetSpec;
     script?: RenderSchedule.Script;
+    /** A stringified representation of the [[ModelGroup]]s by which to structure the contents of the tiles.
+     * Every unique combination of model groups has a corresponding unique string representation.
+     * @see [[BatchedModelGroups.guid]].
+     */
+    modelGroups: string;
 }
 /** @internal */
 export declare function getBatchedTileTreeOwner(iModel: IModelConnection, treeId: BatchedTileTreeId): TileTreeOwner;

package/lib/esm/BatchedTileTreeSupplier.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"BatchedTileTreeSupplier.d.ts","sourceRoot":"","sources":["../../src/BatchedTileTreeSupplier.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EACL,gBAAgB,EAAY,aAAa,EAC1C,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAwB,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAGlF,gBAAgB;AAChB,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,kBAAkB,CAAC;IACzB,MAAM,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;CAChC;AAyBD,gBAAgB;AAChB,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,gBAAgB,EAAE,MAAM,EAAE,iBAAiB,GAAG,aAAa,CAE1G"}
+{"version":3,"file":"BatchedTileTreeSupplier.d.ts","sourceRoot":"","sources":["../../src/BatchedTileTreeSupplier.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EACL,gBAAgB,EAAY,aAAa,EAC1C,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAwB,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAGlF,gBAAgB;AAChB,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,kBAAkB,CAAC;IACzB,MAAM,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;IAC/B;;;OAGG;IACH,WAAW,EAAE,MAAM,CAAC;CACrB;AA2BD,gBAAgB;AAChB,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,gBAAgB,EAAE,MAAM,EAAE,iBAAiB,GAAG,aAAa,CAE1G"}

package/lib/esm/BatchedTileTreeSupplier.js

@@ -2,19 +2,21 @@
 * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
 * See LICENSE.md in the project root for license terms and full copyright notice.
 *--------------------------------------------------------------------------------------------*/
-import { comparePossiblyUndefined, compareStrings, Logger } from "@itwin/core-bentley";
+import { comparePossiblyUndefined, compareStrings, CompressedId64Set, Logger } from "@itwin/core-bentley";
 import { loggerCategory } from "./LoggerCategory";
 import { BatchedTilesetReader } from "./BatchedTilesetReader";
 import { BatchedTileTree } from "./BatchedTileTree";
 class BatchedTileTreeSupplier {
     compareTileTreeIds(lhs, rhs) {
-        return compareStrings(lhs.toString(), rhs.toString())
+        return compareStrings(lhs.spec.baseUrl.toString(), rhs.spec.baseUrl.toString())
+            || compareStrings(lhs.modelGroups, rhs.modelGroups)
             || comparePossiblyUndefined((x, y) => x.compareTo(y), lhs.script, rhs.script);
     }
     async createTileTree(treeId, iModel) {
         const spec = treeId.spec;
         try {
-            const reader = new BatchedTilesetReader(spec, iModel);
+            const modelGroups = treeId.modelGroups ? treeId.modelGroups.split("_").map((x) => CompressedId64Set.decompressSet(x)) : undefined;
+            const reader = new BatchedTilesetReader(spec, iModel, modelGroups);
             const params = await reader.readTileTreeParams();
             params.script = treeId.script;
             return new BatchedTileTree(params);

package/lib/esm/BatchedTileTreeSupplier.js.map

@@ -1 +1 @@
-{"version":3,"file":"BatchedTileTreeSupplier.js","sourceRoot":"","sources":["../../src/BatchedTileTreeSupplier.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,wBAAwB,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAKvF,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,oBAAoB,EAAsB,MAAM,wBAAwB,CAAC;AAClF,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAQpD,MAAM,uBAAuB;IACpB,kBAAkB,CAAC,GAAsB,EAAE,GAAsB;QACtE,OAAO,cAAc,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;eAChD,wBAAwB,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAClF,CAAC;IAEM,KAAK,CAAC,cAAc,CAAC,MAAyB,EAAE,MAAwB;QAC7E,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACzB,IAAI;YACF,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,kBAAkB,EAAE,CAAC;YAEjD,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YAC9B,OAAO,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;SACpC;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,CAAC,YAAY,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;YACzC,OAAO,SAAS,CAAC;SAClB;IACH,CAAC;CACF;AAED,MAAM,uBAAuB,GAAqB,IAAI,uBAAuB,EAAE,CAAC;AAEhF,gBAAgB;AAChB,MAAM,UAAU,uBAAuB,CAAC,MAAwB,EAAE,MAAyB;IACzF,OAAO,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,MAAM,EAAE,uBAAuB,CAAC,CAAC;AACxE,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { comparePossiblyUndefined, compareStrings, Logger } from \"@itwin/core-bentley\";\r\nimport { RenderSchedule } from \"@itwin/core-common\";\r\nimport {\r\n  IModelConnection, TileTree, TileTreeOwner, TileTreeSupplier,\r\n} from \"@itwin/core-frontend\";\r\nimport { loggerCategory } from \"./LoggerCategory\";\r\nimport { BatchedTilesetReader, BatchedTilesetSpec } from \"./BatchedTilesetReader\";\r\nimport { BatchedTileTree } from \"./BatchedTileTree\";\r\n\r\n/** @internal */\r\nexport interface BatchedTileTreeId {\r\n  spec: BatchedTilesetSpec;\r\n  script?: RenderSchedule.Script;\r\n}\r\n\r\nclass BatchedTileTreeSupplier implements TileTreeSupplier {\r\n  public compareTileTreeIds(lhs: BatchedTileTreeId, rhs: BatchedTileTreeId): number {\r\n    return compareStrings(lhs.toString(), rhs.toString())\r\n      || comparePossiblyUndefined((x, y) => x.compareTo(y), lhs.script, rhs.script);\r\n  }\r\n\r\n  public async createTileTree(treeId: BatchedTileTreeId, iModel: IModelConnection): Promise<TileTree | undefined> {\r\n    const spec = treeId.spec;\r\n    try {\r\n      const reader = new BatchedTilesetReader(spec, iModel);\r\n      const params = await reader.readTileTreeParams();\r\n\r\n      params.script = treeId.script;\r\n      return new BatchedTileTree(params);\r\n    } catch (err) {\r\n      Logger.logException(loggerCategory, err);\r\n      return undefined;\r\n    }\r\n  }\r\n}\r\n\r\nconst batchedTileTreeSupplier: TileTreeSupplier = new BatchedTileTreeSupplier();\r\n\r\n/** @internal */\r\nexport function getBatchedTileTreeOwner(iModel: IModelConnection, treeId: BatchedTileTreeId): TileTreeOwner {\r\n  return iModel.tiles.getTileTreeOwner(treeId, batchedTileTreeSupplier);\r\n}\r\n"]}
+{"version":3,"file":"BatchedTileTreeSupplier.js","sourceRoot":"","sources":["../../src/BatchedTileTreeSupplier.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,wBAAwB,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAK1G,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,oBAAoB,EAAsB,MAAM,wBAAwB,CAAC;AAClF,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAapD,MAAM,uBAAuB;IACpB,kBAAkB,CAAC,GAAsB,EAAE,GAAsB;QACtE,OAAO,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;eAC1E,cAAc,CAAC,GAAG,CAAC,WAAW,EAAE,GAAG,CAAC,WAAW,CAAC;eAChD,wBAAwB,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAClF,CAAC;IAEM,KAAK,CAAC,cAAc,CAAC,MAAyB,EAAE,MAAwB;QAC7E,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACzB,IAAI;YACF,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAClI,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YACnE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,kBAAkB,EAAE,CAAC;YAEjD,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YAC9B,OAAO,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;SACpC;QAAC,OAAO,GAAG,EAAE;YACZ,MAAM,CAAC,YAAY,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;YACzC,OAAO,SAAS,CAAC;SAClB;IACH,CAAC;CACF;AAED,MAAM,uBAAuB,GAAqB,IAAI,uBAAuB,EAAE,CAAC;AAEhF,gBAAgB;AAChB,MAAM,UAAU,uBAAuB,CAAC,MAAwB,EAAE,MAAyB;IACzF,OAAO,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,MAAM,EAAE,uBAAuB,CAAC,CAAC;AACxE,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { comparePossiblyUndefined, compareStrings, CompressedId64Set, Logger } from \"@itwin/core-bentley\";\r\nimport { RenderSchedule } from \"@itwin/core-common\";\r\nimport {\r\n  IModelConnection, TileTree, TileTreeOwner, TileTreeSupplier,\r\n} from \"@itwin/core-frontend\";\r\nimport { loggerCategory } from \"./LoggerCategory\";\r\nimport { BatchedTilesetReader, BatchedTilesetSpec } from \"./BatchedTilesetReader\";\r\nimport { BatchedTileTree } from \"./BatchedTileTree\";\r\n\r\n/** @internal */\r\nexport interface BatchedTileTreeId {\r\n  spec: BatchedTilesetSpec;\r\n  script?: RenderSchedule.Script;\r\n  /** A stringified representation of the [[ModelGroup]]s by which to structure the contents of the tiles.\r\n   * Every unique combination of model groups has a corresponding unique string representation.\r\n   * @see [[BatchedModelGroups.guid]].\r\n   */\r\n  modelGroups: string;\r\n}\r\n\r\nclass BatchedTileTreeSupplier implements TileTreeSupplier {\r\n  public compareTileTreeIds(lhs: BatchedTileTreeId, rhs: BatchedTileTreeId): number {\r\n    return compareStrings(lhs.spec.baseUrl.toString(), rhs.spec.baseUrl.toString())\r\n      || compareStrings(lhs.modelGroups, rhs.modelGroups)\r\n      || comparePossiblyUndefined((x, y) => x.compareTo(y), lhs.script, rhs.script);\r\n  }\r\n\r\n  public async createTileTree(treeId: BatchedTileTreeId, iModel: IModelConnection): Promise<TileTree | undefined> {\r\n    const spec = treeId.spec;\r\n    try {\r\n      const modelGroups = treeId.modelGroups ? treeId.modelGroups.split(\"_\").map((x) => CompressedId64Set.decompressSet(x)) : undefined;\r\n      const reader = new BatchedTilesetReader(spec, iModel, modelGroups);\r\n      const params = await reader.readTileTreeParams();\r\n\r\n      params.script = treeId.script;\r\n      return new BatchedTileTree(params);\r\n    } catch (err) {\r\n      Logger.logException(loggerCategory, err);\r\n      return undefined;\r\n    }\r\n  }\r\n}\r\n\r\nconst batchedTileTreeSupplier: TileTreeSupplier = new BatchedTileTreeSupplier();\r\n\r\n/** @internal */\r\nexport function getBatchedTileTreeOwner(iModel: IModelConnection, treeId: BatchedTileTreeId): TileTreeOwner {\r\n  return iModel.tiles.getTileTreeOwner(treeId, batchedTileTreeSupplier);\r\n}\r\n"]}