@itwin/tree-widget-react 3.0.0-dev.0 → 3.0.0-dev.2

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

@@ -0,0 +1,321 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { assert } from "@itwin/core-bentley";
+import { pushToMap } from "../../common/Utils";
+/** @internal */
+export class ModelsTreeIdsCache {
+    constructor(_queryExecutor, _hierarchyConfig) {
+        this._queryExecutor = _queryExecutor;
+        this._hierarchyConfig = _hierarchyConfig;
+        this._categoryElementCounts = new Map();
+    }
+    async *querySubjects() {
+        const subjectsQuery = `
+      SELECT
+        s.ECInstanceId id,
+        s.Parent.Id parentId,
+        (
+          SELECT m.ECInstanceId
+          FROM bis.GeometricModel3d m
+          WHERE m.ECInstanceId = HexToId(json_extract(s.JsonProperties, '$.Subject.Model.TargetPartition'))
+            AND NOT m.IsPrivate
+            AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)
+        ) targetPartitionId,
+        CASE
+          WHEN (
+            json_extract(s.JsonProperties, '$.Subject.Job.Bridge') IS NOT NULL
+            OR json_extract(s.JsonProperties, '$.Subject.Model.Type') = 'Hierarchy'
+          ) THEN 1
+          ELSE 0
+        END hideInHierarchy
+      FROM bis.Subject s
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: subjectsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { id: row.id, parentId: row.parentId, targetPartitionId: row.targetPartitionId, hideInHierarchy: !!row.hideInHierarchy };
+        }
+    }
+    async *queryModels() {
+        const modelsQuery = `
+      SELECT p.ECInstanceId id, p.Parent.Id parentId
+      FROM bis.InformationPartitionElement p
+      INNER JOIN bis.GeometricModel3d m ON m.ModeledElement.Id = p.ECInstanceId
+      WHERE
+        NOT m.IsPrivate
+        ${this._hierarchyConfig.showEmptyModels ? "" : `AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)`}
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: modelsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { id: row.id, parentId: row.parentId };
+        }
+    }
+    async getSubjectInfos() {
+        this._subjectInfos ??= (async () => {
+            const [subjectInfos, targetPartitionSubjects] = await Promise.all([
+                (async () => {
+                    const result = new Map();
+                    for await (const subject of this.querySubjects()) {
+                        const subjectInfo = {
+                            parentSubject: subject.parentId,
+                            hideInHierarchy: subject.hideInHierarchy,
+                            childSubjects: new Set(),
+                            childModels: new Set(),
+                        };
+                        if (subject.targetPartitionId) {
+                            subjectInfo.childModels.add(subject.targetPartitionId);
+                        }
+                        result.set(subject.id, subjectInfo);
+                    }
+                    return result;
+                })(),
+                (async () => {
+                    const result = new Map();
+                    for await (const model of this.queryModels()) {
+                        pushToMap(result, model.id, model.parentId);
+                    }
+                    return result;
+                })(),
+            ]);
+            for (const [subjectId, { parentSubject: parentSubjectId }] of subjectInfos.entries()) {
+                if (parentSubjectId) {
+                    const parentSubjectInfo = subjectInfos.get(parentSubjectId);
+                    assert(!!parentSubjectInfo);
+                    parentSubjectInfo.childSubjects.add(subjectId);
+                }
+            }
+            for (const [partitionId, subjectIds] of targetPartitionSubjects) {
+                subjectIds.forEach((subjectId) => {
+                    const subjectInfo = subjectInfos.get(subjectId);
+                    assert(!!subjectInfo);
+                    subjectInfo.childModels.add(partitionId);
+                });
+            }
+            return subjectInfos;
+        })();
+        return this._subjectInfos;
+    }
+    /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */
+    async getParentSubjectIds() {
+        this._parentSubjectIds ??= (async () => {
+            const subjectInfos = await this.getSubjectInfos();
+            const parentSubjectIds = new Set();
+            subjectInfos.forEach((subjectInfo, subjectId) => {
+                if (subjectInfo.childModels.size > 0) {
+                    parentSubjectIds.add(subjectId);
+                    let currParentId = subjectInfo.parentSubject;
+                    while (currParentId) {
+                        parentSubjectIds.add(currParentId);
+                        currParentId = subjectInfos.get(currParentId)?.parentSubject;
+                    }
+                }
+            });
+            return [...parentSubjectIds];
+        })();
+        return this._parentSubjectIds;
+    }
+    /**
+     * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into
+     * account `hideInHierarchy` flag.
+     */
+    async getChildSubjectIds(parentSubjectIds) {
+        const childSubjectIds = new Array();
+        const subjectInfos = await this.getSubjectInfos();
+        parentSubjectIds.forEach((subjectId) => {
+            forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {
+                if (!childSubjectInfo.hideInHierarchy) {
+                    childSubjectIds.push(childSubjectId);
+                    return "break";
+                }
+                return "continue";
+            });
+        });
+        return childSubjectIds;
+    }
+    /** Returns ECInstanceIDs of all Models under specific parent Subjects, including their child Subjects, etc. */
+    async getSubjectModelIds(subjectIds) {
+        const subjectInfos = await this.getSubjectInfos();
+        const subjectStack = [...subjectIds];
+        const result = new Array();
+        while (true) {
+            const subjectId = subjectStack.pop();
+            if (subjectId === undefined) {
+                break;
+            }
+            const subjectInfo = subjectInfos.get(subjectId);
+            if (!subjectInfo) {
+                continue;
+            }
+            result.push(...subjectInfo.childModels);
+            subjectStack.push(...subjectInfo.childSubjects);
+        }
+        return result;
+    }
+    /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */
+    async getChildSubjectModelIds(parentSubjectIds) {
+        const subjectInfos = await this.getSubjectInfos();
+        const hiddenSubjectIds = new Array();
+        parentSubjectIds.forEach((subjectId) => {
+            forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {
+                if (childSubjectInfo.hideInHierarchy) {
+                    hiddenSubjectIds.push(childSubjectId);
+                    return "continue";
+                }
+                return "break";
+            });
+        });
+        const modelIds = new Array();
+        [...parentSubjectIds, ...hiddenSubjectIds].forEach((subjectId) => {
+            const subjectInfo = subjectInfos.get(subjectId);
+            subjectInfo && modelIds.push(...subjectInfo.childModels);
+        });
+        return modelIds;
+    }
+    /**
+     * Returns a list of Subject ancestor ECInstanceIds from root to target Subject as displayed in the
+     * hierarchy  - taking into account `hideInHierarchy` flag.
+     */
+    async getSubjectAncestorsPath(targetSubjectId) {
+        const subjectInfos = await this.getSubjectInfos();
+        const result = new Array();
+        let currParentId = targetSubjectId;
+        while (currParentId) {
+            const parentInfo = subjectInfos.get(currParentId);
+            if (!parentInfo?.hideInHierarchy) {
+                result.push(currParentId);
+            }
+            currParentId = parentInfo?.parentSubject;
+        }
+        return result.reverse();
+    }
+    async *queryModelElementCounts() {
+        const query = /* sql */ `
+      SELECT Model.Id modelId, COUNT(*) elementCount
+      FROM ${this._hierarchyConfig.elementClassSpecification}
+      GROUP BY Model.Id
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { modelId: row.modelId, elementCount: row.elementCount };
+        }
+    }
+    async *queryModelCategories() {
+        const query = /* sql */ `
+      SELECT Model.Id modelId, Category.Id categoryId
+      FROM ${this._hierarchyConfig.elementClassSpecification}
+      WHERE Parent IS NULL
+      GROUP BY modelId, categoryId
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { modelId: row.modelId, categoryId: row.categoryId };
+        }
+    }
+    async getModelInfos() {
+        this._modelInfos ??= (async () => {
+            const modelInfos = new Map();
+            await Promise.all([
+                (async () => {
+                    for await (const { modelId, categoryId } of this.queryModelCategories()) {
+                        const entry = modelInfos.get(modelId);
+                        if (entry) {
+                            entry.categories.add(categoryId);
+                        }
+                        else {
+                            modelInfos.set(modelId, { categories: new Set([categoryId]), elementCount: 0 });
+                        }
+                    }
+                })(),
+                (async () => {
+                    for await (const { modelId, elementCount } of this.queryModelElementCounts()) {
+                        const entry = modelInfos.get(modelId);
+                        if (entry) {
+                            entry.elementCount = elementCount;
+                        }
+                        else {
+                            modelInfos.set(modelId, { categories: new Set(), elementCount });
+                        }
+                    }
+                })(),
+            ]);
+            return modelInfos;
+        })();
+        return this._modelInfos;
+    }
+    async getModelCategories(modelId) {
+        const modelInfos = await this.getModelInfos();
+        const categories = modelInfos.get(modelId)?.categories;
+        return categories ? [...categories] : [];
+    }
+    async getModelElementCount(modelId) {
+        const modelInfos = await this.getModelInfos();
+        return modelInfos.get(modelId)?.elementCount ?? 0;
+    }
+    async getModelSubjects(modelId) {
+        const result = new Array();
+        const subjectInfos = await this.getSubjectInfos();
+        subjectInfos.forEach((subjectInfo, subjectId) => {
+            if (subjectInfo.childModels.has(modelId)) {
+                result.push(subjectId);
+            }
+        });
+        return result;
+    }
+    async queryCategoryElementsCount(modelId, categoryId) {
+        const reader = this._queryExecutor.createQueryReader({
+            ctes: [
+                /* sql */ `
+            CategoryElements(id) AS (
+              SELECT ECInstanceId id
+              FROM ${this._hierarchyConfig.elementClassSpecification}
+              WHERE
+                Model.Id = ?
+                AND Category.Id = ?
+                AND Parent IS NULL
+
+              UNION ALL
+
+              SELECT c.ECInstanceId id
+              FROM ${this._hierarchyConfig.elementClassSpecification} c
+              JOIN CategoryElements p ON c.Parent.Id = p.id
+            )
+          `,
+            ],
+            ecsql: `SELECT COUNT(*) FROM CategoryElements`,
+            bindings: [
+                { type: "id", value: modelId },
+                { type: "id", value: categoryId },
+            ],
+        }, { rowFormat: "Indexes", limit: "unbounded" });
+        return (await reader.next()).value[0];
+    }
+    async getCategoryElementsCount(modelId, categoryId) {
+        const cacheKey = `${modelId}${categoryId}`;
+        let result = this._categoryElementCounts.get(cacheKey);
+        if (result !== undefined) {
+            return result;
+        }
+        result = await this.queryCategoryElementsCount(modelId, categoryId);
+        this._categoryElementCounts.set(cacheKey, result);
+        return result;
+    }
+    async getCategoryModels(categoryId) {
+        const result = new Set();
+        const modelInfos = await this.getModelInfos();
+        modelInfos?.forEach((modelInfo, modelId) => {
+            if (modelInfo.categories.has(categoryId)) {
+                result.add(modelId);
+            }
+        });
+        return [...result];
+    }
+}
+function forEachChildSubject(subjectInfos, parentSubject, cb) {
+    const parentSubjectInfo = typeof parentSubject === "string" ? subjectInfos.get(parentSubject) : parentSubject;
+    parentSubjectInfo &&
+        parentSubjectInfo.childSubjects.forEach((childSubjectId) => {
+            const childSubjectInfo = subjectInfos.get(childSubjectId);
+            if (cb(childSubjectId, childSubjectInfo) === "break") {
+                return;
+            }
+            forEachChildSubject(subjectInfos, childSubjectInfo, cb);
+        });
+}
+//# sourceMappingURL=ModelsTreeIdsCache.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelsTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../src/components/trees/models-tree/internal/ModelsTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAoB/C,gBAAgB;AAChB,MAAM,OAAO,kBAAkB;IAM7B,YACU,cAA0C,EAC1C,gBAAkD;QADlD,mBAAc,GAAd,cAAc,CAA4B;QAC1C,qBAAgB,GAAhB,gBAAgB,CAAkC;QAP3C,2BAAsB,GAAG,IAAI,GAAG,EAAsB,CAAC;IAQrE,CAAC;IAEI,KAAK,CAAC,CAAC,aAAa;QAC1B,MAAM,aAAa,GAAG;;;;;;;;;wCASc,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;;;;;;;KAUlF,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAChJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,iBAAiB,EAAE,GAAG,CAAC,iBAAiB,EAAE,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;SAChI;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,WAAW;QACxB,MAAM,WAAW,GAAG;;;;;;UAMd,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,6BAA6B,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,mCAAmC;KACjK,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAC9I,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;SAC9C;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,aAAa,KAAK,CAAC,KAAK,IAAI,EAAE;YACjC,MAAM,CAAC,YAAY,EAAE,uBAAuB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChE,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;oBAClD,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE;wBAChD,MAAM,WAAW,GAAgB;4BAC/B,aAAa,EAAE,OAAO,CAAC,QAAQ;4BAC/B,eAAe,EAAE,OAAO,CAAC,eAAe;4BACxC,aAAa,EAAE,IAAI,GAAG,EAAE;4BACxB,WAAW,EAAE,IAAI,GAAG,EAAE;yBACvB,CAAC;wBACF,IAAI,OAAO,CAAC,iBAAiB,EAAE;4BAC7B,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;yBACxD;wBACD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;qBACrC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA+B,CAAC;oBACtD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE;wBAC5C,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;qBAC7C;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YAEH,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE;gBACpF,IAAI,eAAe,EAAE;oBACnB,MAAM,iBAAiB,GAAG,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;oBAC5D,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC;oBAC5B,iBAAiB,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;iBAChD;aACF;YAED,KAAK,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,uBAAuB,EAAE;gBAC/D,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;oBAC/B,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChD,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;oBACtB,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC3C,CAAC,CAAC,CAAC;aACJ;YAED,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,kHAAkH;IAC3G,KAAK,CAAC,mBAAmB;QAC9B,IAAI,CAAC,iBAAiB,KAAK,CAAC,KAAK,IAAI,EAAE;YACrC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAClD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;YAC/C,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE;gBAC9C,IAAI,WAAW,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE;oBACpC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChC,IAAI,YAAY,GAAG,WAAW,CAAC,aAAa,CAAC;oBAC7C,OAAO,YAAY,EAAE;wBACnB,gBAAgB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;wBACnC,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,aAAa,CAAC;qBAC9D;iBACF;YACH,CAAC,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,gBAAgB,CAAC,CAAC;QAC/B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,kBAAkB,CAAC,gBAA8B;QAC5D,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;QAChD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE;oBACrC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACrC,OAAO,OAAO,CAAC;iBAChB;gBACD,OAAO,UAAU,CAAC;YACpB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,+GAA+G;IACxG,KAAK,CAAC,kBAAkB,CAAC,UAAqB;QACnD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,MAAM,YAAY,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,OAAO,IAAI,EAAE;YACX,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,EAAE,CAAC;YACrC,IAAI,SAAS,KAAK,SAAS,EAAE;gBAC3B,MAAM;aACP;YACD,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,WAAW,EAAE;gBAChB,SAAS;aACV;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;YACxC,YAAY,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,aAAa,CAAC,CAAC;SACjD;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,6GAA6G;IACtG,KAAK,CAAC,uBAAuB,CAAC,gBAA8B;QACjE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAElD,MAAM,gBAAgB,GAAG,IAAI,KAAK,EAAc,CAAC;QACjD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,gBAAgB,CAAC,eAAe,EAAE;oBACpC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACtC,OAAO,UAAU,CAAC;iBACnB;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,KAAK,EAAc,CAAC;QACzC,CAAC,GAAG,gBAAgB,EAAE,GAAG,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YAC/D,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,WAAW,IAAI,QAAQ,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,uBAAuB,CAAC,eAA2B;QAC9D,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,IAAI,YAAY,GAA2B,eAAe,CAAC;QAC3D,OAAO,YAAY,EAAE;YACnB,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YAClD,IAAI,CAAC,UAAU,EAAE,eAAe,EAAE;gBAChC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;aAC3B;YACD,YAAY,GAAG,UAAU,EAAE,aAAa,CAAC;SAC1C;QACD,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,CAAC,uBAAuB;QACpC,MAAM,KAAK,GAAG,SAAS,CAAC;;aAEf,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;KAEvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACxI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC;SAChE;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QACjC,MAAM,KAAK,GAAG,SAAS,CAAC;;aAEf,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;KAGvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACxI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAC5D;IACH,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,IAAI,CAAC,WAAW,KAAK,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,EAA6D,CAAC;YACxF,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,CAAC,KAAK,IAAI,EAAE;oBACV,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE;wBACvE,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACtC,IAAI,KAAK,EAAE;4BACT,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;yBAClC;6BAAM;4BACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;yBACjF;qBACF;gBACH,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,IAAI,CAAC,uBAAuB,EAAE,EAAE;wBAC5E,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACtC,IAAI,KAAK,EAAE;4BACT,KAAK,CAAC,YAAY,GAAG,YAAY,CAAC;yBACnC;6BAAM;4BACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;yBAClE;qBACF;gBACH,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YACH,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,OAAmB;QACjD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,UAAU,CAAC;QACvD,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAAC,OAAmB;QACnD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,OAAO,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,YAAY,IAAI,CAAC,CAAC;IACpD,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,OAAmB;QAC/C,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE;YAC9C,IAAI,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBACxC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;aACxB;QACH,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,0BAA0B,CAAC,OAAmB,EAAE,UAAsB;QAClF,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAClD;YACE,IAAI,EAAE;gBACJ,SAAS,CAAC;;;qBAGC,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;;;;;;qBAS/C,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;WAGzD;aACF;YACD,KAAK,EAAE,uCAAuC;YAC9C,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE;gBAC9B,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE;aAClC;SACF,EACD,EAAE,SAAS,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,CAC7C,CAAC;QAEF,OAAO,CAAC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,MAAM,QAAQ,GAAG,GAAG,OAAO,GAAG,UAAU,EAAE,CAAC;QAC3C,IAAI,MAAM,GAAG,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvD,IAAI,MAAM,KAAK,SAAS,EAAE;YACxB,OAAO,MAAM,CAAC;SACf;QAED,MAAM,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QACpE,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAClD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,iBAAiB,CAAC,UAAsB;QACnD,MAAM,MAAM,GAAG,IAAI,GAAG,EAAc,CAAC;QACrC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,EAAE,OAAO,EAAE,EAAE;YACzC,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;gBACxC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;aACrB;QACH,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,MAAM,CAAC,CAAC;IACrB,CAAC;CACF;AAED,SAAS,mBAAmB,CAC1B,YAA0C,EAC1C,aAAuC,EACvC,EAAuF;IAEvF,MAAM,iBAAiB,GAAG,OAAO,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;IAC9G,iBAAiB;QACf,iBAAiB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YACzD,MAAM,gBAAgB,GAAG,YAAY,CAAC,GAAG,CAAC,cAAc,CAAE,CAAC;YAC3D,IAAI,EAAE,CAAC,cAAc,EAAE,gBAAgB,CAAC,KAAK,OAAO,EAAE;gBACpD,OAAO;aACR;YACD,mBAAmB,CAAC,YAAY,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;AACP,CAAC","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 } from \"@itwin/core-bentley\";\nimport { pushToMap } from \"../../common/Utils\";\n\nimport type { ModelsTreeDefinition } from \"../ModelsTreeDefinition\";\nimport type { Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\n\ninterface SubjectInfo {\n  parentSubject: Id64String | undefined;\n  hideInHierarchy: boolean;\n  childSubjects: Id64Set;\n  childModels: Id64Set;\n}\n\ninterface ModelInfo {\n  categories: Id64Set;\n  elementCount: number;\n}\n\ntype ModelsTreeHierarchyConfiguration = ConstructorParameters<typeof ModelsTreeDefinition>[0][\"hierarchyConfig\"];\n\n/** @internal */\nexport class ModelsTreeIdsCache {\n  private readonly _categoryElementCounts = new Map<Id64String, number>();\n  private _subjectInfos: Promise<Map<Id64String, SubjectInfo>> | undefined;\n  private _parentSubjectIds: Promise<Id64Array> | undefined; // the list should contain a subject id if its node should be shown as having children\n  private _modelInfos: Promise<Map<Id64String, ModelInfo>> | undefined;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    private _hierarchyConfig: ModelsTreeHierarchyConfiguration,\n  ) {}\n\n  private async *querySubjects(): AsyncIterableIterator<{ id: Id64String; parentId?: Id64String; targetPartitionId?: Id64String; hideInHierarchy: boolean }> {\n    const subjectsQuery = `\n      SELECT\n        s.ECInstanceId id,\n        s.Parent.Id parentId,\n        (\n          SELECT m.ECInstanceId\n          FROM bis.GeometricModel3d m\n          WHERE m.ECInstanceId = HexToId(json_extract(s.JsonProperties, '$.Subject.Model.TargetPartition'))\n            AND NOT m.IsPrivate\n            AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)\n        ) targetPartitionId,\n        CASE\n          WHEN (\n            json_extract(s.JsonProperties, '$.Subject.Job.Bridge') IS NOT NULL\n            OR json_extract(s.JsonProperties, '$.Subject.Model.Type') = 'Hierarchy'\n          ) THEN 1\n          ELSE 0\n        END hideInHierarchy\n      FROM bis.Subject s\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: subjectsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId, targetPartitionId: row.targetPartitionId, hideInHierarchy: !!row.hideInHierarchy };\n    }\n  }\n\n  private async *queryModels(): AsyncIterableIterator<{ id: Id64String; parentId: Id64String }> {\n    const modelsQuery = `\n      SELECT p.ECInstanceId id, p.Parent.Id parentId\n      FROM bis.InformationPartitionElement p\n      INNER JOIN bis.GeometricModel3d m ON m.ModeledElement.Id = p.ECInstanceId\n      WHERE\n        NOT m.IsPrivate\n        ${this._hierarchyConfig.showEmptyModels ? \"\" : `AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)`}\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: modelsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId };\n    }\n  }\n\n  private async getSubjectInfos() {\n    this._subjectInfos ??= (async () => {\n      const [subjectInfos, targetPartitionSubjects] = await Promise.all([\n        (async () => {\n          const result = new Map<Id64String, SubjectInfo>();\n          for await (const subject of this.querySubjects()) {\n            const subjectInfo: SubjectInfo = {\n              parentSubject: subject.parentId,\n              hideInHierarchy: subject.hideInHierarchy,\n              childSubjects: new Set(),\n              childModels: new Set(),\n            };\n            if (subject.targetPartitionId) {\n              subjectInfo.childModels.add(subject.targetPartitionId);\n            }\n            result.set(subject.id, subjectInfo);\n          }\n          return result;\n        })(),\n        (async () => {\n          const result = new Map<Id64String, Set<Id64String>>();\n          for await (const model of this.queryModels()) {\n            pushToMap(result, model.id, model.parentId);\n          }\n          return result;\n        })(),\n      ]);\n\n      for (const [subjectId, { parentSubject: parentSubjectId }] of subjectInfos.entries()) {\n        if (parentSubjectId) {\n          const parentSubjectInfo = subjectInfos.get(parentSubjectId);\n          assert(!!parentSubjectInfo);\n          parentSubjectInfo.childSubjects.add(subjectId);\n        }\n      }\n\n      for (const [partitionId, subjectIds] of targetPartitionSubjects) {\n        subjectIds.forEach((subjectId) => {\n          const subjectInfo = subjectInfos.get(subjectId);\n          assert(!!subjectInfo);\n          subjectInfo.childModels.add(partitionId);\n        });\n      }\n\n      return subjectInfos;\n    })();\n    return this._subjectInfos;\n  }\n\n  /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */\n  public async getParentSubjectIds(): Promise<Id64String[]> {\n    this._parentSubjectIds ??= (async () => {\n      const subjectInfos = await this.getSubjectInfos();\n      const parentSubjectIds = new Set<Id64String>();\n      subjectInfos.forEach((subjectInfo, subjectId) => {\n        if (subjectInfo.childModels.size > 0) {\n          parentSubjectIds.add(subjectId);\n          let currParentId = subjectInfo.parentSubject;\n          while (currParentId) {\n            parentSubjectIds.add(currParentId);\n            currParentId = subjectInfos.get(currParentId)?.parentSubject;\n          }\n        }\n      });\n      return [...parentSubjectIds];\n    })();\n    return this._parentSubjectIds;\n  }\n\n  /**\n   * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into\n   * account `hideInHierarchy` flag.\n   */\n  public async getChildSubjectIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const childSubjectIds = new Array<Id64String>();\n    const subjectInfos = await this.getSubjectInfos();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (!childSubjectInfo.hideInHierarchy) {\n          childSubjectIds.push(childSubjectId);\n          return \"break\";\n        }\n        return \"continue\";\n      });\n    });\n    return childSubjectIds;\n  }\n\n  /** Returns ECInstanceIDs of all Models under specific parent Subjects, including their child Subjects, etc. */\n  public async getSubjectModelIds(subjectIds: Id64Array): Promise<Id64Array> {\n    const subjectInfos = await this.getSubjectInfos();\n    const subjectStack = [...subjectIds];\n    const result = new Array<Id64String>();\n    while (true) {\n      const subjectId = subjectStack.pop();\n      if (subjectId === undefined) {\n        break;\n      }\n      const subjectInfo = subjectInfos.get(subjectId);\n      if (!subjectInfo) {\n        continue;\n      }\n      result.push(...subjectInfo.childModels);\n      subjectStack.push(...subjectInfo.childSubjects);\n    }\n    return result;\n  }\n\n  /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */\n  public async getChildSubjectModelIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const subjectInfos = await this.getSubjectInfos();\n\n    const hiddenSubjectIds = new Array<Id64String>();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (childSubjectInfo.hideInHierarchy) {\n          hiddenSubjectIds.push(childSubjectId);\n          return \"continue\";\n        }\n        return \"break\";\n      });\n    });\n\n    const modelIds = new Array<Id64String>();\n    [...parentSubjectIds, ...hiddenSubjectIds].forEach((subjectId) => {\n      const subjectInfo = subjectInfos.get(subjectId);\n      subjectInfo && modelIds.push(...subjectInfo.childModels);\n    });\n    return modelIds;\n  }\n\n  /**\n   * Returns a list of Subject ancestor ECInstanceIds from root to target Subject as displayed in the\n   * hierarchy  - taking into account `hideInHierarchy` flag.\n   */\n  public async getSubjectAncestorsPath(targetSubjectId: Id64String): Promise<Id64Array> {\n    const subjectInfos = await this.getSubjectInfos();\n    const result = new Array<Id64String>();\n    let currParentId: Id64String | undefined = targetSubjectId;\n    while (currParentId) {\n      const parentInfo = subjectInfos.get(currParentId);\n      if (!parentInfo?.hideInHierarchy) {\n        result.push(currParentId);\n      }\n      currParentId = parentInfo?.parentSubject;\n    }\n    return result.reverse();\n  }\n\n  private async *queryModelElementCounts() {\n    const query = /* sql */ `\n      SELECT Model.Id modelId, COUNT(*) elementCount\n      FROM ${this._hierarchyConfig.elementClassSpecification}\n      GROUP BY Model.Id\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, elementCount: row.elementCount };\n    }\n  }\n\n  private async *queryModelCategories() {\n    const query = /* sql */ `\n      SELECT Model.Id modelId, Category.Id categoryId\n      FROM ${this._hierarchyConfig.elementClassSpecification}\n      WHERE Parent IS NULL\n      GROUP BY modelId, categoryId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, categoryId: row.categoryId };\n    }\n  }\n\n  private async getModelInfos() {\n    this._modelInfos ??= (async () => {\n      const modelInfos = new Map<Id64String, { categories: Id64Set; elementCount: number }>();\n      await Promise.all([\n        (async () => {\n          for await (const { modelId, categoryId } of this.queryModelCategories()) {\n            const entry = modelInfos.get(modelId);\n            if (entry) {\n              entry.categories.add(categoryId);\n            } else {\n              modelInfos.set(modelId, { categories: new Set([categoryId]), elementCount: 0 });\n            }\n          }\n        })(),\n        (async () => {\n          for await (const { modelId, elementCount } of this.queryModelElementCounts()) {\n            const entry = modelInfos.get(modelId);\n            if (entry) {\n              entry.elementCount = elementCount;\n            } else {\n              modelInfos.set(modelId, { categories: new Set(), elementCount });\n            }\n          }\n        })(),\n      ]);\n      return modelInfos;\n    })();\n    return this._modelInfos;\n  }\n\n  public async getModelCategories(modelId: Id64String): Promise<Id64Array> {\n    const modelInfos = await this.getModelInfos();\n    const categories = modelInfos.get(modelId)?.categories;\n    return categories ? [...categories] : [];\n  }\n\n  public async getModelElementCount(modelId: Id64String): Promise<number> {\n    const modelInfos = await this.getModelInfos();\n    return modelInfos.get(modelId)?.elementCount ?? 0;\n  }\n\n  public async getModelSubjects(modelId: Id64String): Promise<Id64Array> {\n    const result = new Array<Id64String>();\n    const subjectInfos = await this.getSubjectInfos();\n    subjectInfos.forEach((subjectInfo, subjectId) => {\n      if (subjectInfo.childModels.has(modelId)) {\n        result.push(subjectId);\n      }\n    });\n    return result;\n  }\n\n  private async queryCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    const reader = this._queryExecutor.createQueryReader(\n      {\n        ctes: [\n          /* sql */ `\n            CategoryElements(id) AS (\n              SELECT ECInstanceId id\n              FROM ${this._hierarchyConfig.elementClassSpecification}\n              WHERE\n                Model.Id = ?\n                AND Category.Id = ?\n                AND Parent IS NULL\n\n              UNION ALL\n\n              SELECT c.ECInstanceId id\n              FROM ${this._hierarchyConfig.elementClassSpecification} c\n              JOIN CategoryElements p ON c.Parent.Id = p.id\n            )\n          `,\n        ],\n        ecsql: `SELECT COUNT(*) FROM CategoryElements`,\n        bindings: [\n          { type: \"id\", value: modelId },\n          { type: \"id\", value: categoryId },\n        ],\n      },\n      { rowFormat: \"Indexes\", limit: \"unbounded\" },\n    );\n\n    return (await reader.next()).value[0];\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    const cacheKey = `${modelId}${categoryId}`;\n    let result = this._categoryElementCounts.get(cacheKey);\n    if (result !== undefined) {\n      return result;\n    }\n\n    result = await this.queryCategoryElementsCount(modelId, categoryId);\n    this._categoryElementCounts.set(cacheKey, result);\n    return result;\n  }\n\n  public async getCategoryModels(categoryId: Id64String): Promise<Id64Array> {\n    const result = new Set<Id64String>();\n    const modelInfos = await this.getModelInfos();\n    modelInfos?.forEach((modelInfo, modelId) => {\n      if (modelInfo.categories.has(categoryId)) {\n        result.add(modelId);\n      }\n    });\n    return [...result];\n  }\n}\n\nfunction forEachChildSubject(\n  subjectInfos: Map<Id64String, SubjectInfo>,\n  parentSubject: Id64String | SubjectInfo,\n  cb: (childSubjectId: Id64String, childSubjectInfo: SubjectInfo) => \"break\" | \"continue\",\n) {\n  const parentSubjectInfo = typeof parentSubject === \"string\" ? subjectInfos.get(parentSubject) : parentSubject;\n  parentSubjectInfo &&\n    parentSubjectInfo.childSubjects.forEach((childSubjectId) => {\n      const childSubjectInfo = subjectInfos.get(childSubjectId)!;\n      if (cb(childSubjectId, childSubjectInfo) === \"break\") {\n        return;\n      }\n      forEachChildSubject(subjectInfos, childSubjectInfo, cb);\n    });\n}\n"]}

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

@@ -0,0 +1,31 @@
+import type { Id64String } from "@itwin/core-bentley";
+interface ModelsTreeNode {
+    extendedData?: any;
+}
+/**
+ * @internal
+ */
+export declare namespace ModelsTreeNode {
+    /**
+     * Determines if a node represents a subject.
+     */
+    const isSubjectNode: (node: ModelsTreeNode) => boolean;
+    /**
+     * Determines if a node represents a model.
+     */
+    const isModelNode: (node: ModelsTreeNode) => boolean;
+    /**
+     * Determines if a node represents a category.
+     */
+    const isCategoryNode: (node: ModelsTreeNode) => boolean;
+    /**
+     * Retrieves model ID from node's extended data.
+     */
+    const getModelId: (node: ModelsTreeNode) => Id64String | undefined;
+    /**
+     * Retrieves category ID from node's extended data.
+     */
+    const getCategoryId: (node: ModelsTreeNode) => Id64String | undefined;
+}
+export {};
+//# sourceMappingURL=ModelsTreeNode.d.ts.map

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

@@ -0,0 +1,37 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+/**
+ * @internal
+ */
+export var ModelsTreeNode;
+(function (ModelsTreeNode) {
+    /**
+     * Determines if a node represents a subject.
+     */
+    ModelsTreeNode.isSubjectNode = (node) => !!node.extendedData?.isSubject;
+    /**
+     * Determines if a node represents a model.
+     */
+    ModelsTreeNode.isModelNode = (node) => !!node.extendedData?.isModel;
+    /**
+     * Determines if a node represents a category.
+     */
+    ModelsTreeNode.isCategoryNode = (node) => !!node.extendedData?.isCategory;
+    /**
+     * Retrieves model ID from node's extended data.
+     */
+    ModelsTreeNode.getModelId = (node) => {
+        if (node.extendedData?.modelId) {
+            return node.extendedData?.modelId;
+        }
+        const modelIds = node.extendedData?.modelIds?.[0];
+        return modelIds && (Array.isArray(modelIds) ? modelIds[0] : modelIds);
+    };
+    /**
+     * Retrieves category ID from node's extended data.
+     */
+    ModelsTreeNode.getCategoryId = (node) => node.extendedData?.categoryId;
+})(ModelsTreeNode || (ModelsTreeNode = {}));
+//# sourceMappingURL=ModelsTreeNode.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelsTreeNode.js","sourceRoot":"","sources":["../../../../../../src/components/trees/models-tree/internal/ModelsTreeNode.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAQhG;;GAEG;AACH,MAAM,KAAW,cAAc,CAgC9B;AAhCD,WAAiB,cAAc;IAC7B;;OAEG;IACU,4BAAa,GAAG,CAAC,IAAoB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC;IAEtF;;OAEG;IACU,0BAAW,GAAG,CAAC,IAAoB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC;IAElF;;OAEG;IACU,6BAAc,GAAG,CAAC,IAAoB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;IAExF;;OAEG;IACU,yBAAU,GAAG,CAAC,IAAoB,EAA0B,EAAE;QACzE,IAAI,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE;YAC9B,OAAO,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC;SACnC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;QAClD,OAAO,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;IACxE,CAAC,CAAC;IAEF;;OAEG;IACU,4BAAa,GAAG,CAAC,IAAoB,EAA0B,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;AAC/G,CAAC,EAhCgB,cAAc,KAAd,cAAc,QAgC9B","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport type { Id64String } from \"@itwin/core-bentley\";\n\ninterface ModelsTreeNode {\n  extendedData?: any;\n}\n\n/**\n * @internal\n */\nexport namespace ModelsTreeNode {\n  /**\n   * Determines if a node represents a subject.\n   */\n  export const isSubjectNode = (node: ModelsTreeNode) => !!node.extendedData?.isSubject;\n\n  /**\n   * Determines if a node represents a model.\n   */\n  export const isModelNode = (node: ModelsTreeNode) => !!node.extendedData?.isModel;\n\n  /**\n   * Determines if a node represents a category.\n   */\n  export const isCategoryNode = (node: ModelsTreeNode) => !!node.extendedData?.isCategory;\n\n  /**\n   * Retrieves model ID from node's extended data.\n   */\n  export const getModelId = (node: ModelsTreeNode): Id64String | undefined => {\n    if (node.extendedData?.modelId) {\n      return node.extendedData?.modelId;\n    }\n\n    const modelIds = node.extendedData?.modelIds?.[0];\n    return modelIds && (Array.isArray(modelIds) ? modelIds[0] : modelIds);\n  };\n\n  /**\n   * Retrieves category ID from node's extended data.\n   */\n  export const getCategoryId = (node: ModelsTreeNode): Id64String | undefined => node.extendedData?.categoryId;\n}\n"]}

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

@@ -0,0 +1,122 @@
+import type { ModelsTreeIdsCache } from "./ModelsTreeIdsCache";
+import type { Id64Arg, Id64Array, Id64String } from "@itwin/core-bentley";
+import type { GroupingHierarchyNode } from "@itwin/presentation-hierarchies";
+import type { Viewport } from "@itwin/core-frontend";
+import type { HierarchyVisibilityHandler, VisibilityStatus } from "../../common/UseHierarchyVisibility";
+import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
+interface GetCategoryStatusProps {
+    categoryId: Id64String;
+    modelId: Id64String;
+}
+interface ChangeCategoryStateProps extends GetCategoryStatusProps {
+    on: boolean;
+}
+interface GetElementStateProps {
+    elementId: Id64String;
+    modelId: Id64String;
+    categoryId: Id64String;
+}
+interface ChangeElementStateProps extends GetElementStateProps {
+    on: boolean;
+}
+interface ChangeModelStateProps {
+    ids: Id64Arg;
+    on: boolean;
+}
+export declare const SUBJECT_CLASS_NAME: "BisCore.Subject";
+export declare const MODEL_CLASS_NAME: "BisCore.GeometricModel3d";
+export declare const CATEGORY_CLASS_NAME: "BisCore.SpatialCategory";
+export declare const ELEMENT_CLASS_NAME: "BisCore.GeometricElement3d";
+/**
+ * Properties for a method of `ModelsTreeVisibilityHandler` that can be overridden.
+ */
+type OverridableMethodProps<TFunc> = TFunc extends (props: infer TProps) => infer TResult ? TProps & {
+    /** A callback that produces the value from the original implementation. */
+    readonly originalImplementation: () => TResult;
+    /**
+     * Reference to the hierarchy based handler.
+     * @note Calling `getVisibility` or `changeVisibility` of this object invokes the overridden implementation as well.
+     */
+    readonly handler: ModelsTreeVisibilityHandler;
+} : never;
+/**
+ * Function type for an overridden method of `ModelsTreeVisibilityHandler`.
+ */
+type OverridableMethod<TFunc> = TFunc extends (...args: any[]) => infer TResult ? (props: OverridableMethodProps<TFunc>) => TResult : never;
+/**
+ * Functionality of `ModelsTreeVisibilityHandler` that can be overridden.
+ * Each callback will be provided original implementation and reference to a `HierarchyVisibilityHandler`.
+ */
+interface VisibilityHandlerOverrides {
+    getSubjectNodeVisibility?: OverridableMethod<(props: {
+        ids: Id64Array;
+    }) => Promise<VisibilityStatus>>;
+    getModelDisplayStatus?: OverridableMethod<(props: {
+        id: Id64String;
+    }) => Promise<VisibilityStatus>>;
+    getCategoryDisplayStatus?: OverridableMethod<(props: GetCategoryStatusProps) => Promise<VisibilityStatus>>;
+    getElementGroupingNodeDisplayStatus?: OverridableMethod<(props: {
+        node: GroupingHierarchyNode;
+    }) => Promise<VisibilityStatus>>;
+    getElementDisplayStatus?: OverridableMethod<(props: GetElementStateProps) => Promise<VisibilityStatus>>;
+    changeSubjectNodeState?: OverridableMethod<(props: {
+        ids: Id64Array;
+        on: boolean;
+    }) => Promise<void>>;
+    changeModelState?: OverridableMethod<(props: ChangeModelStateProps) => Promise<void>>;
+    changeCategoryState?: OverridableMethod<(props: ChangeCategoryStateProps) => Promise<void>>;
+    changeElementGroupingNodeState?: OverridableMethod<(props: {
+        node: GroupingHierarchyNode;
+        on: boolean;
+    }) => Promise<void>>;
+    changeElementState?: OverridableMethod<(props: ChangeElementStateProps) => Promise<void>>;
+}
+/**
+ * Properties for `ModelsTreeVisibilityHandler`.
+ * @internal
+ */
+export interface ModelsTreeVisibilityHandlerProps {
+    viewport: Viewport;
+    idsCache: ModelsTreeIdsCache;
+    imodelAccess: ECClassHierarchyInspector;
+    overrides?: VisibilityHandlerOverrides;
+}
+/**
+ * Hierarchy based visibility handler.
+ * When determining visibility for nodes, it should take into account the visibility of their children.
+ * @internal
+ */
+export type ModelsTreeVisibilityHandler = HierarchyVisibilityHandler;
+/**
+ * Creates an instance if `ModelsTreeVisibilityHandler`.
+ * @internal
+ */
+export declare function createModelsTreeVisibilityHandler(props: ModelsTreeVisibilityHandlerProps): ModelsTreeVisibilityHandler;
+/**
+ * Enables display of all given models. Also enables display of all categories and clears always and
+ * never drawn lists in the viewport.
+ * @public
+ */
+export declare function showAllModels(models: string[], viewport: Viewport): Promise<void>;
+/**
+ * Disables display of all given models.
+ * @public
+ */
+export declare function hideAllModels(models: string[], viewport: Viewport): Promise<void>;
+/**
+ * Inverts display of all given models.
+ * @public
+ */
+export declare function invertAllModels(models: string[], viewport: Viewport): Promise<void>;
+/**
+ * Based on the value of `enable` argument, either enables or disables display of given models.
+ * @public
+ */
+export declare function toggleModels(models: string[], enable: boolean, viewport: Viewport): Promise<void>;
+/**
+ * Checks if all given models are displayed in given viewport.
+ * @public
+ */
+export declare function areAllModelsVisible(models: string[], viewport: Viewport): boolean;
+export {};
+//# sourceMappingURL=ModelsTreeVisibilityHandler.d.ts.map