@itwin/tree-widget-react 3.6.0 → 3.6.1

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.js.map

@@ -1 +1 @@
-{"version":3,"file":"CategoriesTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,0BAA0B,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AA2B3F,gBAAgB;AAChB,MAAM,OAAO,sBAAsB;IAQjC,YACU,cAA0C,EAClD,QAAqB;QADb,mBAAc,GAAd,cAAc,CAA4B;QAGlD,MAAM,EAAE,aAAa,EAAE,oBAAoB,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC3E,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,qBAAqB,GAAG,oBAAoB,CAAC;IACpD,CAAC;IAEO,KAAK,CAAC,CAAC,eAAe;QAM5B,MAAM,8BAA8B,GAAG,MAAM,IAAI,CAAC,iCAAiC,EAAE,CAAC;QACtF,MAAM,eAAe,GAAG;;;;;UAMlB,8BAA8B;YAC5B,CAAC,CAAC;gEACkD,0BAA0B;;;cAG5E;YACF,CAAC,CAAC,OACN;;UAEE,IAAI,CAAC,cAAc;eACd,kBAAkB;;;;;oCAKG,IAAI,CAAC,qBAAqB;;KAEzD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAClJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,+BAA+B,EAAE,GAAG,CAAC,+BAA+B,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAC9I;IACH,CAAC;IAEO,KAAK,CAAC,oCAAoC;QAChD,MAAM,KAAK,GAAG;;;;;;;;;KASb,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE;YAChF,OAAO,IAAI,CAAC;SACb;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB;QACtC,iKAAiK;QACjK,2EAA2E;QAC3E,2GAA2G;QAC3G,kFAAkF;QAClF,MAAM,yBAAyB,GAAG,sBAAsB,CAAC;QACzD,MAAM,IAAI,GAAG;YACX;UACI,yBAAyB;;;;;cAKrB,0BAA0B;;yDAEiB,IAAI,CAAC,cAAc,sDAAsD,IAAI,CAAC,qBAAqB;;;;;;;;;cAS9I,yBAAyB;mBACpB,0BAA0B;;;;OAItC;SACF,CAAC;QACF,MAAM,gBAAgB,GAAG;2DAC8B,yBAAyB;KAC/E,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACzJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;SAC5C;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB,CAAC,cAAyB;QAChE,MAAM,gBAAgB,GAAG;;;;;UAKnB,kBAAkB;;;+BAGG,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC;KAClD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACnJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAChD;IACH,CAAC;IAEO,KAAK,CAAC,uBAAuB;QACnC,IAAI,CAAC,qBAAqB,KAAK,CAAC,KAAK,IAAI,EAAE;YACzC,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA8B,CAAC;YAClE,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,eAAe,EAAE,EAAE;gBAC1D,IAAI,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBACvE,IAAI,eAAe,KAAK,SAAS,EAAE;oBACjC,eAAe,GAAG,EAAE,+BAA+B,EAAE,eAAe,CAAC,+BAA+B,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;oBAC5H,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;iBACnE;gBACD,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,eAAe,CAAC,EAAE,EAAE,UAAU,EAAE,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC;aAC1G;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEO,KAAK,CAAC,oBAAoB;QAChC,IAAI,CAAC,kBAAkB,KAAK,CAAC,KAAK,IAAI,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAA+B,CAAC;YAChE,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClE,MAAM,oCAAoC,GAAG,IAAI,KAAK,EAAc,CAAC;YACrE,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE;gBAC/D,oCAAoC,CAAC,IAAI,CACvC,GAAG,mBAAmB,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC,CACpI,CAAC;aACH;YAED,IAAI,oCAAoC,CAAC,MAAM,KAAK,CAAC,EAAE;gBACrD,OAAO,gBAAgB,CAAC;aACzB;YAED,IAAI,KAAK,EAAE,MAAM,kBAAkB,IAAI,IAAI,CAAC,yBAAyB,CAAC,oCAAoC,CAAC,EAAE;gBAC3G,gBAAgB,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,kBAAkB,CAAC,QAAQ,EAAE,CAAC,CAAC;aAC1F;YACD,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,2BAA2B;QACvC,IAAI,CAAC,yBAAyB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC7C,MAAM,wBAAwB,GAAG,IAAI,GAAG,EAAuC,CAAC;YAChF,MAAM,CAAC,8BAA8B,EAAE,oBAAoB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC/E,IAAI,CAAC,iCAAiC,EAAE;gBACxC,IAAI,CAAC,uBAAuB,EAAE;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,8BAA8B,IAAI,oBAAoB,CAAC,IAAI,KAAK,CAAC,EAAE;gBACtE,OAAO,wBAAwB,CAAC;aACjC;YAED,IAAI,KAAK,EAAE,MAAM,0BAA0B,IAAI,IAAI,CAAC,yBAAyB,EAAE,EAAE;gBAC/E,MAAM,mBAAmB,GAAG,oBAAoB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC;gBAEpF,wBAAwB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,EAAE;oBAC1D,eAAe,EAAE,mBAAmB,EAAE,eAAe,IAAI,EAAE;oBAC3D,OAAO,EAAE,0BAA0B,CAAC,OAAO;oBAC3C,yBAAyB,EAAE,EAAE;oBAC7B,+BAA+B,EAAE,KAAK;iBACvC,CAAC,CAAC;aACJ;YAED,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,wBAAwB,EAAE;gBACvF,MAAM,yBAAyB,GAAG,wBAAwB,CAAC,GAAG,CAAC,uBAAuB,CAAC,OAAO,CAAC,CAAC;gBAChG,IAAI,yBAAyB,KAAK,SAAS,EAAE;oBAC3C,yBAAyB,CAAC,yBAAyB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;oBAChF,uBAAuB,CAAC,+BAA+B,GAAG,IAAI,CAAC;iBAChE;aACF;YAED,OAAO,wBAAwB,CAAC;QAClC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,yBAAyB,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,+CAA+C,CAC1D,4BAAuC;QAEvC,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAE1E,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QAExG,4BAA4B,CAAC,OAAO,CAAC,CAAC,2BAA2B,EAAE,EAAE;YACnE,MAAM,6BAA6B,GAAG,wBAAwB,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;YAChG,IAAI,6BAA6B,KAAK,SAAS,EAAE;gBAC/C,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,yBAAyB,CAAC,CAAC;gBAC7F,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,eAAe,CAAC,CAAC;aAC1E;QACH,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,sBAAiC;QACtE,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QAEvC,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC1E,MAAM,kBAAkB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC1C,sBAAsB,CAAC,GAAG,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE;YACzD,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;YACpF,IAAI,uBAAuB,KAAK,SAAS,EAAE;gBACzC,OAAO,EAAE,CAAC;aACX;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,uBAAuB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;YACvF,OAAO,IAAI,CAAC,yBAAyB,CAAC,uBAAuB,CAAC,yBAAyB,CAAC,CAAC;QAC3F,CAAC,CAAC,CACH,CAAC;QACF,KAAK,MAAM,UAAU,IAAI,kBAAkB,EAAE;YAC3C,MAAM,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;SAC5B;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAC9B,KAAyG;QAEzG,IAAI,eAAe,IAAI,KAAK,EAAE;YAC5B,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5D,MAAM,eAAe,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACnE,IAAI,eAAe,KAAK,SAAS,EAAE;gBACjC,OAAO,EAAE,CAAC;aACX;YACD,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC,CAAC;SACtJ;QAED,IAAI,YAAY,IAAI,KAAK,EAAE;YACzB,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClE,KAAK,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC,IAAI,oBAAoB,EAAE;gBACjE,IAAI,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,KAAK,CAAC,UAAU,CAAC,EAAE;oBACtG,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE;wBACxD,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;qBACnE;oBAED,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;iBAC5I;aACF;YACD,OAAO,EAAE,CAAC;SACX;QAED,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC1E,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC1F,IAAI,uBAAuB,KAAK,SAAS,EAAE;YACzC,OAAO,EAAE,CAAC;SACX;QAED,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE;YAC5D,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,0BAA0B,EAAE,CAAC,CAAC;SACrF;QAED,OAAO;YACL,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,uBAAuB,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/F,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,0BAA0B,EAAE;SAC3E,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,uCAAuC;QAClD,MAAM,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,CAAC,GAAG,wBAAwB,CAAC,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,KAAK,EAAc,EAAE,CAAC;QACnH,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE;YAC/D,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC;SACzG;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,wCAAwC;QACnD,MAAM,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QACxG,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE;YAC/D,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE;gBACxD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,CAAC,CAAC;aAChE;SACF;QAED,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,wBAAwB,EAAE;YACvF,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE;gBAC5D,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;aACzD;SACF;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,UAAsB;QAClD,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,KAAK,MAAM,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,iBAAiB,EAAE;YAChE,IAAI,eAAe,CAAC,UAAU,KAAK,UAAU,EAAE;gBAC7C,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;aAC5B;SACF;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,iCAAiC;QAC5C,IAAI,CAAC,+BAA+B,KAAK,IAAI,CAAC,oCAAoC,EAAE,CAAC;QACrF,OAAO,IAAI,CAAC,+BAA+B,CAAC;IAC9C,CAAC;CACF;AAED,gBAAgB;AAChB,MAAM,UAAU,gBAAgB,CAAC,QAAqB;IACpD,OAAO,QAAQ,KAAK,IAAI;QACtB,CAAC,CAAC,EAAE,aAAa,EAAE,yBAAyB,EAAE,oBAAoB,EAAE,4BAA4B,EAAE;QAClG,CAAC,CAAC,EAAE,aAAa,EAAE,yBAAyB,EAAE,oBAAoB,EAAE,4BAA4B,EAAE,CAAC;AACvG,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { DEFINITION_CONTAINER_CLASS, SUB_CATEGORY_CLASS } from \"./ClassNameDefinitions.js\";\n\nimport type { Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\n\ninterface DefinitionContainerInfo {\n  modelId: Id64String;\n  parentDefinitionContainerExists: boolean;\n  childCategories: CategoryInfo[];\n  childDefinitionContainers: Id64Array;\n}\n\ninterface CategoriesInfo {\n  childCategories: CategoryInfo[];\n  parentDefinitionContainerExists: boolean;\n}\n\ninterface CategoryInfo {\n  id: Id64String;\n  childCount: number;\n}\n\ninterface SubCategoryInfo {\n  categoryId: Id64String;\n}\n\n/** @internal */\nexport class CategoriesTreeIdsCache {\n  private _definitionContainersInfo: Promise<Map<Id64String, DefinitionContainerInfo>> | undefined;\n  private _modelsCategoriesInfo: Promise<Map<Id64String, CategoriesInfo>> | undefined;\n  private _subCategoriesInfo: Promise<Map<Id64String, SubCategoryInfo>> | undefined;\n  private _categoryClass: string;\n  private _categoryElementClass: string;\n  private _isDefinitionContainerSupported: Promise<boolean> | undefined;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    viewType: \"3d\" | \"2d\",\n  ) {\n    const { categoryClass, categoryElementClass } = getClassesByView(viewType);\n    this._categoryClass = categoryClass;\n    this._categoryElementClass = categoryElementClass;\n  }\n\n  private async *queryCategories(): AsyncIterableIterator<{\n    id: Id64String;\n    modelId: Id64String;\n    parentDefinitionContainerExists: boolean;\n    childCount: number;\n  }> {\n    const isDefinitionContainerSupported = await this.getIsDefinitionContainerSupported();\n    const categoriesQuery = `\n      SELECT\n        this.ECInstanceId id,\n        COUNT(sc.ECInstanceId) childCount,\n        this.Model.Id modelId,\n        ${\n          isDefinitionContainerSupported\n            ? `\n            IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${DEFINITION_CONTAINER_CLASS} dc),\n              true,\n              false\n            )`\n            : \"false\"\n        } parentDefinitionContainerExists\n      FROM\n        ${this._categoryClass} this\n        JOIN ${SUB_CATEGORY_CLASS} sc ON sc.Parent.Id = this.ECInstanceId\n        JOIN BisCore.Model m ON m.ECInstanceId = this.Model.Id\n      WHERE\n        NOT this.IsPrivate\n        AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))\n        AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)\n      GROUP BY this.ECInstanceId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: categoriesQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, modelId: row.modelId, parentDefinitionContainerExists: row.parentDefinitionContainerExists, childCount: row.childCount };\n    }\n  }\n\n  private async queryIsDefinitionContainersSupported(): Promise<boolean> {\n    const query = `\n      SELECT\n        1\n      FROM\n        ECDbMeta.ECSchemaDef s\n        JOIN ECDbMeta.ECClassDef c ON c.Schema.Id = s.ECInstanceId\n      WHERE\n        s.Name = 'BisCore'\n        AND c.Name = 'DefinitionContainer'\n    `;\n\n    for await (const _row of this._queryExecutor.createQueryReader({ ecsql: query })) {\n      return true;\n    }\n    return false;\n  }\n\n  private async *queryDefinitionContainers(): AsyncIterableIterator<{ id: Id64String; modelId: Id64String }> {\n    // DefinitionModel ECInstanceId will always be the same as modeled DefinitionContainer ECInstanceId, if this wasn't the case, we would need to do something like:\n    //  JOIN BisCore.DefinitionModel dm ON dm.ECInstanceId = ${modelIdAccessor}\n    //  JOIN BisCore.DefinitionModelBreaksDownDefinitionContainer dr ON dr.SourceECInstanceId = dm.ECInstanceId\n    //  JOIN BisCore.DefinitionContainer dc ON dc.ECInstanceId = dr.TargetECInstanceId\n    const DEFINITION_CONTAINERS_CTE = \"DefinitionContainers\";\n    const ctes = [\n      `\n        ${DEFINITION_CONTAINERS_CTE}(ECInstanceId, ModelId) AS (\n          SELECT\n            dc.ECInstanceId,\n            dc.Model.Id\n          FROM\n            ${DEFINITION_CONTAINER_CLASS} dc\n          WHERE\n            dc.ECInstanceId IN (SELECT c.Model.Id FROM ${this._categoryClass} c WHERE NOT c.IsPrivate AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = c.ECInstanceId))\n            AND NOT dc.IsPrivate\n\n          UNION ALL\n\n          SELECT\n            pdc.ECInstanceId,\n            pdc.Model.Id\n          FROM\n            ${DEFINITION_CONTAINERS_CTE} cdc\n            JOIN ${DEFINITION_CONTAINER_CLASS} pdc ON pdc.ECInstanceId = cdc.ModelId\n          WHERE\n            NOT pdc.IsPrivate\n        )\n      `,\n    ];\n    const definitionsQuery = `\n      SELECT dc.ECInstanceId id, dc.ModelId modelId FROM ${DEFINITION_CONTAINERS_CTE} dc GROUP BY dc.ECInstanceId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ctes, ecsql: definitionsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, modelId: row.modelId };\n    }\n  }\n\n  private async *queryVisibleSubCategories(categoriesInfo: Id64Array): AsyncIterableIterator<{ id: Id64String; parentId: Id64String }> {\n    const definitionsQuery = `\n      SELECT\n        sc.ECInstanceId id,\n        sc.Parent.Id categoryId\n      FROM\n        ${SUB_CATEGORY_CLASS} sc\n      WHERE\n        NOT sc.IsPrivate\n        AND sc.Parent.Id IN (${categoriesInfo.join(\",\")})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: definitionsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.categoryId };\n    }\n  }\n\n  private async getModelsCategoriesInfo() {\n    this._modelsCategoriesInfo ??= (async () => {\n      const allModelsCategories = new Map<Id64String, CategoriesInfo>();\n      for await (const queriedCategory of this.queryCategories()) {\n        let modelCategories = allModelsCategories.get(queriedCategory.modelId);\n        if (modelCategories === undefined) {\n          modelCategories = { parentDefinitionContainerExists: queriedCategory.parentDefinitionContainerExists, childCategories: [] };\n          allModelsCategories.set(queriedCategory.modelId, modelCategories);\n        }\n        modelCategories.childCategories.push({ id: queriedCategory.id, childCount: queriedCategory.childCount });\n      }\n      return allModelsCategories;\n    })();\n    return this._modelsCategoriesInfo;\n  }\n\n  private async getSubCategoriesInfo() {\n    this._subCategoriesInfo ??= (async () => {\n      const allSubCategories = new Map<Id64String, SubCategoryInfo>();\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      const categoriesWithMoreThanOneSubCategory = new Array<Id64String>();\n      for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n        categoriesWithMoreThanOneSubCategory.push(\n          ...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.childCount > 1).map((categoryInfo) => categoryInfo.id),\n        );\n      }\n\n      if (categoriesWithMoreThanOneSubCategory.length === 0) {\n        return allSubCategories;\n      }\n\n      for await (const queriedSubCategory of this.queryVisibleSubCategories(categoriesWithMoreThanOneSubCategory)) {\n        allSubCategories.set(queriedSubCategory.id, { categoryId: queriedSubCategory.parentId });\n      }\n      return allSubCategories;\n    })();\n    return this._subCategoriesInfo;\n  }\n\n  private async getDefinitionContainersInfo() {\n    this._definitionContainersInfo ??= (async () => {\n      const definitionContainersInfo = new Map<Id64String, DefinitionContainerInfo>();\n      const [isDefinitionContainerSupported, modelsCategoriesInfo] = await Promise.all([\n        this.getIsDefinitionContainerSupported(),\n        this.getModelsCategoriesInfo(),\n      ]);\n      if (!isDefinitionContainerSupported || modelsCategoriesInfo.size === 0) {\n        return definitionContainersInfo;\n      }\n\n      for await (const queriedDefinitionContainer of this.queryDefinitionContainers()) {\n        const modelCategoriesInfo = modelsCategoriesInfo.get(queriedDefinitionContainer.id);\n\n        definitionContainersInfo.set(queriedDefinitionContainer.id, {\n          childCategories: modelCategoriesInfo?.childCategories ?? [],\n          modelId: queriedDefinitionContainer.modelId,\n          childDefinitionContainers: [],\n          parentDefinitionContainerExists: false,\n        });\n      }\n\n      for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {\n        const parentDefinitionContainer = definitionContainersInfo.get(definitionContainerInfo.modelId);\n        if (parentDefinitionContainer !== undefined) {\n          parentDefinitionContainer.childDefinitionContainers.push(definitionContainerId);\n          definitionContainerInfo.parentDefinitionContainerExists = true;\n        }\n      }\n\n      return definitionContainersInfo;\n    })();\n    return this._definitionContainersInfo;\n  }\n\n  public async getDirectChildDefinitionContainersAndCategories(\n    parentDefinitionContainerIds: Id64Array,\n  ): Promise<{ categories: CategoryInfo[]; definitionContainers: Id64Array }> {\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n\n    const result = { definitionContainers: new Array<Id64String>(), categories: new Array<CategoryInfo>() };\n\n    parentDefinitionContainerIds.forEach((parentDefinitionContainerId) => {\n      const parentDefinitionContainerInfo = definitionContainersInfo.get(parentDefinitionContainerId);\n      if (parentDefinitionContainerInfo !== undefined) {\n        result.definitionContainers.push(...parentDefinitionContainerInfo.childDefinitionContainers);\n        result.categories.push(...parentDefinitionContainerInfo.childCategories);\n      }\n    });\n    return result;\n  }\n\n  public async getAllContainedCategories(definitionContainerIds: Id64Array): Promise<Id64Array> {\n    const result = new Array<Id64String>();\n\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n    const indirectCategories = await Promise.all(\n      definitionContainerIds.map(async (definitionContainerId) => {\n        const definitionContainerInfo = definitionContainersInfo.get(definitionContainerId);\n        if (definitionContainerInfo === undefined) {\n          return [];\n        }\n        result.push(...definitionContainerInfo.childCategories.map((category) => category.id));\n        return this.getAllContainedCategories(definitionContainerInfo.childDefinitionContainers);\n      }),\n    );\n    for (const categories of indirectCategories) {\n      result.push(...categories);\n    }\n\n    return result;\n  }\n\n  public async getInstanceKeyPaths(\n    props: { categoryId: Id64String } | { definitionContainerId: Id64String } | { subCategoryId: Id64String },\n  ): Promise<InstanceKey[]> {\n    if (\"subCategoryId\" in props) {\n      const subCategoriesInfo = await this.getSubCategoriesInfo();\n      const subCategoryInfo = subCategoriesInfo.get(props.subCategoryId);\n      if (subCategoryInfo === undefined) {\n        return [];\n      }\n      return [...(await this.getInstanceKeyPaths({ categoryId: subCategoryInfo.categoryId })), { id: props.subCategoryId, className: SUB_CATEGORY_CLASS }];\n    }\n\n    if (\"categoryId\" in props) {\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      for (const [modelId, modelCategoriesInfo] of modelsCategoriesInfo) {\n        if (modelCategoriesInfo.childCategories.find((childCategory) => childCategory.id === props.categoryId)) {\n          if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n            return [{ id: props.categoryId, className: this._categoryClass }];\n          }\n\n          return [...(await this.getInstanceKeyPaths({ definitionContainerId: modelId })), { id: props.categoryId, className: this._categoryClass }];\n        }\n      }\n      return [];\n    }\n\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n    const definitionContainerInfo = definitionContainersInfo.get(props.definitionContainerId);\n    if (definitionContainerInfo === undefined) {\n      return [];\n    }\n\n    if (!definitionContainerInfo.parentDefinitionContainerExists) {\n      return [{ id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS }];\n    }\n\n    return [\n      ...(await this.getInstanceKeyPaths({ definitionContainerId: definitionContainerInfo.modelId })),\n      { id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS },\n    ];\n  }\n\n  public async getAllDefinitionContainersAndCategories(): Promise<{ categories: Id64Array; definitionContainers: Id64Array }> {\n    const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);\n    const result = { definitionContainers: [...definitionContainersInfo.keys()], categories: new Array<Id64String>() };\n    for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n      result.categories.push(...modelCategoriesInfo.childCategories.map((childCategory) => childCategory.id));\n    }\n\n    return result;\n  }\n\n  public async getRootDefinitionContainersAndCategories(): Promise<{ categories: CategoryInfo[]; definitionContainers: Id64Array }> {\n    const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);\n    const result = { definitionContainers: new Array<Id64String>(), categories: new Array<CategoryInfo>() };\n    for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n      if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n        result.categories.push(...modelCategoriesInfo.childCategories);\n      }\n    }\n\n    for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {\n      if (!definitionContainerInfo.parentDefinitionContainerExists) {\n        result.definitionContainers.push(definitionContainerId);\n      }\n    }\n    return result;\n  }\n\n  public async getSubCategories(categoryId: Id64String): Promise<Id64Array> {\n    const subCategoriesInfo = await this.getSubCategoriesInfo();\n    const result = new Array<Id64String>();\n    for (const [subCategoryId, subCategoryInfo] of subCategoriesInfo) {\n      if (subCategoryInfo.categoryId === categoryId) {\n        result.push(subCategoryId);\n      }\n    }\n    return result;\n  }\n\n  public async getIsDefinitionContainerSupported(): Promise<boolean> {\n    this._isDefinitionContainerSupported ??= this.queryIsDefinitionContainersSupported();\n    return this._isDefinitionContainerSupported;\n  }\n}\n\n/** @internal */\nexport function getClassesByView(viewType: \"2d\" | \"3d\") {\n  return viewType === \"2d\"\n    ? { categoryClass: \"BisCore.DrawingCategory\", categoryElementClass: \"BisCore.GeometricElement2d\" }\n    : { categoryClass: \"BisCore.SpatialCategory\", categoryElementClass: \"BisCore.GeometricElement3d\" };\n}\n"]}
+{"version":3,"file":"CategoriesTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,0BAA0B,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AA2B3F,gBAAgB;AAChB,MAAM,OAAO,sBAAsB;IAUjC,YACU,cAA0C,EAClD,QAAqB;QADb,mBAAc,GAAd,cAAc,CAA4B;QAGlD,MAAM,EAAE,aAAa,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC/F,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,qBAAqB,GAAG,oBAAoB,CAAC;QAClD,IAAI,CAAC,mBAAmB,GAAG,kBAAkB,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,2BAA2B;QAIxC,MAAM,KAAK,GAAG;;aAEL,IAAI,CAAC,mBAAmB;aACxB,IAAI,CAAC,qBAAqB;;;KAGlC,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,iEAAiE,EAAE,CACzI,EAAE;YACD,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAC5D;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,eAAe;QAM5B,MAAM,8BAA8B,GAAG,MAAM,IAAI,CAAC,iCAAiC,EAAE,CAAC;QACtF,MAAM,eAAe,GAAG;;;;;UAMlB,8BAA8B;YAC5B,CAAC,CAAC;gEACkD,0BAA0B;;;cAG5E;YACF,CAAC,CAAC,OACN;;UAEE,IAAI,CAAC,cAAc;eACd,kBAAkB;;;;;oCAKG,IAAI,CAAC,qBAAqB;;KAEzD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAClJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,+BAA+B,EAAE,GAAG,CAAC,+BAA+B,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAC9I;IACH,CAAC;IAEO,KAAK,CAAC,oCAAoC;QAChD,MAAM,KAAK,GAAG;;;;;;;;;KASb,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE;YAChF,OAAO,IAAI,CAAC;SACb;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB;QACtC,iKAAiK;QACjK,2EAA2E;QAC3E,2GAA2G;QAC3G,kFAAkF;QAClF,MAAM,yBAAyB,GAAG,sBAAsB,CAAC;QACzD,MAAM,IAAI,GAAG;YACX;UACI,yBAAyB;;;;;cAKrB,0BAA0B;;yDAEiB,IAAI,CAAC,cAAc,sDAAsD,IAAI,CAAC,qBAAqB;;;;;;;;;cAS9I,yBAAyB;mBACpB,0BAA0B;;;;OAItC;SACF,CAAC;QACF,MAAM,gBAAgB,GAAG;2DAC8B,yBAAyB;KAC/E,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACzJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;SAC5C;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB,CAAC,cAAyB;QAChE,MAAM,gBAAgB,GAAG;;;;;UAKnB,kBAAkB;;;+BAGG,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC;KAClD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACnJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAChD;IACH,CAAC;IAEO,KAAK,CAAC,uBAAuB;QACnC,IAAI,CAAC,qBAAqB,KAAK,CAAC,KAAK,IAAI,EAAE;YACzC,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA8B,CAAC;YAClE,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,eAAe,EAAE,EAAE;gBAC1D,IAAI,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBACvE,IAAI,eAAe,KAAK,SAAS,EAAE;oBACjC,eAAe,GAAG,EAAE,+BAA+B,EAAE,eAAe,CAAC,+BAA+B,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;oBAC5H,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;iBACnE;gBACD,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,eAAe,CAAC,EAAE,EAAE,UAAU,EAAE,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC;aAC1G;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEO,KAAK,CAAC,0BAA0B;QACtC,IAAI,CAAC,wBAAwB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC5C,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAAuB,CAAC;YAC/D,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,2BAA2B,EAAE,EAAE;gBACtE,IAAI,UAAU,GAAG,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBACtE,IAAI,UAAU,KAAK,SAAS,EAAE;oBAC5B,UAAU,GAAG,IAAI,GAAG,EAAE,CAAC;oBACvB,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;iBAClE;gBACD,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;aAC5C;YACD,OAAO,uBAAuB,CAAC;QACjC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,wBAAwB,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,oBAAoB;QAChC,IAAI,CAAC,kBAAkB,KAAK,CAAC,KAAK,IAAI,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAA+B,CAAC;YAChE,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClE,MAAM,oCAAoC,GAAG,IAAI,KAAK,EAAc,CAAC;YACrE,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE;gBAC/D,oCAAoC,CAAC,IAAI,CACvC,GAAG,mBAAmB,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC,CACpI,CAAC;aACH;YAED,IAAI,oCAAoC,CAAC,MAAM,KAAK,CAAC,EAAE;gBACrD,OAAO,gBAAgB,CAAC;aACzB;YAED,IAAI,KAAK,EAAE,MAAM,kBAAkB,IAAI,IAAI,CAAC,yBAAyB,CAAC,oCAAoC,CAAC,EAAE;gBAC3G,gBAAgB,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,kBAAkB,CAAC,QAAQ,EAAE,CAAC,CAAC;aAC1F;YACD,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,2BAA2B;QACvC,IAAI,CAAC,yBAAyB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC7C,MAAM,wBAAwB,GAAG,IAAI,GAAG,EAAuC,CAAC;YAChF,MAAM,CAAC,8BAA8B,EAAE,oBAAoB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC/E,IAAI,CAAC,iCAAiC,EAAE;gBACxC,IAAI,CAAC,uBAAuB,EAAE;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,8BAA8B,IAAI,oBAAoB,CAAC,IAAI,KAAK,CAAC,EAAE;gBACtE,OAAO,wBAAwB,CAAC;aACjC;YAED,IAAI,KAAK,EAAE,MAAM,0BAA0B,IAAI,IAAI,CAAC,yBAAyB,EAAE,EAAE;gBAC/E,MAAM,mBAAmB,GAAG,oBAAoB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC;gBAEpF,wBAAwB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,EAAE;oBAC1D,eAAe,EAAE,mBAAmB,EAAE,eAAe,IAAI,EAAE;oBAC3D,OAAO,EAAE,0BAA0B,CAAC,OAAO;oBAC3C,yBAAyB,EAAE,EAAE;oBAC7B,+BAA+B,EAAE,KAAK;iBACvC,CAAC,CAAC;aACJ;YAED,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,wBAAwB,EAAE;gBACvF,MAAM,yBAAyB,GAAG,wBAAwB,CAAC,GAAG,CAAC,uBAAuB,CAAC,OAAO,CAAC,CAAC;gBAChG,IAAI,yBAAyB,KAAK,SAAS,EAAE;oBAC3C,yBAAyB,CAAC,yBAAyB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;oBAChF,uBAAuB,CAAC,+BAA+B,GAAG,IAAI,CAAC;iBAChE;aACF;YAED,OAAO,wBAAwB,CAAC;QAClC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,yBAAyB,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,+CAA+C,CAC1D,4BAAuC;QAEvC,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAE1E,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QAExG,4BAA4B,CAAC,OAAO,CAAC,CAAC,2BAA2B,EAAE,EAAE;YACnE,MAAM,6BAA6B,GAAG,wBAAwB,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;YAChG,IAAI,6BAA6B,KAAK,SAAS,EAAE;gBAC/C,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,yBAAyB,CAAC,CAAC;gBAC7F,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,eAAe,CAAC,CAAC;aAC1E;QACH,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,0BAA0B,CAAC,WAAsB;QAC5D,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,MAAM,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;QAChD,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE;YACpC,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,uBAAuB,EAAE;gBAC3D,IAAI,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;oBAC9B,IAAI,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC5C,IAAI,CAAC,cAAc,EAAE;wBACnB,cAAc,GAAG,IAAI,KAAK,EAAc,CAAC;wBACzC,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;qBACxC;oBACD,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;iBAC9B;aACF;SACF;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,sBAAiC;QACtE,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QAEvC,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC1E,MAAM,kBAAkB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC1C,sBAAsB,CAAC,GAAG,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE;YACzD,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;YACpF,IAAI,uBAAuB,KAAK,SAAS,EAAE;gBACzC,OAAO,EAAE,CAAC;aACX;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,uBAAuB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;YACvF,OAAO,IAAI,CAAC,yBAAyB,CAAC,uBAAuB,CAAC,yBAAyB,CAAC,CAAC;QAC3F,CAAC,CAAC,CACH,CAAC;QACF,KAAK,MAAM,UAAU,IAAI,kBAAkB,EAAE;YAC3C,MAAM,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;SAC5B;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAC9B,KAAyG;QAEzG,IAAI,eAAe,IAAI,KAAK,EAAE;YAC5B,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5D,MAAM,eAAe,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACnE,IAAI,eAAe,KAAK,SAAS,EAAE;gBACjC,OAAO,EAAE,CAAC;aACX;YACD,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC,CAAC;SACtJ;QAED,IAAI,YAAY,IAAI,KAAK,EAAE;YACzB,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClE,KAAK,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC,IAAI,oBAAoB,EAAE;gBACjE,IAAI,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,KAAK,CAAC,UAAU,CAAC,EAAE;oBACtG,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE;wBACxD,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;qBACnE;oBAED,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;iBAC5I;aACF;YACD,OAAO,EAAE,CAAC;SACX;QAED,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC1E,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC1F,IAAI,uBAAuB,KAAK,SAAS,EAAE;YACzC,OAAO,EAAE,CAAC;SACX;QAED,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE;YAC5D,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,0BAA0B,EAAE,CAAC,CAAC;SACrF;QAED,OAAO;YACL,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,uBAAuB,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/F,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,0BAA0B,EAAE;SAC3E,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,uCAAuC;QAClD,MAAM,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,CAAC,GAAG,wBAAwB,CAAC,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,KAAK,EAAc,EAAE,CAAC;QACnH,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE;YAC/D,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC;SACzG;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,wCAAwC;QACnD,MAAM,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QACxG,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE;YAC/D,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE;gBACxD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,CAAC,CAAC;aAChE;SACF;QAED,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,wBAAwB,EAAE;YACvF,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE;gBAC5D,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;aACzD;SACF;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,UAAsB;QAClD,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,KAAK,MAAM,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,iBAAiB,EAAE;YAChE,IAAI,eAAe,CAAC,UAAU,KAAK,UAAU,EAAE;gBAC7C,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;aAC5B;SACF;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,iCAAiC;QAC5C,IAAI,CAAC,+BAA+B,KAAK,IAAI,CAAC,oCAAoC,EAAE,CAAC;QACrF,OAAO,IAAI,CAAC,+BAA+B,CAAC;IAC9C,CAAC;CACF;AAED,gBAAgB;AAChB,MAAM,UAAU,gBAAgB,CAAC,QAAqB;IACpD,OAAO,QAAQ,KAAK,IAAI;QACtB,CAAC,CAAC,EAAE,aAAa,EAAE,yBAAyB,EAAE,oBAAoB,EAAE,4BAA4B,EAAE,kBAAkB,EAAE,0BAA0B,EAAE;QAClJ,CAAC,CAAC,EAAE,aAAa,EAAE,yBAAyB,EAAE,oBAAoB,EAAE,4BAA4B,EAAE,kBAAkB,EAAE,0BAA0B,EAAE,CAAC;AACvJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { DEFINITION_CONTAINER_CLASS, SUB_CATEGORY_CLASS } from \"./ClassNameDefinitions.js\";\n\nimport type { Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\n\ninterface DefinitionContainerInfo {\n  modelId: Id64String;\n  parentDefinitionContainerExists: boolean;\n  childCategories: CategoryInfo[];\n  childDefinitionContainers: Id64Array;\n}\n\ninterface CategoriesInfo {\n  childCategories: CategoryInfo[];\n  parentDefinitionContainerExists: boolean;\n}\n\ninterface CategoryInfo {\n  id: Id64String;\n  childCount: number;\n}\n\ninterface SubCategoryInfo {\n  categoryId: Id64String;\n}\n\n/** @internal */\nexport class CategoriesTreeIdsCache {\n  private _definitionContainersInfo: Promise<Map<Id64String, DefinitionContainerInfo>> | undefined;\n  private _modelsCategoriesInfo: Promise<Map<Id64String, CategoriesInfo>> | undefined;\n  private _subCategoriesInfo: Promise<Map<Id64String, SubCategoryInfo>> | undefined;\n  private _elementModelsCategories: Promise<Map<Id64String, Id64Set>> | undefined;\n  private _categoryClass: string;\n  private _categoryElementClass: string;\n  private _categoryModelClass: string;\n  private _isDefinitionContainerSupported: Promise<boolean> | undefined;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    viewType: \"3d\" | \"2d\",\n  ) {\n    const { categoryClass, categoryElementClass, categoryModelClass } = getClassesByView(viewType);\n    this._categoryClass = categoryClass;\n    this._categoryElementClass = categoryElementClass;\n    this._categoryModelClass = categoryModelClass;\n  }\n\n  private async *queryElementModelCategories(): AsyncIterableIterator<{\n    modelId: Id64String;\n    categoryId: Id64String;\n  }> {\n    const query = `\n      SELECT this.Model.Id modelId, this.Category.Id categoryId\n      FROM ${this._categoryModelClass} m\n      JOIN ${this._categoryElementClass} this ON m.ECInstanceId = this.Model.Id\n      WHERE m.IsPrivate = false\n      GROUP BY modelId, categoryId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/element-models-and-categories-query\" },\n    )) {\n      yield { modelId: row.modelId, categoryId: row.categoryId };\n    }\n  }\n\n  private async *queryCategories(): AsyncIterableIterator<{\n    id: Id64String;\n    modelId: Id64String;\n    parentDefinitionContainerExists: boolean;\n    childCount: number;\n  }> {\n    const isDefinitionContainerSupported = await this.getIsDefinitionContainerSupported();\n    const categoriesQuery = `\n      SELECT\n        this.ECInstanceId id,\n        COUNT(sc.ECInstanceId) childCount,\n        this.Model.Id modelId,\n        ${\n          isDefinitionContainerSupported\n            ? `\n            IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${DEFINITION_CONTAINER_CLASS} dc),\n              true,\n              false\n            )`\n            : \"false\"\n        } parentDefinitionContainerExists\n      FROM\n        ${this._categoryClass} this\n        JOIN ${SUB_CATEGORY_CLASS} sc ON sc.Parent.Id = this.ECInstanceId\n        JOIN BisCore.Model m ON m.ECInstanceId = this.Model.Id\n      WHERE\n        NOT this.IsPrivate\n        AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))\n        AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)\n      GROUP BY this.ECInstanceId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: categoriesQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, modelId: row.modelId, parentDefinitionContainerExists: row.parentDefinitionContainerExists, childCount: row.childCount };\n    }\n  }\n\n  private async queryIsDefinitionContainersSupported(): Promise<boolean> {\n    const query = `\n      SELECT\n        1\n      FROM\n        ECDbMeta.ECSchemaDef s\n        JOIN ECDbMeta.ECClassDef c ON c.Schema.Id = s.ECInstanceId\n      WHERE\n        s.Name = 'BisCore'\n        AND c.Name = 'DefinitionContainer'\n    `;\n\n    for await (const _row of this._queryExecutor.createQueryReader({ ecsql: query })) {\n      return true;\n    }\n    return false;\n  }\n\n  private async *queryDefinitionContainers(): AsyncIterableIterator<{ id: Id64String; modelId: Id64String }> {\n    // DefinitionModel ECInstanceId will always be the same as modeled DefinitionContainer ECInstanceId, if this wasn't the case, we would need to do something like:\n    //  JOIN BisCore.DefinitionModel dm ON dm.ECInstanceId = ${modelIdAccessor}\n    //  JOIN BisCore.DefinitionModelBreaksDownDefinitionContainer dr ON dr.SourceECInstanceId = dm.ECInstanceId\n    //  JOIN BisCore.DefinitionContainer dc ON dc.ECInstanceId = dr.TargetECInstanceId\n    const DEFINITION_CONTAINERS_CTE = \"DefinitionContainers\";\n    const ctes = [\n      `\n        ${DEFINITION_CONTAINERS_CTE}(ECInstanceId, ModelId) AS (\n          SELECT\n            dc.ECInstanceId,\n            dc.Model.Id\n          FROM\n            ${DEFINITION_CONTAINER_CLASS} dc\n          WHERE\n            dc.ECInstanceId IN (SELECT c.Model.Id FROM ${this._categoryClass} c WHERE NOT c.IsPrivate AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = c.ECInstanceId))\n            AND NOT dc.IsPrivate\n\n          UNION ALL\n\n          SELECT\n            pdc.ECInstanceId,\n            pdc.Model.Id\n          FROM\n            ${DEFINITION_CONTAINERS_CTE} cdc\n            JOIN ${DEFINITION_CONTAINER_CLASS} pdc ON pdc.ECInstanceId = cdc.ModelId\n          WHERE\n            NOT pdc.IsPrivate\n        )\n      `,\n    ];\n    const definitionsQuery = `\n      SELECT dc.ECInstanceId id, dc.ModelId modelId FROM ${DEFINITION_CONTAINERS_CTE} dc GROUP BY dc.ECInstanceId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ctes, ecsql: definitionsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, modelId: row.modelId };\n    }\n  }\n\n  private async *queryVisibleSubCategories(categoriesInfo: Id64Array): AsyncIterableIterator<{ id: Id64String; parentId: Id64String }> {\n    const definitionsQuery = `\n      SELECT\n        sc.ECInstanceId id,\n        sc.Parent.Id categoryId\n      FROM\n        ${SUB_CATEGORY_CLASS} sc\n      WHERE\n        NOT sc.IsPrivate\n        AND sc.Parent.Id IN (${categoriesInfo.join(\",\")})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: definitionsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.categoryId };\n    }\n  }\n\n  private async getModelsCategoriesInfo() {\n    this._modelsCategoriesInfo ??= (async () => {\n      const allModelsCategories = new Map<Id64String, CategoriesInfo>();\n      for await (const queriedCategory of this.queryCategories()) {\n        let modelCategories = allModelsCategories.get(queriedCategory.modelId);\n        if (modelCategories === undefined) {\n          modelCategories = { parentDefinitionContainerExists: queriedCategory.parentDefinitionContainerExists, childCategories: [] };\n          allModelsCategories.set(queriedCategory.modelId, modelCategories);\n        }\n        modelCategories.childCategories.push({ id: queriedCategory.id, childCount: queriedCategory.childCount });\n      }\n      return allModelsCategories;\n    })();\n    return this._modelsCategoriesInfo;\n  }\n\n  private async getElementModelsCategories() {\n    this._elementModelsCategories ??= (async () => {\n      const elementModelsCategories = new Map<Id64String, Id64Set>();\n      for await (const queriedCategory of this.queryElementModelCategories()) {\n        let modelEntry = elementModelsCategories.get(queriedCategory.modelId);\n        if (modelEntry === undefined) {\n          modelEntry = new Set();\n          elementModelsCategories.set(queriedCategory.modelId, modelEntry);\n        }\n        modelEntry.add(queriedCategory.categoryId);\n      }\n      return elementModelsCategories;\n    })();\n    return this._elementModelsCategories;\n  }\n\n  private async getSubCategoriesInfo() {\n    this._subCategoriesInfo ??= (async () => {\n      const allSubCategories = new Map<Id64String, SubCategoryInfo>();\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      const categoriesWithMoreThanOneSubCategory = new Array<Id64String>();\n      for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n        categoriesWithMoreThanOneSubCategory.push(\n          ...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.childCount > 1).map((categoryInfo) => categoryInfo.id),\n        );\n      }\n\n      if (categoriesWithMoreThanOneSubCategory.length === 0) {\n        return allSubCategories;\n      }\n\n      for await (const queriedSubCategory of this.queryVisibleSubCategories(categoriesWithMoreThanOneSubCategory)) {\n        allSubCategories.set(queriedSubCategory.id, { categoryId: queriedSubCategory.parentId });\n      }\n      return allSubCategories;\n    })();\n    return this._subCategoriesInfo;\n  }\n\n  private async getDefinitionContainersInfo() {\n    this._definitionContainersInfo ??= (async () => {\n      const definitionContainersInfo = new Map<Id64String, DefinitionContainerInfo>();\n      const [isDefinitionContainerSupported, modelsCategoriesInfo] = await Promise.all([\n        this.getIsDefinitionContainerSupported(),\n        this.getModelsCategoriesInfo(),\n      ]);\n      if (!isDefinitionContainerSupported || modelsCategoriesInfo.size === 0) {\n        return definitionContainersInfo;\n      }\n\n      for await (const queriedDefinitionContainer of this.queryDefinitionContainers()) {\n        const modelCategoriesInfo = modelsCategoriesInfo.get(queriedDefinitionContainer.id);\n\n        definitionContainersInfo.set(queriedDefinitionContainer.id, {\n          childCategories: modelCategoriesInfo?.childCategories ?? [],\n          modelId: queriedDefinitionContainer.modelId,\n          childDefinitionContainers: [],\n          parentDefinitionContainerExists: false,\n        });\n      }\n\n      for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {\n        const parentDefinitionContainer = definitionContainersInfo.get(definitionContainerInfo.modelId);\n        if (parentDefinitionContainer !== undefined) {\n          parentDefinitionContainer.childDefinitionContainers.push(definitionContainerId);\n          definitionContainerInfo.parentDefinitionContainerExists = true;\n        }\n      }\n\n      return definitionContainersInfo;\n    })();\n    return this._definitionContainersInfo;\n  }\n\n  public async getDirectChildDefinitionContainersAndCategories(\n    parentDefinitionContainerIds: Id64Array,\n  ): Promise<{ categories: CategoryInfo[]; definitionContainers: Id64Array }> {\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n\n    const result = { definitionContainers: new Array<Id64String>(), categories: new Array<CategoryInfo>() };\n\n    parentDefinitionContainerIds.forEach((parentDefinitionContainerId) => {\n      const parentDefinitionContainerInfo = definitionContainersInfo.get(parentDefinitionContainerId);\n      if (parentDefinitionContainerInfo !== undefined) {\n        result.definitionContainers.push(...parentDefinitionContainerInfo.childDefinitionContainers);\n        result.categories.push(...parentDefinitionContainerInfo.childCategories);\n      }\n    });\n    return result;\n  }\n\n  public async getCategoriesElementModels(categoryIds: Id64Array): Promise<Map<Id64String, Id64Array>> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    const result = new Map<Id64String, Id64Array>();\n    for (const categoryId of categoryIds) {\n      for (const [modelId, categories] of elementModelsCategories) {\n        if (categories.has(categoryId)) {\n          let categoryModels = result.get(categoryId);\n          if (!categoryModels) {\n            categoryModels = new Array<Id64String>();\n            result.set(categoryId, categoryModels);\n          }\n          categoryModels.push(modelId);\n        }\n      }\n    }\n    return result;\n  }\n\n  public async getAllContainedCategories(definitionContainerIds: Id64Array): Promise<Id64Array> {\n    const result = new Array<Id64String>();\n\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n    const indirectCategories = await Promise.all(\n      definitionContainerIds.map(async (definitionContainerId) => {\n        const definitionContainerInfo = definitionContainersInfo.get(definitionContainerId);\n        if (definitionContainerInfo === undefined) {\n          return [];\n        }\n        result.push(...definitionContainerInfo.childCategories.map((category) => category.id));\n        return this.getAllContainedCategories(definitionContainerInfo.childDefinitionContainers);\n      }),\n    );\n    for (const categories of indirectCategories) {\n      result.push(...categories);\n    }\n\n    return result;\n  }\n\n  public async getInstanceKeyPaths(\n    props: { categoryId: Id64String } | { definitionContainerId: Id64String } | { subCategoryId: Id64String },\n  ): Promise<InstanceKey[]> {\n    if (\"subCategoryId\" in props) {\n      const subCategoriesInfo = await this.getSubCategoriesInfo();\n      const subCategoryInfo = subCategoriesInfo.get(props.subCategoryId);\n      if (subCategoryInfo === undefined) {\n        return [];\n      }\n      return [...(await this.getInstanceKeyPaths({ categoryId: subCategoryInfo.categoryId })), { id: props.subCategoryId, className: SUB_CATEGORY_CLASS }];\n    }\n\n    if (\"categoryId\" in props) {\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      for (const [modelId, modelCategoriesInfo] of modelsCategoriesInfo) {\n        if (modelCategoriesInfo.childCategories.find((childCategory) => childCategory.id === props.categoryId)) {\n          if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n            return [{ id: props.categoryId, className: this._categoryClass }];\n          }\n\n          return [...(await this.getInstanceKeyPaths({ definitionContainerId: modelId })), { id: props.categoryId, className: this._categoryClass }];\n        }\n      }\n      return [];\n    }\n\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n    const definitionContainerInfo = definitionContainersInfo.get(props.definitionContainerId);\n    if (definitionContainerInfo === undefined) {\n      return [];\n    }\n\n    if (!definitionContainerInfo.parentDefinitionContainerExists) {\n      return [{ id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS }];\n    }\n\n    return [\n      ...(await this.getInstanceKeyPaths({ definitionContainerId: definitionContainerInfo.modelId })),\n      { id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS },\n    ];\n  }\n\n  public async getAllDefinitionContainersAndCategories(): Promise<{ categories: Id64Array; definitionContainers: Id64Array }> {\n    const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);\n    const result = { definitionContainers: [...definitionContainersInfo.keys()], categories: new Array<Id64String>() };\n    for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n      result.categories.push(...modelCategoriesInfo.childCategories.map((childCategory) => childCategory.id));\n    }\n\n    return result;\n  }\n\n  public async getRootDefinitionContainersAndCategories(): Promise<{ categories: CategoryInfo[]; definitionContainers: Id64Array }> {\n    const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);\n    const result = { definitionContainers: new Array<Id64String>(), categories: new Array<CategoryInfo>() };\n    for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n      if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n        result.categories.push(...modelCategoriesInfo.childCategories);\n      }\n    }\n\n    for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {\n      if (!definitionContainerInfo.parentDefinitionContainerExists) {\n        result.definitionContainers.push(definitionContainerId);\n      }\n    }\n    return result;\n  }\n\n  public async getSubCategories(categoryId: Id64String): Promise<Id64Array> {\n    const subCategoriesInfo = await this.getSubCategoriesInfo();\n    const result = new Array<Id64String>();\n    for (const [subCategoryId, subCategoryInfo] of subCategoriesInfo) {\n      if (subCategoryInfo.categoryId === categoryId) {\n        result.push(subCategoryId);\n      }\n    }\n    return result;\n  }\n\n  public async getIsDefinitionContainerSupported(): Promise<boolean> {\n    this._isDefinitionContainerSupported ??= this.queryIsDefinitionContainersSupported();\n    return this._isDefinitionContainerSupported;\n  }\n}\n\n/** @internal */\nexport function getClassesByView(viewType: \"2d\" | \"3d\") {\n  return viewType === \"2d\"\n    ? { categoryClass: \"BisCore.DrawingCategory\", categoryElementClass: \"BisCore.GeometricElement2d\", categoryModelClass: \"BisCore.GeometricModel2d\" }\n    : { categoryClass: \"BisCore.SpatialCategory\", categoryElementClass: \"BisCore.GeometricElement3d\", categoryModelClass: \"BisCore.GeometricModel3d\" };\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/CategoriesVisibilityHandler.d.ts

@@ -20,15 +20,17 @@ export declare class CategoriesVisibilityHandler implements HierarchyVisibilityH
     /** Returns visibility status of the tree node. */
     getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus>;
     changeVisibility(node: HierarchyNode, on: boolean): Promise<void>;
-    private getSubCategoryVisibility;
+    private getSubCategoriesVisibility;
     private getDefinitionContainerVisibility;
     private getCategoriesVisibility;
-    private getCategoryVisibilityFromOverrides;
+    private enableCategoriesElementModelsVisibility;
     private changeSubCategoryVisibility;
     private changeCategoryVisibility;
     private changeDefinitionContainerVisibility;
     private onDisplayStyleChanged;
     private onViewedCategoriesChanged;
+    private onViewedCategoriesPerModelChanged;
+    private onViewedModelsChanged;
     private onVisibilityChangeInternal;
     private static getInstanceIdsFromHierarchyNode;
     private changeCategoryState;

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/CategoriesVisibilityHandler.js

@@ -2,7 +2,8 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { BeEvent } from "@itwin/core-bentley";
+import { assert, BeEvent } from "@itwin/core-bentley";
+import { PerModelCategoryVisibility } from "@itwin/core-frontend";
 import { HierarchyNode } from "@itwin/presentation-hierarchies";
 import { enableCategoryDisplay, enableSubCategoryDisplay } from "../../common/CategoriesVisibilityUtils.js";
 import { createVisibilityStatus } from "../../common/Tooltip.js";
@@ -11,18 +12,24 @@ import { CategoriesTreeNode } from "./CategoriesTreeNode.js";
 export class CategoriesVisibilityHandler {
     constructor(props) {
         this.onVisibilityChange = new BeEvent();
-        // eslint-disable-next-line @typescript-eslint/naming-convention
         this.onDisplayStyleChanged = () => {
             this.onVisibilityChangeInternal();
         };
-        // eslint-disable-next-line @typescript-eslint/naming-convention
         this.onViewedCategoriesChanged = () => {
             this.onVisibilityChangeInternal();
         };
+        this.onViewedCategoriesPerModelChanged = () => {
+            this.onVisibilityChangeInternal();
+        };
+        this.onViewedModelsChanged = () => {
+            this.onVisibilityChangeInternal();
+        };
         this._idsCache = props.idsCache;
         this._viewport = props.viewport;
         this._viewport.onDisplayStyleChanged.addListener(this.onDisplayStyleChanged);
         this._viewport.onViewedCategoriesChanged.addListener(this.onViewedCategoriesChanged);
+        this._viewport.onViewedCategoriesPerModelChanged.addListener(this.onViewedCategoriesPerModelChanged);
+        this._viewport.onViewedModelsChanged.addListener(this.onViewedModelsChanged);
     }
     dispose() {
         this[Symbol.dispose]();
@@ -30,6 +37,8 @@ export class CategoriesVisibilityHandler {
     [Symbol.dispose]() {
         this._viewport.onDisplayStyleChanged.removeListener(this.onDisplayStyleChanged);
         this._viewport.onViewedCategoriesChanged.removeListener(this.onViewedCategoriesChanged);
+        this._viewport.onViewedCategoriesPerModelChanged.removeListener(this.onViewedCategoriesPerModelChanged);
+        this._viewport.onViewedModelsChanged.removeListener(this.onViewedModelsChanged);
         clearTimeout(this._pendingVisibilityChange);
     }
     /** Returns visibility status of the tree node. */
@@ -38,7 +47,10 @@ export class CategoriesVisibilityHandler {
             return { state: "hidden", isDisabled: true };
         }
         if (CategoriesTreeNode.isSubCategoryNode(node)) {
-            return createVisibilityStatus(this.getSubCategoryVisibility(node));
+            if (!node.extendedData?.categoryId) {
+                return { state: "hidden", isDisabled: true };
+            }
+            return createVisibilityStatus(await this.getSubCategoriesVisibility(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node), node.extendedData.categoryId));
         }
         if (CategoriesTreeNode.isCategoryNode(node)) {
             return createVisibilityStatus(await this.getCategoriesVisibility(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node)));
@@ -53,7 +65,7 @@ export class CategoriesVisibilityHandler {
             return;
         }
         if (CategoriesTreeNode.isCategoryNode(node)) {
-            return this.changeCategoryVisibility(node, on);
+            return this.changeCategoryVisibility(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node), on);
         }
         if (CategoriesTreeNode.isSubCategoryNode(node)) {
             return this.changeSubCategoryVisibility(node, on);
@@ -62,135 +74,118 @@ export class CategoriesVisibilityHandler {
             return this.changeDefinitionContainerVisibility(node, on);
         }
     }
-    getSubCategoryVisibility(node) {
-        const parentCategoryId = node.extendedData?.categoryId;
-        if (!parentCategoryId) {
-            return "hidden";
+    async getSubCategoriesVisibility(subCategoryIds, parentCategoryId) {
+        let visibility = "unknown";
+        const categoryModels = [...(await this._idsCache.getCategoriesElementModels([parentCategoryId])).values()].flat();
+        let nonDefaultModelDisplayStatesCount = 0;
+        for (const modelId of categoryModels) {
+            if (!this._viewport.view.viewsModel(modelId)) {
+                if (visibility === "visible") {
+                    return "partial";
+                }
+                visibility = "hidden";
+                ++nonDefaultModelDisplayStatesCount;
+                continue;
+            }
+            const override = this._viewport.perModelCategoryVisibility.getOverride(modelId, parentCategoryId);
+            if (override === PerModelCategoryVisibility.Override.Show) {
+                if (visibility === "hidden") {
+                    return "partial";
+                }
+                visibility = "visible";
+                ++nonDefaultModelDisplayStatesCount;
+                continue;
+            }
+            if (override === PerModelCategoryVisibility.Override.Hide) {
+                if (visibility === "visible") {
+                    return "partial";
+                }
+                visibility = "hidden";
+                ++nonDefaultModelDisplayStatesCount;
+                continue;
+            }
         }
-        const categoryOverrideResult = this.getCategoryVisibilityFromOverrides(parentCategoryId);
-        if (categoryOverrideResult === "hidden" || categoryOverrideResult === "visible") {
-            return categoryOverrideResult;
+        if (categoryModels.length > 0 && nonDefaultModelDisplayStatesCount === categoryModels.length) {
+            assert(visibility === "visible" || visibility === "hidden");
+            return visibility;
         }
         if (!this._viewport.view.viewsCategory(parentCategoryId)) {
-            return "hidden";
+            return visibility === "visible" ? "partial" : "hidden";
         }
-        const subCategoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);
-        let visibleCount = 0;
-        let hiddenCount = 0;
-        for (const subCategoryId of subCategoryIds) {
-            const isVisible = this._viewport.isSubCategoryVisible(subCategoryId);
-            if (isVisible) {
-                ++visibleCount;
-            }
-            else {
-                ++hiddenCount;
-            }
-            if (visibleCount > 0 && hiddenCount > 0) {
+        if (subCategoryIds.length === 0) {
+            if (visibility === "hidden") {
                 return "partial";
             }
+            return "visible";
         }
-        return visibleCount > 0 ? "visible" : "hidden";
-    }
-    async getDefinitionContainerVisibility(node) {
-        const childrenResult = await this._idsCache.getAllContainedCategories(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node));
-        let hiddenCount = 0;
-        let visibleCount = 0;
-        for (const categoryId of childrenResult) {
-            const categoryVisibility = await this.getCategoriesVisibility([categoryId]);
-            if (categoryVisibility === "partial") {
+        for (const subCategoryId of subCategoryIds) {
+            const isSubCategoryVisible = this._viewport.isSubCategoryVisible(subCategoryId);
+            if (isSubCategoryVisible && visibility === "hidden") {
                 return "partial";
             }
-            if (categoryVisibility === "hidden") {
-                ++hiddenCount;
-            }
-            else {
-                ++visibleCount;
-            }
-            if (hiddenCount > 0 && visibleCount > 0) {
+            if (!isSubCategoryVisible && visibility === "visible") {
                 return "partial";
             }
+            visibility = isSubCategoryVisible ? "visible" : "hidden";
         }
-        return hiddenCount > 0 ? "hidden" : "visible";
+        assert(visibility === "visible" || visibility === "hidden");
+        return visibility;
+    }
+    async getDefinitionContainerVisibility(node) {
+        const categoryIds = await this._idsCache.getAllContainedCategories(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node));
+        return this.getCategoriesVisibility(categoryIds);
     }
     async getCategoriesVisibility(categoryIds) {
-        const overrideResult = this.getCategoryVisibilityFromOverrides(categoryIds);
-        if (overrideResult !== "none") {
-            return overrideResult;
-        }
-        let visibleCount = 0;
-        let hiddenCount = 0;
-        for (const categoryId of categoryIds) {
-            const isVisible = this._viewport.view.viewsCategory(categoryId);
-            if (isVisible) {
-                ++visibleCount;
-            }
-            else {
-                ++hiddenCount;
-            }
-            if (visibleCount > 0 && hiddenCount > 0) {
+        const subCategoriesVisibilities = await Promise.all(categoryIds.map(async (categoryId) => {
+            const subCategories = await this._idsCache.getSubCategories(categoryId);
+            return this.getSubCategoriesVisibility(subCategories, categoryId);
+        }));
+        let visibility = "unknown";
+        for (const subCategoriesVisibility of subCategoriesVisibilities) {
+            if (subCategoriesVisibility === "partial") {
                 return "partial";
             }
-        }
-        if (hiddenCount > 0) {
-            return "hidden";
-        }
-        const subCategories = (await Promise.all(categoryIds.map(async (id) => this._idsCache.getSubCategories(id)))).reduce((acc, val) => acc.concat(val), []);
-        let visibleSubCategoryCount = 0;
-        let hiddenSubCategoryCount = 0;
-        for (const subCategory of subCategories) {
-            const isVisible = this._viewport.isSubCategoryVisible(subCategory);
-            if (isVisible) {
-                ++visibleSubCategoryCount;
-            }
-            else {
-                ++hiddenSubCategoryCount;
+            if (subCategoriesVisibility === "hidden" && visibility === "visible") {
+                return "partial";
             }
-            if (hiddenSubCategoryCount > 0 && visibleSubCategoryCount > 0) {
+            if (subCategoriesVisibility === "visible" && visibility === "hidden") {
                 return "partial";
             }
+            visibility = subCategoriesVisibility;
         }
-        return hiddenSubCategoryCount > 0 ? "hidden" : "visible";
+        assert(visibility !== "unknown");
+        return visibility;
     }
-    getCategoryVisibilityFromOverrides(categoryIds) {
-        let showOverrides = 0;
-        let hideOverrides = 0;
-        for (const currentOverride of this._viewport.perModelCategoryVisibility) {
-            if (categoryIds.includes(currentOverride.categoryId)) {
-                if (currentOverride.visible) {
-                    ++showOverrides;
-                }
-                else {
-                    ++hideOverrides;
-                }
-                if (showOverrides > 0 && hideOverrides > 0) {
-                    return "partial";
-                }
-            }
+    async enableCategoriesElementModelsVisibility(categoryIds) {
+        const categoriesModelsMap = await this._idsCache.getCategoriesElementModels(categoryIds);
+        const modelIds = [...categoriesModelsMap.values()].flat();
+        const hiddenModels = modelIds.filter((modelId) => !this._viewport.view.viewsModel(modelId));
+        if (hiddenModels.length > 0) {
+            this._viewport.changeModelDisplay(hiddenModels, true);
         }
-        if (showOverrides === 0 && hideOverrides === 0) {
-            return "none";
-        }
-        return showOverrides > 0 ? "visible" : "hidden";
     }
     async changeSubCategoryVisibility(node, on) {
         const parentCategoryId = node.extendedData?.categoryId;
-        // make sure parent category is enabled
+        // make sure parent category and models are enabled
         if (on && parentCategoryId) {
-            await this.changeCategoryState([parentCategoryId], true, false);
+            await Promise.all([this.enableCategoriesElementModelsVisibility([parentCategoryId]), this.changeCategoryState([parentCategoryId], true, false)]);
         }
         const subCategoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);
         subCategoryIds.forEach((id) => {
             this.changeSubCategoryState(id, on);
         });
     }
-    async changeCategoryVisibility(node, on) {
-        const categoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);
+    async changeCategoryVisibility(categoryIds, on) {
+        // make sure models are enabled
+        if (on) {
+            await this.enableCategoriesElementModelsVisibility(categoryIds);
+        }
         return this.changeCategoryState(categoryIds, on, on);
     }
     async changeDefinitionContainerVisibility(node, on) {
         const definitionContainerId = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);
         const childCategories = await this._idsCache.getAllContainedCategories(definitionContainerId);
-        return this.changeCategoryState(childCategories, on, on);
+        return this.changeCategoryVisibility(childCategories, on);
     }
     onVisibilityChangeInternal() {
         if (this._pendingVisibilityChange) {

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/CategoriesVisibilityHandler.js.map

@@ -1 +1 @@
-{"version":3,"file":"CategoriesVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,MAAM,2CAA2C,CAAC;AAC5G,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAa7D,gBAAgB;AAChB,MAAM,OAAO,2BAA2B;IAKtC,YAAY,KAAuC;QAiB5C,uBAAkB,GAAG,IAAI,OAAO,EAAE,CAAC;QA4L1C,gEAAgE;QACxD,0BAAqB,GAAG,GAAG,EAAE;YACnC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QAEF,gEAAgE;QACxD,8BAAyB,GAAG,GAAG,EAAE;YACvC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QApNA,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,WAAW,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC7E,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,WAAW,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IACvF,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IACzB,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,cAAc,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,cAAc,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACxF,YAAY,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9C,CAAC;IAID,kDAAkD;IAC3C,KAAK,CAAC,mBAAmB,CAAC,IAAmB;QAClD,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE;YACxC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;SAC9C;QAED,IAAI,kBAAkB,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE;YAC9C,OAAO,sBAAsB,CAAC,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC;SACpE;QAED,IAAI,kBAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE;YAC3C,OAAO,sBAAsB,CAAC,MAAM,IAAI,CAAC,uBAAuB,CAAC,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SACtI;QAED,IAAI,kBAAkB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE;YACtD,OAAO,sBAAsB,CAAC,MAAM,IAAI,CAAC,gCAAgC,CAAC,IAAI,CAAC,CAAC,CAAC;SAClF;QAED,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IAC/C,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAmB,EAAE,EAAW;QAC5D,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE;YACxC,OAAO;SACR;QAED,IAAI,kBAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE;YAC3C,OAAO,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;SAChD;QAED,IAAI,kBAAkB,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE;YAC9C,OAAO,IAAI,CAAC,2BAA2B,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;SACnD;QAED,IAAI,kBAAkB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE;YACtD,OAAO,IAAI,CAAC,mCAAmC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;SAC3D;IACH,CAAC;IAEO,wBAAwB,CAAC,IAAmB;QAClD,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;QACvD,IAAI,CAAC,gBAAgB,EAAE;YACrB,OAAO,QAAQ,CAAC;SACjB;QAED,MAAM,sBAAsB,GAAG,IAAI,CAAC,kCAAkC,CAAC,gBAAgB,CAAC,CAAC;QACzF,IAAI,sBAAsB,KAAK,QAAQ,IAAI,sBAAsB,KAAK,SAAS,EAAE;YAC/E,OAAO,sBAAsB,CAAC;SAC/B;QAED,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,EAAE;YACxD,OAAO,QAAQ,CAAC;SACjB;QACD,MAAM,cAAc,GAAG,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QACzF,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE;YAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;YACrE,IAAI,SAAS,EAAE;gBACb,EAAE,YAAY,CAAC;aAChB;iBAAM;gBACL,EAAE,WAAW,CAAC;aACf;YACD,IAAI,YAAY,GAAG,CAAC,IAAI,WAAW,GAAG,CAAC,EAAE;gBACvC,OAAO,SAAS,CAAC;aAClB;SACF;QACD,OAAO,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,gCAAgC,CAAC,IAAmB;QAChE,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC,CAAC;QACzI,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,KAAK,MAAM,UAAU,IAAI,cAAc,EAAE;YACvC,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;YAC5E,IAAI,kBAAkB,KAAK,SAAS,EAAE;gBACpC,OAAO,SAAS,CAAC;aAClB;YAED,IAAI,kBAAkB,KAAK,QAAQ,EAAE;gBACnC,EAAE,WAAW,CAAC;aACf;iBAAM;gBACL,EAAE,YAAY,CAAC;aAChB;YAED,IAAI,WAAW,GAAG,CAAC,IAAI,YAAY,GAAG,CAAC,EAAE;gBACvC,OAAO,SAAS,CAAC;aAClB;SACF;QAED,OAAO,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,WAAsB;QAC1D,MAAM,cAAc,GAAG,IAAI,CAAC,kCAAkC,CAAC,WAAW,CAAC,CAAC;QAC5E,IAAI,cAAc,KAAK,MAAM,EAAE;YAC7B,OAAO,cAAc,CAAC;SACvB;QACD,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE;YACpC,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAChE,IAAI,SAAS,EAAE;gBACb,EAAE,YAAY,CAAC;aAChB;iBAAM;gBACL,EAAE,WAAW,CAAC;aACf;YACD,IAAI,YAAY,GAAG,CAAC,IAAI,WAAW,GAAG,CAAC,EAAE;gBACvC,OAAO,SAAS,CAAC;aAClB;SACF;QAED,IAAI,WAAW,GAAG,CAAC,EAAE;YACnB,OAAO,QAAQ,CAAC;SACjB;QAED,MAAM,aAAa,GAAG,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;QACxJ,IAAI,uBAAuB,GAAG,CAAC,CAAC;QAChC,IAAI,sBAAsB,GAAG,CAAC,CAAC;QAE/B,KAAK,MAAM,WAAW,IAAI,aAAa,EAAE;YACvC,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;YACnE,IAAI,SAAS,EAAE;gBACb,EAAE,uBAAuB,CAAC;aAC3B;iBAAM;gBACL,EAAE,sBAAsB,CAAC;aAC1B;YACD,IAAI,sBAAsB,GAAG,CAAC,IAAI,uBAAuB,GAAG,CAAC,EAAE;gBAC7D,OAAO,SAAS,CAAC;aAClB;SACF;QAED,OAAO,sBAAsB,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3D,CAAC;IAEO,kCAAkC,CAAC,WAAsB;QAC/D,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,aAAa,GAAG,CAAC,CAAC;QAEtB,KAAK,MAAM,eAAe,IAAI,IAAI,CAAC,SAAS,CAAC,0BAA0B,EAAE;YACvE,IAAI,WAAW,CAAC,QAAQ,CAAC,eAAe,CAAC,UAAU,CAAC,EAAE;gBACpD,IAAI,eAAe,CAAC,OAAO,EAAE;oBAC3B,EAAE,aAAa,CAAC;iBACjB;qBAAM;oBACL,EAAE,aAAa,CAAC;iBACjB;gBAED,IAAI,aAAa,GAAG,CAAC,IAAI,aAAa,GAAG,CAAC,EAAE;oBAC1C,OAAO,SAAS,CAAC;iBAClB;aACF;SACF;QAED,IAAI,aAAa,KAAK,CAAC,IAAI,aAAa,KAAK,CAAC,EAAE;YAC9C,OAAO,MAAM,CAAC;SACf;QAED,OAAO,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;IAClD,CAAC;IAEO,KAAK,CAAC,2BAA2B,CAAC,IAAmB,EAAE,EAAW;QACxE,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;QAEvD,uCAAuC;QACvC,IAAI,EAAE,IAAI,gBAAgB,EAAE;YAC1B,MAAM,IAAI,CAAC,mBAAmB,CAAC,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;SACjE;QAED,MAAM,cAAc,GAAG,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QACzF,cAAc,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;YAC5B,IAAI,CAAC,sBAAsB,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,wBAAwB,CAAC,IAAmB,EAAE,EAAW;QACrE,MAAM,WAAW,GAAG,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QACtF,OAAO,IAAI,CAAC,mBAAmB,CAAC,WAAW,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;IACvD,CAAC;IAEO,KAAK,CAAC,mCAAmC,CAAC,IAAmB,EAAE,EAAW;QAChF,MAAM,qBAAqB,GAAG,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QAChG,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,qBAAqB,CAAC,CAAC;QAC9F,OAAO,IAAI,CAAC,mBAAmB,CAAC,eAAe,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;IAC3D,CAAC;IAYO,0BAA0B;QAChC,IAAI,IAAI,CAAC,wBAAwB,EAAE;YACjC,OAAO;SACR;QAED,IAAI,CAAC,wBAAwB,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9C,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,CAAC;YACrC,IAAI,CAAC,wBAAwB,GAAG,SAAS,CAAC;QAC5C,CAAC,EAAE,CAAC,CAAC,CAAC;IACR,CAAC;IAEO,MAAM,CAAC,+BAA+B,CAAC,IAAmB;QAChE,OAAO,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,0BAA0B,CAAC,EAAE,CAAC;IAC1I,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,GAAa,EAAE,OAAgB,EAAE,sBAA+B;QAChG,MAAM,qBAAqB,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;IACpF,CAAC;IAEO,sBAAsB,CAAC,GAAW,EAAE,OAAgB;QAC1D,wBAAwB,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IACzD,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { BeEvent } from \"@itwin/core-bentley\";\nimport { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport { enableCategoryDisplay, enableSubCategoryDisplay } from \"../../common/CategoriesVisibilityUtils.js\";\nimport { createVisibilityStatus } from \"../../common/Tooltip.js\";\nimport { CategoriesTreeNode } from \"./CategoriesTreeNode.js\";\n\nimport type { Id64Array } from \"@itwin/core-bentley\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { HierarchyVisibilityHandler, VisibilityStatus } from \"../../common/UseHierarchyVisibility.js\";\nimport type { CategoriesTreeIdsCache } from \"./CategoriesTreeIdsCache.js\";\n\n/** @internal */\nexport interface CategoriesVisibilityHandlerProps {\n  viewport: Viewport;\n  idsCache: CategoriesTreeIdsCache;\n}\n\n/** @internal */\nexport class CategoriesVisibilityHandler implements HierarchyVisibilityHandler {\n  private _pendingVisibilityChange: any;\n  private _viewport: Viewport;\n  private _idsCache: CategoriesTreeIdsCache;\n\n  constructor(props: CategoriesVisibilityHandlerProps) {\n    this._idsCache = props.idsCache;\n    this._viewport = props.viewport;\n    this._viewport.onDisplayStyleChanged.addListener(this.onDisplayStyleChanged);\n    this._viewport.onViewedCategoriesChanged.addListener(this.onViewedCategoriesChanged);\n  }\n\n  public dispose() {\n    this[Symbol.dispose]();\n  }\n\n  public [Symbol.dispose]() {\n    this._viewport.onDisplayStyleChanged.removeListener(this.onDisplayStyleChanged);\n    this._viewport.onViewedCategoriesChanged.removeListener(this.onViewedCategoriesChanged);\n    clearTimeout(this._pendingVisibilityChange);\n  }\n\n  public onVisibilityChange = new BeEvent();\n\n  /** Returns visibility status of the tree node. */\n  public async getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> {\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return { state: \"hidden\", isDisabled: true };\n    }\n\n    if (CategoriesTreeNode.isSubCategoryNode(node)) {\n      return createVisibilityStatus(this.getSubCategoryVisibility(node));\n    }\n\n    if (CategoriesTreeNode.isCategoryNode(node)) {\n      return createVisibilityStatus(await this.getCategoriesVisibility(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node)));\n    }\n\n    if (CategoriesTreeNode.isDefinitionContainerNode(node)) {\n      return createVisibilityStatus(await this.getDefinitionContainerVisibility(node));\n    }\n\n    return { state: \"hidden\", isDisabled: true };\n  }\n\n  public async changeVisibility(node: HierarchyNode, on: boolean) {\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return;\n    }\n\n    if (CategoriesTreeNode.isCategoryNode(node)) {\n      return this.changeCategoryVisibility(node, on);\n    }\n\n    if (CategoriesTreeNode.isSubCategoryNode(node)) {\n      return this.changeSubCategoryVisibility(node, on);\n    }\n\n    if (CategoriesTreeNode.isDefinitionContainerNode(node)) {\n      return this.changeDefinitionContainerVisibility(node, on);\n    }\n  }\n\n  private getSubCategoryVisibility(node: HierarchyNode): VisibilityStatus[\"state\"] {\n    const parentCategoryId = node.extendedData?.categoryId;\n    if (!parentCategoryId) {\n      return \"hidden\";\n    }\n\n    const categoryOverrideResult = this.getCategoryVisibilityFromOverrides(parentCategoryId);\n    if (categoryOverrideResult === \"hidden\" || categoryOverrideResult === \"visible\") {\n      return categoryOverrideResult;\n    }\n\n    if (!this._viewport.view.viewsCategory(parentCategoryId)) {\n      return \"hidden\";\n    }\n    const subCategoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);\n    let visibleCount = 0;\n    let hiddenCount = 0;\n    for (const subCategoryId of subCategoryIds) {\n      const isVisible = this._viewport.isSubCategoryVisible(subCategoryId);\n      if (isVisible) {\n        ++visibleCount;\n      } else {\n        ++hiddenCount;\n      }\n      if (visibleCount > 0 && hiddenCount > 0) {\n        return \"partial\";\n      }\n    }\n    return visibleCount > 0 ? \"visible\" : \"hidden\";\n  }\n\n  private async getDefinitionContainerVisibility(node: HierarchyNode): Promise<VisibilityStatus[\"state\"]> {\n    const childrenResult = await this._idsCache.getAllContainedCategories(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node));\n    let hiddenCount = 0;\n    let visibleCount = 0;\n    for (const categoryId of childrenResult) {\n      const categoryVisibility = await this.getCategoriesVisibility([categoryId]);\n      if (categoryVisibility === \"partial\") {\n        return \"partial\";\n      }\n\n      if (categoryVisibility === \"hidden\") {\n        ++hiddenCount;\n      } else {\n        ++visibleCount;\n      }\n\n      if (hiddenCount > 0 && visibleCount > 0) {\n        return \"partial\";\n      }\n    }\n\n    return hiddenCount > 0 ? \"hidden\" : \"visible\";\n  }\n\n  private async getCategoriesVisibility(categoryIds: Id64Array): Promise<VisibilityStatus[\"state\"]> {\n    const overrideResult = this.getCategoryVisibilityFromOverrides(categoryIds);\n    if (overrideResult !== \"none\") {\n      return overrideResult;\n    }\n    let visibleCount = 0;\n    let hiddenCount = 0;\n    for (const categoryId of categoryIds) {\n      const isVisible = this._viewport.view.viewsCategory(categoryId);\n      if (isVisible) {\n        ++visibleCount;\n      } else {\n        ++hiddenCount;\n      }\n      if (visibleCount > 0 && hiddenCount > 0) {\n        return \"partial\";\n      }\n    }\n\n    if (hiddenCount > 0) {\n      return \"hidden\";\n    }\n\n    const subCategories = (await Promise.all(categoryIds.map(async (id) => this._idsCache.getSubCategories(id)))).reduce((acc, val) => acc.concat(val), []);\n    let visibleSubCategoryCount = 0;\n    let hiddenSubCategoryCount = 0;\n\n    for (const subCategory of subCategories) {\n      const isVisible = this._viewport.isSubCategoryVisible(subCategory);\n      if (isVisible) {\n        ++visibleSubCategoryCount;\n      } else {\n        ++hiddenSubCategoryCount;\n      }\n      if (hiddenSubCategoryCount > 0 && visibleSubCategoryCount > 0) {\n        return \"partial\";\n      }\n    }\n\n    return hiddenSubCategoryCount > 0 ? \"hidden\" : \"visible\";\n  }\n\n  private getCategoryVisibilityFromOverrides(categoryIds: Id64Array): VisibilityStatus[\"state\"] | \"none\" {\n    let showOverrides = 0;\n    let hideOverrides = 0;\n\n    for (const currentOverride of this._viewport.perModelCategoryVisibility) {\n      if (categoryIds.includes(currentOverride.categoryId)) {\n        if (currentOverride.visible) {\n          ++showOverrides;\n        } else {\n          ++hideOverrides;\n        }\n\n        if (showOverrides > 0 && hideOverrides > 0) {\n          return \"partial\";\n        }\n      }\n    }\n\n    if (showOverrides === 0 && hideOverrides === 0) {\n      return \"none\";\n    }\n\n    return showOverrides > 0 ? \"visible\" : \"hidden\";\n  }\n\n  private async changeSubCategoryVisibility(node: HierarchyNode, on: boolean) {\n    const parentCategoryId = node.extendedData?.categoryId;\n\n    // make sure parent category is enabled\n    if (on && parentCategoryId) {\n      await this.changeCategoryState([parentCategoryId], true, false);\n    }\n\n    const subCategoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);\n    subCategoryIds.forEach((id) => {\n      this.changeSubCategoryState(id, on);\n    });\n  }\n\n  private async changeCategoryVisibility(node: HierarchyNode, on: boolean) {\n    const categoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);\n    return this.changeCategoryState(categoryIds, on, on);\n  }\n\n  private async changeDefinitionContainerVisibility(node: HierarchyNode, on: boolean) {\n    const definitionContainerId = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);\n    const childCategories = await this._idsCache.getAllContainedCategories(definitionContainerId);\n    return this.changeCategoryState(childCategories, on, on);\n  }\n\n  // eslint-disable-next-line @typescript-eslint/naming-convention\n  private onDisplayStyleChanged = () => {\n    this.onVisibilityChangeInternal();\n  };\n\n  // eslint-disable-next-line @typescript-eslint/naming-convention\n  private onViewedCategoriesChanged = () => {\n    this.onVisibilityChangeInternal();\n  };\n\n  private onVisibilityChangeInternal() {\n    if (this._pendingVisibilityChange) {\n      return;\n    }\n\n    this._pendingVisibilityChange = setTimeout(() => {\n      this.onVisibilityChange.raiseEvent();\n      this._pendingVisibilityChange = undefined;\n    }, 0);\n  }\n\n  private static getInstanceIdsFromHierarchyNode(node: HierarchyNode) {\n    return HierarchyNode.isInstancesNode(node) ? node.key.instanceKeys.map((instanceKey) => instanceKey.id) : /* istanbul ignore next */ [];\n  }\n\n  private async changeCategoryState(ids: string[], enabled: boolean, enableAllSubCategories: boolean) {\n    await enableCategoryDisplay(this._viewport, ids, enabled, enableAllSubCategories);\n  }\n\n  private changeSubCategoryState(key: string, enabled: boolean) {\n    enableSubCategoryDisplay(this._viewport, key, enabled);\n  }\n}\n"]}
+{"version":3,"file":"CategoriesVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAC;AAClE,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,MAAM,2CAA2C,CAAC;AAC5G,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAc7D,gBAAgB;AAChB,MAAM,OAAO,2BAA2B;IAKtC,YAAY,KAAuC;QAqB5C,uBAAkB,GAAG,IAAI,OAAO,EAAE,CAAC;QA6KlC,0BAAqB,GAAG,GAAG,EAAE;YACnC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QAEM,8BAAyB,GAAG,GAAG,EAAE;YACvC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QAEM,sCAAiC,GAAG,GAAG,EAAE;YAC/C,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QAEM,0BAAqB,GAAG,GAAG,EAAE;YACnC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QA/MA,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,WAAW,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC7E,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,WAAW,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACrF,IAAI,CAAC,SAAS,CAAC,iCAAiC,CAAC,WAAW,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;QACrG,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,WAAW,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IAC/E,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IACzB,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,cAAc,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,cAAc,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACxF,IAAI,CAAC,SAAS,CAAC,iCAAiC,CAAC,cAAc,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;QACxG,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,cAAc,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAChF,YAAY,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9C,CAAC;IAID,kDAAkD;IAC3C,KAAK,CAAC,mBAAmB,CAAC,IAAmB;QAClD,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE;YACxC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;SAC9C;QAED,IAAI,kBAAkB,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE;YAC9C,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,UAAU,EAAE;gBAClC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;aAC9C;YACD,OAAO,sBAAsB,CAC3B,MAAM,IAAI,CAAC,0BAA0B,CAAC,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,CACvI,CAAC;SACH;QAED,IAAI,kBAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE;YAC3C,OAAO,sBAAsB,CAAC,MAAM,IAAI,CAAC,uBAAuB,CAAC,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SACtI;QAED,IAAI,kBAAkB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE;YACtD,OAAO,sBAAsB,CAAC,MAAM,IAAI,CAAC,gCAAgC,CAAC,IAAI,CAAC,CAAC,CAAC;SAClF;QAED,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IAC/C,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAmB,EAAE,EAAW;QAC5D,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE;YACxC,OAAO;SACR;QAED,IAAI,kBAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE;YAC3C,OAAO,IAAI,CAAC,wBAAwB,CAAC,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;SAC7G;QAED,IAAI,kBAAkB,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE;YAC9C,OAAO,IAAI,CAAC,2BAA2B,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;SACnD;QAED,IAAI,kBAAkB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE;YACtD,OAAO,IAAI,CAAC,mCAAmC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;SAC3D;IACH,CAAC;IAEO,KAAK,CAAC,0BAA0B,CAAC,cAAyB,EAAE,gBAA4B;QAC9F,IAAI,UAAU,GAAqC,SAAS,CAAC;QAC7D,MAAM,cAAc,GAAG,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAClH,IAAI,iCAAiC,GAAG,CAAC,CAAC;QAC1C,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE;YACpC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE;gBAC5C,IAAI,UAAU,KAAK,SAAS,EAAE;oBAC5B,OAAO,SAAS,CAAC;iBAClB;gBACD,UAAU,GAAG,QAAQ,CAAC;gBACtB,EAAE,iCAAiC,CAAC;gBACpC,SAAS;aACV;YACD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,0BAA0B,CAAC,WAAW,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;YAClG,IAAI,QAAQ,KAAK,0BAA0B,CAAC,QAAQ,CAAC,IAAI,EAAE;gBACzD,IAAI,UAAU,KAAK,QAAQ,EAAE;oBAC3B,OAAO,SAAS,CAAC;iBAClB;gBACD,UAAU,GAAG,SAAS,CAAC;gBACvB,EAAE,iCAAiC,CAAC;gBACpC,SAAS;aACV;YACD,IAAI,QAAQ,KAAK,0BAA0B,CAAC,QAAQ,CAAC,IAAI,EAAE;gBACzD,IAAI,UAAU,KAAK,SAAS,EAAE;oBAC5B,OAAO,SAAS,CAAC;iBAClB;gBACD,UAAU,GAAG,QAAQ,CAAC;gBACtB,EAAE,iCAAiC,CAAC;gBACpC,SAAS;aACV;SACF;QACD,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,IAAI,iCAAiC,KAAK,cAAc,CAAC,MAAM,EAAE;YAC5F,MAAM,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,QAAQ,CAAC,CAAC;YAC5D,OAAO,UAAU,CAAC;SACnB;QAED,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,EAAE;YACxD,OAAO,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;SACxD;QAED,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE;YAC/B,IAAI,UAAU,KAAK,QAAQ,EAAE;gBAC3B,OAAO,SAAS,CAAC;aAClB;YACD,OAAO,SAAS,CAAC;SAClB;QAED,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE;YAC1C,MAAM,oBAAoB,GAAG,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;YAChF,IAAI,oBAAoB,IAAI,UAAU,KAAK,QAAQ,EAAE;gBACnD,OAAO,SAAS,CAAC;aAClB;YACD,IAAI,CAAC,oBAAoB,IAAI,UAAU,KAAK,SAAS,EAAE;gBACrD,OAAO,SAAS,CAAC;aAClB;YACD,UAAU,GAAG,oBAAoB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;SAC1D;QACD,MAAM,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,QAAQ,CAAC,CAAC;QAC5D,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,gCAAgC,CAAC,IAAmB;QAChE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC,CAAC;QACtI,OAAO,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,WAAsB;QAC1D,MAAM,yBAAyB,GAAG,MAAM,OAAO,CAAC,GAAG,CACjD,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE;YACnC,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;YACxE,OAAO,IAAI,CAAC,0BAA0B,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;QACpE,CAAC,CAAC,CACH,CAAC;QACF,IAAI,UAAU,GAAqC,SAAS,CAAC;QAE7D,KAAK,MAAM,uBAAuB,IAAI,yBAAyB,EAAE;YAC/D,IAAI,uBAAuB,KAAK,SAAS,EAAE;gBACzC,OAAO,SAAS,CAAC;aAClB;YACD,IAAI,uBAAuB,KAAK,QAAQ,IAAI,UAAU,KAAK,SAAS,EAAE;gBACpE,OAAO,SAAS,CAAC;aAClB;YACD,IAAI,uBAAuB,KAAK,SAAS,IAAI,UAAU,KAAK,QAAQ,EAAE;gBACpE,OAAO,SAAS,CAAC;aAClB;YACD,UAAU,GAAG,uBAAuB,CAAC;SACtC;QACD,MAAM,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC;QACjC,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,uCAAuC,CAAC,WAAsB;QAC1E,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,0BAA0B,CAAC,WAAW,CAAC,CAAC;QACzF,MAAM,QAAQ,GAAG,CAAC,GAAG,mBAAmB,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC1D,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC;QAC5F,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;YAC3B,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;SACvD;IACH,CAAC;IAEO,KAAK,CAAC,2BAA2B,CAAC,IAAmB,EAAE,EAAW;QACxE,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;QAEvD,mDAAmD;QACnD,IAAI,EAAE,IAAI,gBAAgB,EAAE;YAC1B,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC,gBAAgB,CAAC,CAAC,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;SAClJ;QAED,MAAM,cAAc,GAAG,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QACzF,cAAc,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;YAC5B,IAAI,CAAC,sBAAsB,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC;IACO,KAAK,CAAC,wBAAwB,CAAC,WAAsB,EAAE,EAAW;QACxE,+BAA+B;QAC/B,IAAI,EAAE,EAAE;YACN,MAAM,IAAI,CAAC,uCAAuC,CAAC,WAAW,CAAC,CAAC;SACjE;QACD,OAAO,IAAI,CAAC,mBAAmB,CAAC,WAAW,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;IACvD,CAAC;IAEO,KAAK,CAAC,mCAAmC,CAAC,IAAmB,EAAE,EAAW;QAChF,MAAM,qBAAqB,GAAG,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QAChG,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,qBAAqB,CAAC,CAAC;QAC9F,OAAO,IAAI,CAAC,wBAAwB,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;IAC5D,CAAC;IAkBO,0BAA0B;QAChC,IAAI,IAAI,CAAC,wBAAwB,EAAE;YACjC,OAAO;SACR;QAED,IAAI,CAAC,wBAAwB,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9C,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,CAAC;YACrC,IAAI,CAAC,wBAAwB,GAAG,SAAS,CAAC;QAC5C,CAAC,EAAE,CAAC,CAAC,CAAC;IACR,CAAC;IAEO,MAAM,CAAC,+BAA+B,CAAC,IAAmB;QAChE,OAAO,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,0BAA0B,CAAC,EAAE,CAAC;IAC1I,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,GAAa,EAAE,OAAgB,EAAE,sBAA+B;QAChG,MAAM,qBAAqB,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;IACpF,CAAC;IAEO,sBAAsB,CAAC,GAAW,EAAE,OAAgB;QAC1D,wBAAwB,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IACzD,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { assert, BeEvent } from \"@itwin/core-bentley\";\nimport { PerModelCategoryVisibility } from \"@itwin/core-frontend\";\nimport { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport { enableCategoryDisplay, enableSubCategoryDisplay } from \"../../common/CategoriesVisibilityUtils.js\";\nimport { createVisibilityStatus } from \"../../common/Tooltip.js\";\nimport { CategoriesTreeNode } from \"./CategoriesTreeNode.js\";\n\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { Visibility } from \"../../common/Tooltip.js\";\nimport type { Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyVisibilityHandler, VisibilityStatus } from \"../../common/UseHierarchyVisibility.js\";\nimport type { CategoriesTreeIdsCache } from \"./CategoriesTreeIdsCache.js\";\n\n/** @internal */\nexport interface CategoriesVisibilityHandlerProps {\n  viewport: Viewport;\n  idsCache: CategoriesTreeIdsCache;\n}\n\n/** @internal */\nexport class CategoriesVisibilityHandler implements HierarchyVisibilityHandler {\n  private _pendingVisibilityChange: any;\n  private _viewport: Viewport;\n  private _idsCache: CategoriesTreeIdsCache;\n\n  constructor(props: CategoriesVisibilityHandlerProps) {\n    this._idsCache = props.idsCache;\n    this._viewport = props.viewport;\n    this._viewport.onDisplayStyleChanged.addListener(this.onDisplayStyleChanged);\n    this._viewport.onViewedCategoriesChanged.addListener(this.onViewedCategoriesChanged);\n    this._viewport.onViewedCategoriesPerModelChanged.addListener(this.onViewedCategoriesPerModelChanged);\n    this._viewport.onViewedModelsChanged.addListener(this.onViewedModelsChanged);\n  }\n\n  public dispose() {\n    this[Symbol.dispose]();\n  }\n\n  public [Symbol.dispose]() {\n    this._viewport.onDisplayStyleChanged.removeListener(this.onDisplayStyleChanged);\n    this._viewport.onViewedCategoriesChanged.removeListener(this.onViewedCategoriesChanged);\n    this._viewport.onViewedCategoriesPerModelChanged.removeListener(this.onViewedCategoriesPerModelChanged);\n    this._viewport.onViewedModelsChanged.removeListener(this.onViewedModelsChanged);\n    clearTimeout(this._pendingVisibilityChange);\n  }\n\n  public onVisibilityChange = new BeEvent();\n\n  /** Returns visibility status of the tree node. */\n  public async getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> {\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return { state: \"hidden\", isDisabled: true };\n    }\n\n    if (CategoriesTreeNode.isSubCategoryNode(node)) {\n      if (!node.extendedData?.categoryId) {\n        return { state: \"hidden\", isDisabled: true };\n      }\n      return createVisibilityStatus(\n        await this.getSubCategoriesVisibility(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node), node.extendedData.categoryId),\n      );\n    }\n\n    if (CategoriesTreeNode.isCategoryNode(node)) {\n      return createVisibilityStatus(await this.getCategoriesVisibility(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node)));\n    }\n\n    if (CategoriesTreeNode.isDefinitionContainerNode(node)) {\n      return createVisibilityStatus(await this.getDefinitionContainerVisibility(node));\n    }\n\n    return { state: \"hidden\", isDisabled: true };\n  }\n\n  public async changeVisibility(node: HierarchyNode, on: boolean) {\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return;\n    }\n\n    if (CategoriesTreeNode.isCategoryNode(node)) {\n      return this.changeCategoryVisibility(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node), on);\n    }\n\n    if (CategoriesTreeNode.isSubCategoryNode(node)) {\n      return this.changeSubCategoryVisibility(node, on);\n    }\n\n    if (CategoriesTreeNode.isDefinitionContainerNode(node)) {\n      return this.changeDefinitionContainerVisibility(node, on);\n    }\n  }\n\n  private async getSubCategoriesVisibility(subCategoryIds: Id64Array, parentCategoryId: Id64String): Promise<Visibility> {\n    let visibility: \"visible\" | \"hidden\" | \"unknown\" = \"unknown\";\n    const categoryModels = [...(await this._idsCache.getCategoriesElementModels([parentCategoryId])).values()].flat();\n    let nonDefaultModelDisplayStatesCount = 0;\n    for (const modelId of categoryModels) {\n      if (!this._viewport.view.viewsModel(modelId)) {\n        if (visibility === \"visible\") {\n          return \"partial\";\n        }\n        visibility = \"hidden\";\n        ++nonDefaultModelDisplayStatesCount;\n        continue;\n      }\n      const override = this._viewport.perModelCategoryVisibility.getOverride(modelId, parentCategoryId);\n      if (override === PerModelCategoryVisibility.Override.Show) {\n        if (visibility === \"hidden\") {\n          return \"partial\";\n        }\n        visibility = \"visible\";\n        ++nonDefaultModelDisplayStatesCount;\n        continue;\n      }\n      if (override === PerModelCategoryVisibility.Override.Hide) {\n        if (visibility === \"visible\") {\n          return \"partial\";\n        }\n        visibility = \"hidden\";\n        ++nonDefaultModelDisplayStatesCount;\n        continue;\n      }\n    }\n    if (categoryModels.length > 0 && nonDefaultModelDisplayStatesCount === categoryModels.length) {\n      assert(visibility === \"visible\" || visibility === \"hidden\");\n      return visibility;\n    }\n\n    if (!this._viewport.view.viewsCategory(parentCategoryId)) {\n      return visibility === \"visible\" ? \"partial\" : \"hidden\";\n    }\n\n    if (subCategoryIds.length === 0) {\n      if (visibility === \"hidden\") {\n        return \"partial\";\n      }\n      return \"visible\";\n    }\n\n    for (const subCategoryId of subCategoryIds) {\n      const isSubCategoryVisible = this._viewport.isSubCategoryVisible(subCategoryId);\n      if (isSubCategoryVisible && visibility === \"hidden\") {\n        return \"partial\";\n      }\n      if (!isSubCategoryVisible && visibility === \"visible\") {\n        return \"partial\";\n      }\n      visibility = isSubCategoryVisible ? \"visible\" : \"hidden\";\n    }\n    assert(visibility === \"visible\" || visibility === \"hidden\");\n    return visibility;\n  }\n\n  private async getDefinitionContainerVisibility(node: HierarchyNode): Promise<Visibility> {\n    const categoryIds = await this._idsCache.getAllContainedCategories(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node));\n    return this.getCategoriesVisibility(categoryIds);\n  }\n\n  private async getCategoriesVisibility(categoryIds: Id64Array): Promise<Visibility> {\n    const subCategoriesVisibilities = await Promise.all(\n      categoryIds.map(async (categoryId) => {\n        const subCategories = await this._idsCache.getSubCategories(categoryId);\n        return this.getSubCategoriesVisibility(subCategories, categoryId);\n      }),\n    );\n    let visibility: \"visible\" | \"hidden\" | \"unknown\" = \"unknown\";\n\n    for (const subCategoriesVisibility of subCategoriesVisibilities) {\n      if (subCategoriesVisibility === \"partial\") {\n        return \"partial\";\n      }\n      if (subCategoriesVisibility === \"hidden\" && visibility === \"visible\") {\n        return \"partial\";\n      }\n      if (subCategoriesVisibility === \"visible\" && visibility === \"hidden\") {\n        return \"partial\";\n      }\n      visibility = subCategoriesVisibility;\n    }\n    assert(visibility !== \"unknown\");\n    return visibility;\n  }\n\n  private async enableCategoriesElementModelsVisibility(categoryIds: Id64Array) {\n    const categoriesModelsMap = await this._idsCache.getCategoriesElementModels(categoryIds);\n    const modelIds = [...categoriesModelsMap.values()].flat();\n    const hiddenModels = modelIds.filter((modelId) => !this._viewport.view.viewsModel(modelId));\n    if (hiddenModels.length > 0) {\n      this._viewport.changeModelDisplay(hiddenModels, true);\n    }\n  }\n\n  private async changeSubCategoryVisibility(node: HierarchyNode, on: boolean) {\n    const parentCategoryId = node.extendedData?.categoryId;\n\n    // make sure parent category and models are enabled\n    if (on && parentCategoryId) {\n      await Promise.all([this.enableCategoriesElementModelsVisibility([parentCategoryId]), this.changeCategoryState([parentCategoryId], true, false)]);\n    }\n\n    const subCategoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);\n    subCategoryIds.forEach((id) => {\n      this.changeSubCategoryState(id, on);\n    });\n  }\n  private async changeCategoryVisibility(categoryIds: Id64Array, on: boolean) {\n    // make sure models are enabled\n    if (on) {\n      await this.enableCategoriesElementModelsVisibility(categoryIds);\n    }\n    return this.changeCategoryState(categoryIds, on, on);\n  }\n\n  private async changeDefinitionContainerVisibility(node: HierarchyNode, on: boolean) {\n    const definitionContainerId = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);\n    const childCategories = await this._idsCache.getAllContainedCategories(definitionContainerId);\n    return this.changeCategoryVisibility(childCategories, on);\n  }\n\n  private onDisplayStyleChanged = () => {\n    this.onVisibilityChangeInternal();\n  };\n\n  private onViewedCategoriesChanged = () => {\n    this.onVisibilityChangeInternal();\n  };\n\n  private onViewedCategoriesPerModelChanged = () => {\n    this.onVisibilityChangeInternal();\n  };\n\n  private onViewedModelsChanged = () => {\n    this.onVisibilityChangeInternal();\n  };\n\n  private onVisibilityChangeInternal() {\n    if (this._pendingVisibilityChange) {\n      return;\n    }\n\n    this._pendingVisibilityChange = setTimeout(() => {\n      this.onVisibilityChange.raiseEvent();\n      this._pendingVisibilityChange = undefined;\n    }, 0);\n  }\n\n  private static getInstanceIdsFromHierarchyNode(node: HierarchyNode) {\n    return HierarchyNode.isInstancesNode(node) ? node.key.instanceKeys.map((instanceKey) => instanceKey.id) : /* istanbul ignore next */ [];\n  }\n\n  private async changeCategoryState(ids: string[], enabled: boolean, enableAllSubCategories: boolean) {\n    await enableCategoryDisplay(this._viewport, ids, enabled, enableAllSubCategories);\n  }\n\n  private changeSubCategoryState(key: string, enabled: boolean) {\n    enableSubCategoryDisplay(this._viewport, key, enabled);\n  }\n}\n"]}

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

@@ -0,0 +1,18 @@
+import type { IModelConnection } from "@itwin/core-frontend";
+import type { SchemaContext } from "@itwin/ecschema-metadata";
+import type { useIModelTree } from "@itwin/presentation-hierarchies-react";
+import type { FunctionProps } from "./Utils.js";
+type IModelAccess = FunctionProps<typeof useIModelTree>["imodelAccess"];
+export interface UseIModelAccessProps {
+    imodel: IModelConnection;
+    getSchemaContext: (imodel: IModelConnection) => SchemaContext;
+    imodelAccess?: IModelAccess;
+    hierarchyLevelSizeLimit?: number;
+}
+/** @internal */
+export declare function useIModelAccess({ imodel, getSchemaContext, imodelAccess: providedIModelAccess, hierarchyLevelSizeLimit }: UseIModelAccessProps): {
+    imodelAccess: IModelAccess;
+    currentHierarchyLevelSizeLimit: number;
+};
+export {};
+//# sourceMappingURL=UseIModelAccess.d.ts.map

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

@@ -0,0 +1,30 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { useMemo } from "react";
+import { createECSchemaProvider, createECSqlQueryExecutor } from "@itwin/presentation-core-interop";
+import { createLimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+import { createCachingECClassHierarchyInspector } from "@itwin/presentation-shared";
+/** @internal */
+export function useIModelAccess({ imodel, getSchemaContext, imodelAccess: providedIModelAccess, hierarchyLevelSizeLimit }) {
+    const defaultHierarchyLevelSizeLimit = hierarchyLevelSizeLimit ?? 1000;
+    const imodelAccess = useMemo(() => {
+        return providedIModelAccess ?? createIModelAccess({ getSchemaContext, imodel, hierarchyLevelSizeLimit: defaultHierarchyLevelSizeLimit });
+    }, [providedIModelAccess, getSchemaContext, imodel, defaultHierarchyLevelSizeLimit]);
+    return {
+        imodelAccess,
+        currentHierarchyLevelSizeLimit: defaultHierarchyLevelSizeLimit,
+    };
+}
+function createIModelAccess({ imodel, getSchemaContext, hierarchyLevelSizeLimit, }) {
+    const schemas = getSchemaContext(imodel);
+    const schemaProvider = createECSchemaProvider(schemas);
+    return {
+        imodelKey: imodel.key,
+        ...schemaProvider,
+        ...createCachingECClassHierarchyInspector({ schemaProvider, cacheSize: 100 }),
+        ...createLimitingECSqlQueryExecutor(createECSqlQueryExecutor(imodel), hierarchyLevelSizeLimit),
+    };
+}
+//# sourceMappingURL=UseIModelAccess.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"UseIModelAccess.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/UseIModelAccess.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AACpG,OAAO,EAAE,gCAAgC,EAAE,MAAM,iCAAiC,CAAC;AACnF,OAAO,EAAE,sCAAsC,EAAE,MAAM,4BAA4B,CAAC;AAgBpF,gBAAgB;AAChB,MAAM,UAAU,eAAe,CAAC,EAAC,MAAM,EAAE,gBAAgB,EAAE,YAAY,EAAE,oBAAoB,EAAE,uBAAuB,EAAuB;IAIzI,MAAM,8BAA8B,GAAG,uBAAuB,IAAI,IAAI,CAAC;IACvE,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,EAAE;QAChC,OAAO,oBAAoB,IAAI,kBAAkB,CAAC,EAAE,gBAAgB,EAAE,MAAM,EAAE,uBAAuB,EAAE,8BAA8B,EAAE,CAAC,CAAC;IAC3I,CAAC,EAAE,CAAC,oBAAoB,EAAE,gBAAgB,EAAE,MAAM,EAAE,8BAA8B,CAAC,CAAC,CAAC;IAErF,OAAO;QACL,YAAY;QACZ,8BAA8B,EAAE,8BAA8B;KAC/D,CAAA;AACL,CAAC;AAED,SAAS,kBAAkB,CAAC,EAC1B,MAAM,EACN,gBAAgB,EAChB,uBAAuB,GAKxB;IACC,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;IACzC,MAAM,cAAc,GAAG,sBAAsB,CAAC,OAAO,CAAC,CAAC;IACvD,OAAO;QACL,SAAS,EAAE,MAAM,CAAC,GAAG;QACrB,GAAG,cAAc;QACjB,GAAG,sCAAsC,CAAC,EAAE,cAAc,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC;QAC7E,GAAG,gCAAgC,CAAC,wBAAwB,CAAC,MAAM,CAAC,EAAE,uBAAuB,CAAC;KAC/F,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useMemo } from \"react\";\nimport { createECSchemaProvider, createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { createLimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport { createCachingECClassHierarchyInspector } from \"@itwin/presentation-shared\";\n\nimport type { IModelConnection } from \"@itwin/core-frontend\";\nimport type { SchemaContext } from \"@itwin/ecschema-metadata\";\nimport type { useIModelTree } from \"@itwin/presentation-hierarchies-react\";\nimport type { FunctionProps } from \"./Utils.js\";\n\ntype IModelAccess = FunctionProps<typeof useIModelTree>[\"imodelAccess\"];\n\nexport interface UseIModelAccessProps {\n    imodel: IModelConnection;\n    getSchemaContext: (imodel: IModelConnection) => SchemaContext;\n    imodelAccess?: IModelAccess;\n    hierarchyLevelSizeLimit?: number;\n}\n\n/** @internal */\nexport function useIModelAccess({imodel, getSchemaContext, imodelAccess: providedIModelAccess, hierarchyLevelSizeLimit}: UseIModelAccessProps): {\n  imodelAccess: IModelAccess;\n  currentHierarchyLevelSizeLimit: number;\n} {\n    const defaultHierarchyLevelSizeLimit = hierarchyLevelSizeLimit ?? 1000;\n    const imodelAccess = useMemo(() => {\n      return providedIModelAccess ?? createIModelAccess({ getSchemaContext, imodel, hierarchyLevelSizeLimit: defaultHierarchyLevelSizeLimit });\n    }, [providedIModelAccess, getSchemaContext, imodel, defaultHierarchyLevelSizeLimit]);\n\n    return {\n      imodelAccess,\n      currentHierarchyLevelSizeLimit: defaultHierarchyLevelSizeLimit,\n    }\n}\n\nfunction createIModelAccess({\n  imodel,\n  getSchemaContext,\n  hierarchyLevelSizeLimit,\n}: {\n  imodel: IModelConnection;\n  getSchemaContext: (imodel: IModelConnection) => SchemaContext;\n  hierarchyLevelSizeLimit: number;\n}) {\n  const schemas = getSchemaContext(imodel);\n  const schemaProvider = createECSchemaProvider(schemas);\n  return {\n    imodelKey: imodel.key,\n    ...schemaProvider,\n    ...createCachingECClassHierarchyInspector({ schemaProvider, cacheSize: 100 }),\n    ...createLimitingECSqlQueryExecutor(createECSqlQueryExecutor(imodel), hierarchyLevelSizeLimit),\n  };\n}\n"]}

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

@@ -1,7 +1,5 @@
 /// <reference types="react" />
 import type { Id64Array } from "@itwin/core-bentley";
-import type { IModelConnection } from "@itwin/core-frontend";
-import type { SchemaContext } from "@itwin/ecschema-metadata";
 /** @beta */
 export type FunctionProps<THook extends (props: any) => any> = Parameters<THook>[0];
 /** @internal */
@@ -11,17 +9,5 @@ export declare function parseIdsSelectorResult(selectorResult: any): Id64Array;
 /** @internal */
 export declare function pushToMap<TKey, TValue>(targetMap: Map<TKey, Set<TValue>>, key: TKey, value: TValue): void;
 /** @internal */
-export declare function createIModelAccess({ imodel, getSchemaContext }: {
-    imodel: IModelConnection;
-    getSchemaContext: (imodel: IModelConnection) => SchemaContext;
-}): {
-    createQueryReader(query: import("@itwin/presentation-shared").ECSqlQueryDef, config?: (import("@itwin/presentation-shared").ECSqlQueryReaderOptions & {
-        limit?: number | "unbounded" | undefined;
-    }) | undefined): AsyncIterableIterator<import("@itwin/presentation-shared").ECSqlQueryRow>;
-    classDerivesFrom(derivedClassFullName: string, candidateBaseClassFullName: string): boolean | Promise<boolean>;
-    getSchema(schemaName: string): Promise<import("@itwin/presentation-shared").EC.Schema | undefined>;
-    imodelKey: string;
-};
-/** @internal */
 export declare function useLatest<T>(value: T): import("react").MutableRefObject<T>;
 //# sourceMappingURL=Utils.d.ts.map