@itwin/tree-widget-react 4.0.0-alpha.6 → 4.0.0-alpha.8

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

@@ -2,41 +2,91 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { DEFINITION_CONTAINER_CLASS, SUB_CATEGORY_CLASS } from "./ClassNameDefinitions.js";
+import { DEFINITION_CONTAINER_CLASS_NAME, MODEL_CLASS_NAME, SUB_CATEGORY_CLASS_NAME } from "../../common/internal/ClassNameDefinitions.js";
+import { ModelCategoryElementsCountCache } from "../../common/internal/ModelCategoryElementsCountCache.js";
+import { getClassesByView, getDistinctMapValues } from "../../common/internal/Utils.js";
 /** @internal */
 export class CategoriesTreeIdsCache {
     constructor(_queryExecutor, viewType) {
         this._queryExecutor = _queryExecutor;
-        const { categoryClass, categoryElementClass } = getClassesByView(viewType);
+        const { categoryClass, elementClass, modelClass } = getClassesByView(viewType);
         this._categoryClass = categoryClass;
-        this._categoryElementClass = categoryElementClass;
+        this._categoryElementClass = elementClass;
+        this._categoryModelClass = modelClass;
+        this._categoryElementCounts = new ModelCategoryElementsCountCache(_queryExecutor, elementClass);
+    }
+    [Symbol.dispose]() {
+        this._categoryElementCounts[Symbol.dispose]();
+    }
+    async *queryFilteredElementsModels(filteredElementIds) {
+        const query = `
+      SELECT Model.Id modelId, ECInstanceId id
+      FROM ${this._categoryElementClass}
+      WHERE ECInstanceId IN (${filteredElementIds.join(", ")})
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { modelId: row.modelId, id: row.id };
+        }
+    }
+    async getFilteredElementsModels(filteredElementIds) {
+        if (filteredElementIds.length === 0) {
+            return new Map();
+        }
+        this._filteredElementsModels ??= (async () => {
+            const filteredElementsModels = new Map();
+            for await (const { modelId, id } of this.queryFilteredElementsModels(filteredElementIds)) {
+                filteredElementsModels.set(id, modelId);
+            }
+            return filteredElementsModels;
+        })();
+        return this._filteredElementsModels;
+    }
+    clearFilteredElementsModels() {
+        this._filteredElementsModels = undefined;
+    }
+    async *queryElementModelCategories() {
+        const query = `
+      SELECT this.Model.Id modelId, this.Category.Id categoryId
+      FROM ${this._categoryModelClass} m
+      JOIN ${this._categoryElementClass} this ON m.ECInstanceId = this.Model.Id
+      WHERE this.Parent.Id IS NULL AND m.IsPrivate = false
+      GROUP BY modelId, categoryId
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: "tree-widget/categories-tree/element-models-and-categories-query" })) {
+            yield { modelId: row.modelId, categoryId: row.categoryId };
+        }
     }
     async *queryCategories() {
         const isDefinitionContainerSupported = await this.getIsDefinitionContainerSupported();
         const categoriesQuery = `
       SELECT
         this.ECInstanceId id,
-        COUNT(sc.ECInstanceId) childCount,
+        COUNT(sc.ECInstanceId) subCategoryChildCount,
         this.Model.Id modelId,
         ${isDefinitionContainerSupported
             ? `
-            IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${DEFINITION_CONTAINER_CLASS} dc),
+            IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${DEFINITION_CONTAINER_CLASS_NAME} dc),
               true,
               false
             )`
             : "false"} parentDefinitionContainerExists
       FROM
         ${this._categoryClass} this
-        JOIN ${SUB_CATEGORY_CLASS} sc ON sc.Parent.Id = this.ECInstanceId
-        JOIN BisCore.Model m ON m.ECInstanceId = this.Model.Id
+        JOIN ${SUB_CATEGORY_CLASS_NAME} sc ON sc.Parent.Id = this.ECInstanceId
+        JOIN ${MODEL_CLASS_NAME} m ON m.ECInstanceId = this.Model.Id
       WHERE
         NOT this.IsPrivate
         AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))
         AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)
       GROUP BY this.ECInstanceId
     `;
-        for await (const row of this._queryExecutor.createQueryReader({ ecsql: categoriesQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
-            yield { id: row.id, modelId: row.modelId, parentDefinitionContainerExists: row.parentDefinitionContainerExists, childCount: row.childCount };
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: categoriesQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: "tree-widget/categories-tree/categories-query" })) {
+            yield {
+                id: row.id,
+                modelId: row.modelId,
+                parentDefinitionContainerExists: row.parentDefinitionContainerExists,
+                subCategoryChildCount: row.subCategoryChildCount,
+            };
         }
     }
     async queryIsDefinitionContainersSupported() {
@@ -50,7 +100,7 @@ export class CategoriesTreeIdsCache {
         s.Name = 'BisCore'
         AND c.Name = 'DefinitionContainer'
     `;
-        for await (const _row of this._queryExecutor.createQueryReader({ ecsql: query })) {
+        for await (const _row of this._queryExecutor.createQueryReader({ ecsql: query }, { restartToken: "tree-widget/categories-tree/is-definition-container-supported-query" })) {
             return true;
         }
         return false;
@@ -68,9 +118,9 @@ export class CategoriesTreeIdsCache {
             dc.ECInstanceId,
             dc.Model.Id
           FROM
-            ${DEFINITION_CONTAINER_CLASS} dc
+            ${DEFINITION_CONTAINER_CLASS_NAME} dc
           WHERE
-            dc.ECInstanceId IN (SELECT c.Model.Id FROM ${this._categoryClass} c WHERE NOT c.IsPrivate)
+            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))
             AND NOT dc.IsPrivate
 
           UNION ALL
@@ -80,7 +130,7 @@ export class CategoriesTreeIdsCache {
             pdc.Model.Id
           FROM
             ${DEFINITION_CONTAINERS_CTE} cdc
-            JOIN ${DEFINITION_CONTAINER_CLASS} pdc ON pdc.ECInstanceId = cdc.ModelId
+            JOIN ${DEFINITION_CONTAINER_CLASS_NAME} pdc ON pdc.ECInstanceId = cdc.ModelId
           WHERE
             NOT pdc.IsPrivate
         )
@@ -89,22 +139,22 @@ export class CategoriesTreeIdsCache {
         const definitionsQuery = `
       SELECT dc.ECInstanceId id, dc.ModelId modelId FROM ${DEFINITION_CONTAINERS_CTE} dc GROUP BY dc.ECInstanceId
     `;
-        for await (const row of this._queryExecutor.createQueryReader({ ctes, ecsql: definitionsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+        for await (const row of this._queryExecutor.createQueryReader({ ctes, ecsql: definitionsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: "tree-widget/categories-tree/definition-containers-query" })) {
             yield { id: row.id, modelId: row.modelId };
         }
     }
-    async *queryVisibleSubCategories(categoriesInfo) {
+    async *queryVisibleSubCategories(categoryIds) {
         const definitionsQuery = `
       SELECT
         sc.ECInstanceId id,
         sc.Parent.Id categoryId
       FROM
-        ${SUB_CATEGORY_CLASS} sc
+        ${SUB_CATEGORY_CLASS_NAME} sc
       WHERE
         NOT sc.IsPrivate
-        AND sc.Parent.Id IN (${categoriesInfo.join(",")})
+        AND sc.Parent.Id IN (${categoryIds.join(",")})
     `;
-        for await (const row of this._queryExecutor.createQueryReader({ ecsql: definitionsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: definitionsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: "tree-widget/categories-tree/sub-categories-query" })) {
             yield { id: row.id, parentId: row.categoryId };
         }
     }
@@ -117,19 +167,90 @@ export class CategoriesTreeIdsCache {
                     modelCategories = { parentDefinitionContainerExists: queriedCategory.parentDefinitionContainerExists, childCategories: [] };
                     allModelsCategories.set(queriedCategory.modelId, modelCategories);
                 }
-                modelCategories.childCategories.push({ id: queriedCategory.id, childCount: queriedCategory.childCount });
+                modelCategories.childCategories.push({ id: queriedCategory.id, subCategoryChildCount: queriedCategory.subCategoryChildCount });
             }
             return allModelsCategories;
         })();
         return this._modelsCategoriesInfo;
     }
+    async getElementModelsCategories() {
+        this._elementModelsCategories ??= (async () => {
+            const [modelCategories, modelWithCategoryModeledElements] = await Promise.all([
+                (async () => {
+                    const elementModelsCategories = new Map();
+                    for await (const queriedCategory of this.queryElementModelCategories()) {
+                        let modelEntry = elementModelsCategories.get(queriedCategory.modelId);
+                        if (modelEntry === undefined) {
+                            modelEntry = { categoryIds: new Set() };
+                            elementModelsCategories.set(queriedCategory.modelId, modelEntry);
+                        }
+                        modelEntry.categoryIds.add(queriedCategory.categoryId);
+                    }
+                    return elementModelsCategories;
+                })(),
+                this.getModelWithCategoryModeledElements(),
+            ]);
+            const result = new Map();
+            const subModels = getDistinctMapValues(modelWithCategoryModeledElements);
+            for (const [modelId, modelEntry] of modelCategories) {
+                const isSubModel = subModels.has(modelId);
+                result.set(modelId, { categoryIds: modelEntry.categoryIds, isSubModel });
+            }
+            return result;
+        })();
+        return this._elementModelsCategories;
+    }
+    async *queryModeledElements() {
+        const query = `
+      SELECT
+        pe.ECInstanceId modeledElementId,
+        pe.Category.Id categoryId,
+        pe.Model.Id modelId
+      FROM ${this._categoryModelClass} m
+      JOIN ${this._categoryElementClass} pe ON pe.ECInstanceId = m.ModeledElement.Id
+      WHERE
+        m.IsPrivate = false
+        AND m.ECInstanceId IN (SELECT Model.Id FROM ${this._categoryElementClass})
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: "tree-widget/categories-tree/modeled-elements-query" })) {
+            yield { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };
+        }
+    }
+    async getModelWithCategoryModeledElements() {
+        this._modelWithCategoryModeledElements ??= (async () => {
+            const modelWithCategoryModeledElements = new Map();
+            for await (const { modelId, categoryId, modeledElementId } of this.queryModeledElements()) {
+                const key = `${modelId}-${categoryId}`;
+                const entry = modelWithCategoryModeledElements.get(key);
+                if (entry === undefined) {
+                    modelWithCategoryModeledElements.set(key, new Set([modeledElementId]));
+                }
+                else {
+                    entry.add(modeledElementId);
+                }
+            }
+            return modelWithCategoryModeledElements;
+        })();
+        return this._modelWithCategoryModeledElements;
+    }
+    async getCategoriesModeledElements(modelId, categoryIds) {
+        const modelWithCategoryModeledElements = await this.getModelWithCategoryModeledElements();
+        const result = new Array();
+        for (const categoryId of categoryIds) {
+            const entry = modelWithCategoryModeledElements.get(`${modelId}-${categoryId}`);
+            if (entry !== undefined) {
+                result.push(...entry);
+            }
+        }
+        return result;
+    }
     async getSubCategoriesInfo() {
         this._subCategoriesInfo ??= (async () => {
             const allSubCategories = new Map();
             const modelsCategoriesInfo = await this.getModelsCategoriesInfo();
             const categoriesWithMoreThanOneSubCategory = new Array();
             for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {
-                categoriesWithMoreThanOneSubCategory.push(...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.childCount > 1).map((categoryInfo) => categoryInfo.id));
+                categoriesWithMoreThanOneSubCategory.push(...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.subCategoryChildCount > 1).map((categoryInfo) => categoryInfo.id));
             }
             if (categoriesWithMoreThanOneSubCategory.length === 0) {
                 return allSubCategories;
@@ -156,14 +277,14 @@ export class CategoriesTreeIdsCache {
                 definitionContainersInfo.set(queriedDefinitionContainer.id, {
                     childCategories: modelCategoriesInfo?.childCategories ?? [],
                     modelId: queriedDefinitionContainer.modelId,
-                    childDefinitionContainers: [],
+                    childDefinitionContainerIds: [],
                     parentDefinitionContainerExists: false,
                 });
             }
             for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {
                 const parentDefinitionContainer = definitionContainersInfo.get(definitionContainerInfo.modelId);
                 if (parentDefinitionContainer !== undefined) {
-                    parentDefinitionContainer.childDefinitionContainers.push(definitionContainerId);
+                    parentDefinitionContainer.childDefinitionContainerIds.push(definitionContainerId);
                     definitionContainerInfo.parentDefinitionContainerExists = true;
                 }
             }
@@ -177,12 +298,37 @@ export class CategoriesTreeIdsCache {
         parentDefinitionContainerIds.forEach((parentDefinitionContainerId) => {
             const parentDefinitionContainerInfo = definitionContainersInfo.get(parentDefinitionContainerId);
             if (parentDefinitionContainerInfo !== undefined) {
-                result.definitionContainers.push(...parentDefinitionContainerInfo.childDefinitionContainers);
+                result.definitionContainers.push(...parentDefinitionContainerInfo.childDefinitionContainerIds);
                 result.categories.push(...parentDefinitionContainerInfo.childCategories);
             }
         });
         return result;
     }
+    async getCategoriesElementModels(categoryIds, includeSubModels) {
+        const elementModelsCategories = await this.getElementModelsCategories();
+        const result = new Map();
+        for (const categoryId of categoryIds) {
+            for (const [modelId, { categoryIds: categories, isSubModel }] of elementModelsCategories) {
+                if ((includeSubModels || !isSubModel) && categories.has(categoryId)) {
+                    let categoryModels = result.get(categoryId);
+                    if (!categoryModels) {
+                        categoryModels = new Array();
+                        result.set(categoryId, categoryModels);
+                    }
+                    categoryModels.push(modelId);
+                }
+            }
+        }
+        return result;
+    }
+    async getModelCategoryIds(modelId) {
+        const elementModelsCategories = await this.getElementModelsCategories();
+        return [...(elementModelsCategories.get(modelId)?.categoryIds ?? [])];
+    }
+    async hasSubModel(elementId) {
+        const elementModelsCategories = await this.getElementModelsCategories();
+        return elementModelsCategories.has(elementId);
+    }
     async getAllContainedCategories(definitionContainerIds) {
         const result = new Array();
         const definitionContainersInfo = await this.getDefinitionContainersInfo();
@@ -192,7 +338,7 @@ export class CategoriesTreeIdsCache {
                 return [];
             }
             result.push(...definitionContainerInfo.childCategories.map((category) => category.id));
-            return this.getAllContainedCategories(definitionContainerInfo.childDefinitionContainers);
+            return this.getAllContainedCategories(definitionContainerInfo.childDefinitionContainerIds);
         }));
         for (const categories of indirectCategories) {
             result.push(...categories);
@@ -206,7 +352,7 @@ export class CategoriesTreeIdsCache {
             if (subCategoryInfo === undefined) {
                 return [];
             }
-            return [...(await this.getInstanceKeyPaths({ categoryId: subCategoryInfo.categoryId })), { id: props.subCategoryId, className: SUB_CATEGORY_CLASS }];
+            return [...(await this.getInstanceKeyPaths({ categoryId: subCategoryInfo.categoryId })), { id: props.subCategoryId, className: SUB_CATEGORY_CLASS_NAME }];
         }
         if ("categoryId" in props) {
             const modelsCategoriesInfo = await this.getModelsCategoriesInfo();
@@ -226,13 +372,16 @@ export class CategoriesTreeIdsCache {
             return [];
         }
         if (!definitionContainerInfo.parentDefinitionContainerExists) {
-            return [{ id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS }];
+            return [{ id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS_NAME }];
         }
         return [
             ...(await this.getInstanceKeyPaths({ definitionContainerId: definitionContainerInfo.modelId })),
-            { id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS },
+            { id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS_NAME },
         ];
     }
+    async getCategoryElementsCount(modelId, categoryId) {
+        return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);
+    }
     async getAllDefinitionContainersAndCategories() {
         const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);
         const result = { definitionContainers: [...definitionContainersInfo.keys()], categories: new Array() };
@@ -256,12 +405,19 @@ export class CategoriesTreeIdsCache {
         }
         return result;
     }
-    async getSubCategories(categoryId) {
+    async getSubCategories(categoryIds) {
         const subCategoriesInfo = await this.getSubCategoriesInfo();
-        const result = new Array();
-        for (const [subCategoryId, subCategoryInfo] of subCategoriesInfo) {
-            if (subCategoryInfo.categoryId === categoryId) {
-                result.push(subCategoryId);
+        const result = new Map();
+        for (const categoryId of categoryIds) {
+            for (const [subCategoryId, subCategoryInfo] of subCategoriesInfo) {
+                if (subCategoryInfo.categoryId === categoryId) {
+                    let categoryEntry = result.get(categoryId);
+                    if (!categoryEntry) {
+                        categoryEntry = [];
+                        result.set(categoryId, categoryEntry);
+                    }
+                    categoryEntry.push(subCategoryId);
+                }
             }
         }
         return result;
@@ -271,10 +427,4 @@ export class CategoriesTreeIdsCache {
         return this._isDefinitionContainerSupported;
     }
 }
-/** @internal */
-export function getClassesByView(viewType) {
-    return viewType === "2d"
-        ? { categoryClass: "BisCore.DrawingCategory", categoryElementClass: "BisCore.GeometricElement2d" }
-        : { categoryClass: "BisCore.SpatialCategory", categoryElementClass: "BisCore.GeometricElement3d" };
-}
 //# sourceMappingURL=CategoriesTreeIdsCache.js.map

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;;;;;;;;;cAS9D,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)\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,+BAA+B,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,MAAM,+CAA+C,CAAC;AAC3I,OAAO,EAAE,+BAA+B,EAAE,MAAM,0DAA0D,CAAC;AAC3G,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,gCAAgC,CAAC;AA+BxF,gBAAgB;AAChB,MAAM,OAAO,sBAAsB;IAajC,YACU,cAA0C,EAClD,QAAqB;QADb,mBAAc,GAAd,cAAc,CAA4B;QAGlD,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC/E,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,qBAAqB,GAAG,YAAY,CAAC;QAC1C,IAAI,CAAC,mBAAmB,GAAG,UAAU,CAAC;QACtC,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,cAAc,EAAE,YAAY,CAAC,CAAC;IAClG,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,2BAA2B,CAAC,kBAA6B;QAItE,MAAM,KAAK,GAAG;;aAEL,IAAI,CAAC,qBAAqB;+BACR,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC;KACvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACxI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC;SAC5C;IACH,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,kBAA6B;QAClE,IAAI,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE;YACnC,OAAO,IAAI,GAAG,EAAsB,CAAC;SACtC;QAED,IAAI,CAAC,uBAAuB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC3C,MAAM,sBAAsB,GAAG,IAAI,GAAG,EAAE,CAAC;YACzC,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,IAAI,CAAC,2BAA2B,CAAC,kBAAkB,CAAC,EAAE;gBACxF,sBAAsB,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;aACzC;YACD,OAAO,sBAAsB,CAAC;QAChC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;IAEM,2BAA2B;QAChC,IAAI,CAAC,uBAAuB,GAAG,SAAS,CAAC;IAC3C,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,+BAA+B;;;cAGjF;YACF,CAAC,CAAC,OACN;;UAEE,IAAI,CAAC,cAAc;eACd,uBAAuB;eACvB,gBAAgB;;;;oCAIK,IAAI,CAAC,qBAAqB;;KAEzD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,eAAe,EAAE,EAC1B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,8CAA8C,EAAE,CACtH,EAAE;YACD,MAAM;gBACJ,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,+BAA+B,EAAE,GAAG,CAAC,+BAA+B;gBACpE,qBAAqB,EAAE,GAAG,CAAC,qBAAqB;aACjD,CAAC;SACH;IACH,CAAC;IAEO,KAAK,CAAC,oCAAoC;QAChD,MAAM,KAAK,GAAG;;;;;;;;;KASb,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC5D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,YAAY,EAAE,qEAAqE,EAAE,CACxF,EAAE;YACD,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,+BAA+B;;yDAEY,IAAI,CAAC,cAAc,sDAAsD,IAAI,CAAC,qBAAqB;;;;;;;;;cAS9I,yBAAyB;mBACpB,+BAA+B;;;;OAI3C;SACF,CAAC;QACF,MAAM,gBAAgB,GAAG;2DAC8B,yBAAyB;KAC/E,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,EACjC,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,yDAAyD,EAAE,CACjI,EAAE;YACD,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;SAC5C;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB,CAAC,WAAsB;QAC7D,MAAM,gBAAgB,GAAG;;;;;UAKnB,uBAAuB;;;+BAGF,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC;KAC/C,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAC3B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,kDAAkD,EAAE,CAC1H,EAAE;YACD,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,EAA2B,CAAC;YAC/D,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,qBAAqB,EAAE,eAAe,CAAC,qBAAqB,EAAE,CAAC,CAAC;aAChI;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,CAAC,eAAe,EAAE,gCAAgC,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC5E,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAAsC,CAAC;oBAC9E,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,2BAA2B,EAAE,EAAE;wBACtE,IAAI,UAAU,GAAG,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;wBACtE,IAAI,UAAU,KAAK,SAAS,EAAE;4BAC5B,UAAU,GAAG,EAAE,WAAW,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;4BACxC,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;yBAClE;wBACD,UAAU,CAAC,WAAW,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;qBACxD;oBACD,OAAO,uBAAuB,CAAC;gBACjC,CAAC,CAAC,EAAE;gBACJ,IAAI,CAAC,mCAAmC,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,EAAmE,CAAC;YAC1F,MAAM,SAAS,GAAG,oBAAoB,CAAC,gCAAgC,CAAC,CAAC;YACzE,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,eAAe,EAAE;gBACnD,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC1C,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,UAAU,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC;aAC1E;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,wBAAwB,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QAMjC,MAAM,KAAK,GAAG;;;;;aAKL,IAAI,CAAC,mBAAmB;aACxB,IAAI,CAAC,qBAAqB;;;sDAGe,IAAI,CAAC,qBAAqB;KAC3E,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,oDAAoD,EAAE,CAC5H,EAAE;YACD,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,cAAc,EAAE,GAAG,CAAC,cAAc,EAAE,CAAC;SACxI;IACH,CAAC;IAEO,KAAK,CAAC,mCAAmC;QAC/C,IAAI,CAAC,iCAAiC,KAAK,CAAC,KAAK,IAAI,EAAE;YACrD,MAAM,gCAAgC,GAAG,IAAI,GAAG,EAAoC,CAAC;YACrF,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE;gBACzF,MAAM,GAAG,GAAqB,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;gBACzD,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACxD,IAAI,KAAK,KAAK,SAAS,EAAE;oBACvB,gCAAgC,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;iBACxE;qBAAM;oBACL,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;iBAC7B;aACF;YACD,OAAO,gCAAgC,CAAC;QAC1C,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iCAAiC,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,4BAA4B,CAAC,OAAmB,EAAE,WAAsB;QACnF,MAAM,gCAAgC,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;QAC1F,MAAM,MAAM,GAAG,IAAI,KAAK,EAAa,CAAC;QACtC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE;YACpC,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;YAC/E,IAAI,KAAK,KAAK,SAAS,EAAE;gBACvB,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;aACvB;SACF;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,oBAAoB;QAChC,IAAI,CAAC,kBAAkB,KAAK,CAAC,KAAK,IAAI,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAkC,CAAC;YACnE,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,qBAAqB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC,CAC/I,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,EAAkD,CAAC;YAC3F,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,2BAA2B,EAAE,EAAE;oBAC/B,+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,2BAA2B,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;oBAClF,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,EAAyB,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QAEnH,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,2BAA2B,CAAC,CAAC;gBAC/F,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,EAAE,gBAA0B;QACxF,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,MAAM,MAAM,GAAG,IAAI,GAAG,EAA8B,CAAC;QACrD,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE;YACpC,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,IAAI,uBAAuB,EAAE;gBACxF,IAAI,CAAC,gBAAgB,IAAI,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;oBACnE,IAAI,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC5C,IAAI,CAAC,cAAc,EAAE;wBACnB,cAAc,GAAG,IAAI,KAAK,EAAW,CAAC;wBACtC,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,mBAAmB,CAAC,OAAmB;QAClD,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC;IACxE,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,SAAqB;QAC5C,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,OAAO,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAChD,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,2BAA2B,CAAC,CAAC;QAC7F,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,uBAAuB,EAAE,CAAC,CAAC;SAC3J;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,+BAA+B,EAAE,CAAC,CAAC;SAC1F;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,+BAA+B,EAAE;SAChF,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,KAAK,CAAC,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,WAAsB;QAClD,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAoC,CAAC;QAC3D,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE;YACpC,KAAK,MAAM,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,iBAAiB,EAAE;gBAChE,IAAI,eAAe,CAAC,UAAU,KAAK,UAAU,EAAE;oBAC7C,IAAI,aAAa,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC3C,IAAI,CAAC,aAAa,EAAE;wBAClB,aAAa,GAAG,EAAE,CAAC;wBACnB,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;qBACvC;oBACD,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;iBACnC;aACF;SACF;QAED,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","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_NAME, MODEL_CLASS_NAME, SUB_CATEGORY_CLASS_NAME } from \"../../common/internal/ClassNameDefinitions.js\";\nimport { ModelCategoryElementsCountCache } from \"../../common/internal/ModelCategoryElementsCountCache.js\";\nimport { getClassesByView, getDistinctMapValues } from \"../../common/internal/Utils.js\";\n\nimport type { CategoryId, DefinitionContainerId, ElementId, ModelId, SubCategoryId } from \"../../common/internal/Types.js\";\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  childDefinitionContainerIds: Id64Array;\n}\n\ninterface CategoriesInfo {\n  childCategories: CategoryInfo[];\n  parentDefinitionContainerExists: boolean;\n}\n\n/** @internal */\nexport interface CategoryInfo {\n  id: CategoryId;\n  subCategoryChildCount: number;\n}\n\ninterface SubCategoryInfo {\n  categoryId: Id64String;\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\n/** @internal */\nexport class CategoriesTreeIdsCache implements Disposable {\n  private _definitionContainersInfo: Promise<Map<DefinitionContainerId, DefinitionContainerInfo>> | undefined;\n  private _modelsCategoriesInfo: Promise<Map<ModelId, CategoriesInfo>> | undefined;\n  private _elementModelsCategories: Promise<Map<ModelId, { categoryIds: Id64Set; isSubModel: boolean }>> | undefined;\n  private _subCategoriesInfo: Promise<Map<SubCategoryId, SubCategoryInfo>> | undefined;\n  private readonly _categoryElementCounts: ModelCategoryElementsCountCache;\n  private _modelWithCategoryModeledElements: Promise<Map<ModelCategoryKey, Set<ElementId>>> | undefined;\n  private _categoryClass: string;\n  private _categoryElementClass: string;\n  private _categoryModelClass: string;\n  private _isDefinitionContainerSupported: Promise<boolean> | undefined;\n  private _filteredElementsModels: Promise<Map<ElementId, ModelId>> | undefined;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    viewType: \"3d\" | \"2d\",\n  ) {\n    const { categoryClass, elementClass, modelClass } = getClassesByView(viewType);\n    this._categoryClass = categoryClass;\n    this._categoryElementClass = elementClass;\n    this._categoryModelClass = modelClass;\n    this._categoryElementCounts = new ModelCategoryElementsCountCache(_queryExecutor, elementClass);\n  }\n\n  public [Symbol.dispose]() {\n    this._categoryElementCounts[Symbol.dispose]();\n  }\n\n  private async *queryFilteredElementsModels(filteredElementIds: Id64Array): AsyncIterableIterator<{\n    modelId: Id64String;\n    id: ElementId;\n  }> {\n    const query = `\n      SELECT Model.Id modelId, ECInstanceId id\n      FROM ${this._categoryElementClass}\n      WHERE ECInstanceId IN (${filteredElementIds.join(\", \")})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, id: row.id };\n    }\n  }\n\n  public async getFilteredElementsModels(filteredElementIds: Id64Array) {\n    if (filteredElementIds.length === 0) {\n      return new Map<ElementId, ModelId>();\n    }\n\n    this._filteredElementsModels ??= (async () => {\n      const filteredElementsModels = new Map();\n      for await (const { modelId, id } of this.queryFilteredElementsModels(filteredElementIds)) {\n        filteredElementsModels.set(id, modelId);\n      }\n      return filteredElementsModels;\n    })();\n    return this._filteredElementsModels;\n  }\n\n  public clearFilteredElementsModels() {\n    this._filteredElementsModels = undefined;\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 this.Parent.Id IS NULL AND 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: CategoryId;\n    modelId: Id64String;\n    parentDefinitionContainerExists: boolean;\n    subCategoryChildCount: number;\n  }> {\n    const isDefinitionContainerSupported = await this.getIsDefinitionContainerSupported();\n    const categoriesQuery = `\n      SELECT\n        this.ECInstanceId id,\n        COUNT(sc.ECInstanceId) subCategoryChildCount,\n        this.Model.Id modelId,\n        ${\n          isDefinitionContainerSupported\n            ? `\n            IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${DEFINITION_CONTAINER_CLASS_NAME} dc),\n              true,\n              false\n            )`\n            : \"false\"\n        } parentDefinitionContainerExists\n      FROM\n        ${this._categoryClass} this\n        JOIN ${SUB_CATEGORY_CLASS_NAME} sc ON sc.Parent.Id = this.ECInstanceId\n        JOIN ${MODEL_CLASS_NAME} 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(\n      { ecsql: categoriesQuery },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/categories-query\" },\n    )) {\n      yield {\n        id: row.id,\n        modelId: row.modelId,\n        parentDefinitionContainerExists: row.parentDefinitionContainerExists,\n        subCategoryChildCount: row.subCategoryChildCount,\n      };\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(\n      { ecsql: query },\n      { restartToken: \"tree-widget/categories-tree/is-definition-container-supported-query\" },\n    )) {\n      return true;\n    }\n    return false;\n  }\n\n  private async *queryDefinitionContainers(): AsyncIterableIterator<{ id: DefinitionContainerId; 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_NAME} 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_NAME} 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(\n      { ctes, ecsql: definitionsQuery },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/definition-containers-query\" },\n    )) {\n      yield { id: row.id, modelId: row.modelId };\n    }\n  }\n\n  private async *queryVisibleSubCategories(categoryIds: Id64Array): AsyncIterableIterator<{ id: SubCategoryId; parentId: CategoryId }> {\n    const definitionsQuery = `\n      SELECT\n        sc.ECInstanceId id,\n        sc.Parent.Id categoryId\n      FROM\n        ${SUB_CATEGORY_CLASS_NAME} sc\n      WHERE\n        NOT sc.IsPrivate\n        AND sc.Parent.Id IN (${categoryIds.join(\",\")})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: definitionsQuery },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/sub-categories-query\" },\n    )) {\n      yield { id: row.id, parentId: row.categoryId };\n    }\n  }\n\n  private async getModelsCategoriesInfo() {\n    this._modelsCategoriesInfo ??= (async () => {\n      const allModelsCategories = new Map<ModelId, 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, subCategoryChildCount: queriedCategory.subCategoryChildCount });\n      }\n      return allModelsCategories;\n    })();\n    return this._modelsCategoriesInfo;\n  }\n\n  private async getElementModelsCategories() {\n    this._elementModelsCategories ??= (async () => {\n      const [modelCategories, modelWithCategoryModeledElements] = await Promise.all([\n        (async () => {\n          const elementModelsCategories = new Map<ModelId, { categoryIds: Id64Set; }>();\n          for await (const queriedCategory of this.queryElementModelCategories()) {\n            let modelEntry = elementModelsCategories.get(queriedCategory.modelId);\n            if (modelEntry === undefined) {\n              modelEntry = { categoryIds: new Set() };\n              elementModelsCategories.set(queriedCategory.modelId, modelEntry);\n            }\n            modelEntry.categoryIds.add(queriedCategory.categoryId);\n          }\n          return elementModelsCategories;\n        })(),\n        this.getModelWithCategoryModeledElements(),\n      ]);\n      const result = new Map<ModelId, { categoryIds: Set<CategoryId>; isSubModel: boolean; }>();\n      const subModels = getDistinctMapValues(modelWithCategoryModeledElements);\n      for (const [modelId, modelEntry] of modelCategories) {\n        const isSubModel = subModels.has(modelId);\n        result.set(modelId, { categoryIds: modelEntry.categoryIds, isSubModel });\n      }\n      return result;\n    })();\n    return this._elementModelsCategories;\n  }\n\n  private async *queryModeledElements(): AsyncIterableIterator<{\n    modelId: Id64String;\n    modeledElementId: Id64String;\n    categoryId: Id64String;\n    rootCategoryId: Id64String;\n  }> {\n    const query = `\n      SELECT\n        pe.ECInstanceId modeledElementId,\n        pe.Category.Id categoryId,\n        pe.Model.Id modelId\n      FROM ${this._categoryModelClass} m\n      JOIN ${this._categoryElementClass} pe ON pe.ECInstanceId = m.ModeledElement.Id\n      WHERE\n        m.IsPrivate = false\n        AND m.ECInstanceId IN (SELECT Model.Id FROM ${this._categoryElementClass})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/categories-tree/modeled-elements-query\" },\n    )) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };\n    }\n  }\n\n  private async getModelWithCategoryModeledElements() {\n    this._modelWithCategoryModeledElements ??= (async () => {\n      const modelWithCategoryModeledElements = new Map<ModelCategoryKey, Set<ElementId>>();\n      for await (const { modelId, categoryId, modeledElementId } of this.queryModeledElements()) {\n        const key: ModelCategoryKey = `${modelId}-${categoryId}`;\n        const entry = modelWithCategoryModeledElements.get(key);\n        if (entry === undefined) {\n          modelWithCategoryModeledElements.set(key, new Set([modeledElementId]));\n        } else {\n          entry.add(modeledElementId);\n        }\n      }\n      return modelWithCategoryModeledElements;\n    })();\n    return this._modelWithCategoryModeledElements;\n  }\n\n  public async getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Array): Promise<Id64Array> {\n    const modelWithCategoryModeledElements = await this.getModelWithCategoryModeledElements();\n    const result = new Array<ElementId>();\n    for (const categoryId of categoryIds) {\n      const entry = modelWithCategoryModeledElements.get(`${modelId}-${categoryId}`);\n      if (entry !== undefined) {\n        result.push(...entry);\n      }\n    }\n    return result;\n  }\n\n  private async getSubCategoriesInfo() {\n    this._subCategoriesInfo ??= (async () => {\n      const allSubCategories = new Map<SubCategoryId, SubCategoryInfo>();\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      const categoriesWithMoreThanOneSubCategory = new Array<CategoryId>();\n      for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n        categoriesWithMoreThanOneSubCategory.push(\n          ...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.subCategoryChildCount > 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<DefinitionContainerId, 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          childDefinitionContainerIds: [],\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.childDefinitionContainerIds.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: Array<DefinitionContainerId> }> {\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n\n    const result = { definitionContainers: new Array<DefinitionContainerId>(), categories: new Array<CategoryInfo>() };\n\n    parentDefinitionContainerIds.forEach((parentDefinitionContainerId) => {\n      const parentDefinitionContainerInfo = definitionContainersInfo.get(parentDefinitionContainerId);\n      if (parentDefinitionContainerInfo !== undefined) {\n        result.definitionContainers.push(...parentDefinitionContainerInfo.childDefinitionContainerIds);\n        result.categories.push(...parentDefinitionContainerInfo.childCategories);\n      }\n    });\n    return result;\n  }\n\n  public async getCategoriesElementModels(categoryIds: Id64Array, includeSubModels?: boolean): Promise<Map<CategoryId, Array<ModelId>>> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    const result = new Map<CategoryId, Array<ModelId>>();\n    for (const categoryId of categoryIds) {\n      for (const [modelId, { categoryIds: categories, isSubModel }] of elementModelsCategories) {\n        if ((includeSubModels || !isSubModel) && categories.has(categoryId)) {\n          let categoryModels = result.get(categoryId);\n          if (!categoryModels) {\n            categoryModels = new Array<ModelId>();\n            result.set(categoryId, categoryModels);\n          }\n          categoryModels.push(modelId);\n        }\n      }\n    }\n    return result;\n  }\n\n  public async getModelCategoryIds(modelId: Id64String): Promise<Id64Array> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    return [...(elementModelsCategories.get(modelId)?.categoryIds ?? [])];\n  }\n\n  public async hasSubModel(elementId: Id64String): Promise<boolean> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    return elementModelsCategories.has(elementId);\n  }\n\n  public async getAllContainedCategories(definitionContainerIds: Id64Array): Promise<Id64Array> {\n    const result = new Array<CategoryId>();\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.childDefinitionContainerIds);\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_NAME }];\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_NAME }];\n    }\n\n    return [\n      ...(await this.getInstanceKeyPaths({ definitionContainerId: definitionContainerInfo.modelId })),\n      { id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS_NAME },\n    ];\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  public async getAllDefinitionContainersAndCategories(): Promise<{ categories: Array<CategoryId>; definitionContainers: Array<DefinitionContainerId> }> {\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: Array<DefinitionContainerId> }> {\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(categoryIds: Id64Array): Promise<Map<CategoryId, Array<SubCategoryId>>> {\n    const subCategoriesInfo = await this.getSubCategoriesInfo();\n    const result = new Map<CategoryId, Array<SubCategoryId>>();\n    for (const categoryId of categoryIds) {\n      for (const [subCategoryId, subCategoryInfo] of subCategoriesInfo) {\n        if (subCategoryInfo.categoryId === categoryId) {\n          let categoryEntry = result.get(categoryId);\n          if (!categoryEntry) {\n            categoryEntry = [];\n            result.set(categoryId, categoryEntry);\n          }\n          categoryEntry.push(subCategoryId);\n        }\n      }\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"]}

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

@@ -1,3 +1,4 @@
+import type { Id64String } from "@itwin/core-bentley";
 import type { HierarchyNodeKey } from "@itwin/presentation-hierarchies";
 interface CategoriesTreeNode {
     key: HierarchyNodeKey;
@@ -5,9 +6,7 @@ interface CategoriesTreeNode {
         [id: string]: any;
     };
 }
-/**
- * @internal
- */
+/** @internal */
 export declare namespace CategoriesTreeNode {
     /**
      * Determines if node represents a definition container.
@@ -17,10 +16,26 @@ export declare namespace CategoriesTreeNode {
      * Determines if node represents a category.
      */
     const isCategoryNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+    /**
+     * Determines if node represents a model.
+     */
+    const isModelNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+    /**
+     * Determines if node represents an element.
+     */
+    const isElementNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
     /**
      * Determines if node represents a sub-category.
      */
     const isSubCategoryNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+    /**
+     * Retrieves model ID from node's extended data.
+     */
+    const getModelId: (node: Pick<CategoriesTreeNode, "extendedData">) => Id64String | undefined;
+    /**
+     * Retrieves category ID from node's extended data.
+     */
+    const getCategoryId: (node: Pick<CategoriesTreeNode, "extendedData">) => Id64String | undefined;
 }
 export {};
 //# sourceMappingURL=CategoriesTreeNode.d.ts.map

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

@@ -2,9 +2,7 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-/**
- * @internal
- */
+/** @internal */
 export var CategoriesTreeNode;
 (function (CategoriesTreeNode) {
     /**
@@ -15,9 +13,38 @@ export var CategoriesTreeNode;
      * Determines if node represents a category.
      */
     CategoriesTreeNode.isCategoryNode = (node) => node.extendedData && "isCategory" in node.extendedData && !!node.extendedData.isCategory;
+    /**
+     * Determines if node represents a model.
+     */
+    CategoriesTreeNode.isModelNode = (node) => node.extendedData && "isModel" in node.extendedData && !!node.extendedData.isModel;
+    /**
+     * Determines if node represents an element.
+     */
+    CategoriesTreeNode.isElementNode = (node) => node.extendedData && "isElement" in node.extendedData && !!node.extendedData.isElement;
     /**
      * Determines if node represents a sub-category.
      */
     CategoriesTreeNode.isSubCategoryNode = (node) => node.extendedData && "isSubCategory" in node.extendedData && !!node.extendedData.isSubCategory;
+    /**
+     * Retrieves model ID from node's extended data.
+     */
+    CategoriesTreeNode.getModelId = (node) => {
+        // Elements have modelId property set
+        if (node.extendedData?.modelId) {
+            return node.extendedData?.modelId;
+        }
+        // This is for categories:
+        // a) If category is at root, then it won't have 'modelId' or 'modelIds' set - it will return undefined;
+        // b) If category is under subModel then it will have 'modelIds' set just like in models tree.
+        const modelIds = node.extendedData?.modelIds?.[0];
+        return modelIds && (Array.isArray(modelIds) ? modelIds[0] : modelIds);
+    };
+    /**
+     * Retrieves category ID from node's extended data.
+     */
+    CategoriesTreeNode.getCategoryId = (node) => {
+        // Elements have categoryId property set
+        return node.extendedData?.categoryId;
+    };
 })(CategoriesTreeNode || (CategoriesTreeNode = {}));
 //# sourceMappingURL=CategoriesTreeNode.js.map