@itwin/tree-widget-react 1.0.0 → 1.1.1

package/lib/cjs/components/trees/models-tree/Utils.js

@@ -4,7 +4,7 @@
 * See LICENSE.md in the project root for license terms and full copyright notice.
 *--------------------------------------------------------------------------------------------*/
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.customizeModelsTreeNodeItem = exports.createSearchRuleset = exports.createRuleset = exports.CachingElementIdsContainer = void 0;
+exports.addModelsTreeNodeItemIcons = exports.createSearchRuleset = exports.createRuleset = exports.CachingElementIdsContainer = void 0;
 const presentation_common_1 = require("@itwin/presentation-common");
 /** @internal */
 class CachingElementIdsContainer {
@@ -27,7 +27,11 @@ class CachingElementIdsContainer {
 exports.CachingElementIdsContainer = CachingElementIdsContainer;
 /** @internal */
 function createRuleset(props) {
-    const elementClassSpecification = props.elementClassSpecification ?? { schemaName: "BisCore", className: "GeometricElement3d" };
+    const context = {
+        elementClassSpecification: props.elementClassSpecification ?? { schemaName: "BisCore", className: "GeometricElement3d" },
+        groupElements: !!props.enableElementsClassGrouping,
+        showEmptyModels: !!props.showEmptyModels,
+    };
     return {
         id: "tree-widget-react/ModelsTree",
         requiredSchemas: [
@@ -40,20 +44,7 @@ function createRuleset(props) {
                 ruleType: "RootNodes",
                 autoExpand: true,
                 specifications: [
-                    {
-                        specType: "InstanceNodesOfSpecificClasses",
-                        classes: [
-                            {
-                                schemaName: "BisCore",
-                                classNames: [
-                                    "Subject",
-                                ],
-                            },
-                        ],
-                        instanceFilter: `this.Parent = NULL`,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
+                    createRootSubjectSpecification(),
                 ],
                 customizationRules: [
                     {
@@ -69,46 +60,8 @@ function createRuleset(props) {
                 ruleType: "ChildNodes",
                 condition: `ParentNode.IsOfClass("Subject", "BisCore")`,
                 specifications: [
-                    {
-                        specType: "RelatedInstanceNodes",
-                        relationshipPaths: [
-                            {
-                                relationship: {
-                                    schemaName: "BisCore",
-                                    className: "SubjectOwnsSubjects",
-                                },
-                                direction: "Forward",
-                                targetClass: {
-                                    schemaName: "BisCore",
-                                    className: "Subject",
-                                },
-                            },
-                        ],
-                        instanceFilter: `json_extract(this.JsonProperties, "$.Subject.Job.Bridge") <> NULL OR ifnull(json_extract(this.JsonProperties, "$.Subject.Model.Type"), "") = "Hierarchy"`,
-                        hideNodesInHierarchy: true,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
-                    {
-                        specType: "RelatedInstanceNodes",
-                        relationshipPaths: [
-                            {
-                                relationship: {
-                                    schemaName: "BisCore",
-                                    className: "SubjectOwnsSubjects",
-                                },
-                                direction: "Forward",
-                                targetClass: {
-                                    schemaName: "BisCore",
-                                    className: "Subject",
-                                },
-                            },
-                        ],
-                        instanceFilter: `json_extract(this.JsonProperties, "$.Subject.Job.Bridge") = NULL AND ifnull(json_extract(this.JsonProperties, "$.Subject.Model.Type"), "") <> "Hierarchy"`,
-                        hideIfNoChildren: true,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
+                    createRelatedHierarchySubjectSpecification(),
+                    createRelatedNonHierarchySubjectSpecification(),
                 ],
                 customizationRules: [
                     {
@@ -124,70 +77,8 @@ function createRuleset(props) {
                 ruleType: "ChildNodes",
                 condition: `ParentNode.IsOfClass("Subject", "BisCore")`,
                 specifications: [
-                    {
-                        specType: "InstanceNodesOfSpecificClasses",
-                        classes: {
-                            schemaName: "BisCore",
-                            classNames: [
-                                "GeometricModel3d",
-                            ],
-                            arePolymorphic: true,
-                        },
-                        relatedInstances: [
-                            {
-                                relationshipPath: {
-                                    relationship: {
-                                        schemaName: "BisCore",
-                                        className: "ModelModelsElement",
-                                    },
-                                    direction: "Forward",
-                                    targetClass: {
-                                        schemaName: "BisCore",
-                                        className: "InformationPartitionElement",
-                                    },
-                                },
-                                alias: "partition",
-                                isRequired: true,
-                            },
-                        ],
-                        instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, "$.Subject.Model.TargetPartition") = printf("0x%x", partition.ECInstanceId)) AND NOT this.IsPrivate AND json_extract(partition.JsonProperties, "$.PhysicalPartition.Model.Content") = NULL AND json_extract(partition.JsonProperties, "$.GraphicalPartition3d.Model.Content") = NULL AND this.HasRelatedInstance("BisCore:ModelContainsElements", "Forward", "${elementClassSpecification.schemaName}:${elementClassSpecification.className}")`,
-                        hasChildren: "Always",
-                        hideIfNoChildren: true,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
-                    {
-                        specType: "InstanceNodesOfSpecificClasses",
-                        classes: {
-                            schemaName: "BisCore",
-                            classNames: [
-                                "GeometricModel3d",
-                            ],
-                            arePolymorphic: true,
-                        },
-                        relatedInstances: [
-                            {
-                                relationshipPath: {
-                                    relationship: {
-                                        schemaName: "BisCore",
-                                        className: "ModelModelsElement",
-                                    },
-                                    direction: "Forward",
-                                    targetClass: {
-                                        schemaName: "BisCore",
-                                        className: "InformationPartitionElement",
-                                    },
-                                },
-                                alias: "partition",
-                                isRequired: true,
-                            },
-                        ],
-                        instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, "$.Subject.Model.TargetPartition") = printf("0x%x", partition.ECInstanceId)) AND NOT this.IsPrivate AND (json_extract(partition.JsonProperties, "$.PhysicalPartition.Model.Content") <> NULL OR json_extract(partition.JsonProperties, "$.GraphicalPartition3d.Model.Content") <> NULL) AND this.HasRelatedInstance("BisCore:ModelContainsElements", "Forward", "${elementClassSpecification.schemaName}:${elementClassSpecification.className}")`,
-                        hasChildren: "Always",
-                        hideNodesInHierarchy: true,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
+                    createNonContentModelsSpecification(context),
+                    createContentModelsSpecification(context),
                 ],
                 customizationRules: [
                     {
@@ -203,22 +94,7 @@ function createRuleset(props) {
                 ruleType: "ChildNodes",
                 condition: `ParentNode.IsOfClass("ISubModeledElement", "BisCore")`,
                 specifications: [
-                    {
-                        specType: "RelatedInstanceNodes",
-                        relationshipPaths: [
-                            {
-                                relationship: {
-                                    schemaName: "BisCore",
-                                    className: "ModelModelsElement",
-                                },
-                                direction: "Backward",
-                            },
-                        ],
-                        instanceFilter: `NOT this.IsPrivate AND this.HasRelatedInstance("BisCore:ModelContainsElements", "Forward", "${elementClassSpecification.schemaName}:${elementClassSpecification.className}")`,
-                        hideNodesInHierarchy: true,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
+                    createElementModelSpecification(context),
                 ],
                 customizationRules: [
                     {
@@ -234,33 +110,7 @@ function createRuleset(props) {
                 ruleType: "ChildNodes",
                 condition: `ParentNode.IsOfClass("GeometricModel3d", "BisCore")`,
                 specifications: [
-                    {
-                        specType: "RelatedInstanceNodes",
-                        relationshipPaths: [
-                            [
-                                {
-                                    relationship: {
-                                        schemaName: "BisCore",
-                                        className: "ModelContainsElements",
-                                    },
-                                    direction: "Forward",
-                                    targetClass: elementClassSpecification,
-                                },
-                                {
-                                    relationship: {
-                                        schemaName: "BisCore",
-                                        className: "GeometricElement3dIsInCategory",
-                                    },
-                                    direction: "Forward",
-                                },
-                            ],
-                        ],
-                        instanceFilter: `NOT this.IsPrivate`,
-                        suppressSimilarAncestorsCheck: true,
-                        hideIfNoChildren: true,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
+                    createModelCategoriesSpecification(context),
                 ],
                 customizationRules: [
                     {
@@ -277,22 +127,7 @@ function createRuleset(props) {
                 ruleType: "ChildNodes",
                 condition: `ParentNode.IsOfClass("SpatialCategory", "BisCore")`,
                 specifications: [
-                    {
-                        specType: "RelatedInstanceNodes",
-                        relationshipPaths: [
-                            {
-                                relationship: {
-                                    schemaName: "BisCore",
-                                    className: "GeometricElement3dIsInCategory",
-                                },
-                                direction: "Backward",
-                                targetClass: elementClassSpecification,
-                            },
-                        ],
-                        instanceFilter: `this.Model.Id = parent.parent.ECInstanceId ANDALSO this.Parent = NULL`,
-                        groupByClass: !!props.enableElementsClassGrouping,
-                        groupByLabel: false,
-                    },
+                    createCategoryElementsSpecification(context),
                 ],
                 customizationRules: [
                     {
@@ -310,21 +145,7 @@ function createRuleset(props) {
                 ruleType: "ChildNodes",
                 condition: `ParentNode.IsOfClass("GeometricElement3d", "BisCore")`,
                 specifications: [
-                    {
-                        specType: "RelatedInstanceNodes",
-                        relationshipPaths: [
-                            {
-                                relationship: {
-                                    schemaName: "BisCore",
-                                    className: "ElementOwnsChildElements",
-                                },
-                                direction: "Forward",
-                                targetClass: elementClassSpecification,
-                            },
-                        ],
-                        groupByClass: !!props.enableElementsClassGrouping,
-                        groupByLabel: false,
-                    },
+                    createEementElementsSpecification(context),
                 ],
                 customizationRules: [
                     {
@@ -392,28 +213,18 @@ function createRuleset(props) {
 exports.createRuleset = createRuleset;
 /** @internal */
 function createSearchRuleset(props) {
-    const elementClassSpecification = props.elementClassSpecification ?? { schemaName: "BisCore", className: "GeometricElement3d" };
+    const context = {
+        elementClassSpecification: props.elementClassSpecification ?? { schemaName: "BisCore", className: "GeometricElement3d" },
+        groupElements: false,
+        showEmptyModels: !!props.showEmptyModels,
+    };
     return {
         id: "tree-widget-react/ModelsTreeSearch",
         rules: [
             {
                 ruleType: "RootNodes",
                 specifications: [
-                    {
-                        specType: "InstanceNodesOfSpecificClasses",
-                        classes: [
-                            {
-                                schemaName: "BisCore",
-                                classNames: [
-                                    "Subject",
-                                ],
-                            },
-                        ],
-                        instanceFilter: "this.Parent = NULL",
-                        arePolymorphic: false,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
+                    createRootSubjectSpecification(),
                 ],
                 customizationRules: [
                     {
@@ -429,45 +240,10 @@ function createSearchRuleset(props) {
                 ruleType: "ChildNodes",
                 condition: `ParentNode.IsOfClass("Subject", "BisCore")`,
                 specifications: [
+                    createRelatedHierarchySubjectSpecification(),
                     {
-                        specType: "RelatedInstanceNodes",
-                        relationshipPaths: [
-                            {
-                                relationship: {
-                                    schemaName: "BisCore",
-                                    className: "SubjectOwnsSubjects",
-                                },
-                                direction: "Forward",
-                                targetClass: {
-                                    schemaName: "BisCore",
-                                    className: "Subject",
-                                },
-                            },
-                        ],
-                        instanceFilter: `json_extract(this.JsonProperties, "$.Subject.Job.Bridge") <> NULL OR ifnull(json_extract(this.JsonProperties, "$.Subject.Model.Type"), "") = "Hierarchy"`,
-                        hideNodesInHierarchy: true,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
-                    {
-                        specType: "RelatedInstanceNodes",
-                        relationshipPaths: [
-                            {
-                                relationship: {
-                                    schemaName: "BisCore",
-                                    className: "SubjectOwnsSubjects",
-                                },
-                                direction: "Forward",
-                                targetClass: {
-                                    schemaName: "BisCore",
-                                    className: "Subject",
-                                },
-                            },
-                        ],
-                        instanceFilter: `json_extract(this.JsonProperties, "$.Subject.Job.Bridge") = NULL AND ifnull(json_extract(this.JsonProperties, "$.Subject.Model.Type"), "") <> "Hierarchy"`,
+                        ...createRelatedNonHierarchySubjectSpecification(),
                         hideExpression: `NOT ThisNode.HasChildren ANDALSO NOT ThisNode.ChildrenArtifacts.AnyMatches(x => x.isContentModel)`,
-                        groupByClass: false,
-                        groupByLabel: false,
                     },
                 ],
                 customizationRules: [
@@ -497,36 +273,7 @@ function createSearchRuleset(props) {
                 ruleType: "ChildNodes",
                 condition: `ParentNode.IsOfClass("Subject", "BisCore")`,
                 specifications: [
-                    {
-                        specType: "InstanceNodesOfSpecificClasses",
-                        classes: {
-                            schemaName: "BisCore",
-                            classNames: [
-                                "GeometricModel3d",
-                            ],
-                        },
-                        arePolymorphic: true,
-                        relatedInstances: [
-                            {
-                                relationshipPath: {
-                                    relationship: {
-                                        schemaName: "BisCore",
-                                        className: "ModelModelsElement",
-                                    },
-                                    direction: "Forward",
-                                    targetClass: {
-                                        schemaName: "BisCore",
-                                        className: "InformationPartitionElement",
-                                    },
-                                },
-                                alias: "partition",
-                                isRequired: true,
-                            },
-                        ],
-                        instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, "$.Subject.Model.TargetPartition") = printf("0x%x", partition.ECInstanceId)) AND NOT this.IsPrivate AND json_extract(partition.JsonProperties, "$.PhysicalPartition.Model.Content") = NULL AND json_extract(partition.JsonProperties, "$.GraphicalPartition3d.Model.Content") = NULL AND this.HasRelatedInstance("BisCore:ModelContainsElements", "Forward", "${elementClassSpecification.schemaName}:${elementClassSpecification.className}")`,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
+                    createNonContentModelsSpecification(context),
                 ],
                 customizationRules: [
                     {
@@ -542,37 +289,7 @@ function createSearchRuleset(props) {
                 ruleType: "ChildNodes",
                 condition: `ParentNode.IsOfClass("Subject", "BisCore")`,
                 specifications: [
-                    {
-                        specType: "InstanceNodesOfSpecificClasses",
-                        classes: {
-                            schemaName: "BisCore",
-                            classNames: [
-                                "GeometricModel3d",
-                            ],
-                        },
-                        arePolymorphic: true,
-                        relatedInstances: [
-                            {
-                                relationshipPath: {
-                                    relationship: {
-                                        schemaName: "BisCore",
-                                        className: "ModelModelsElement",
-                                    },
-                                    direction: "Forward",
-                                    targetClass: {
-                                        schemaName: "BisCore",
-                                        className: "InformationPartitionElement",
-                                    },
-                                },
-                                alias: "partition",
-                                isRequired: true,
-                            },
-                        ],
-                        instanceFilter: `(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, "$.Subject.Model.TargetPartition") = printf("0x%x", partition.ECInstanceId)) AND NOT this.IsPrivate AND (json_extract(partition.JsonProperties, "$.PhysicalPartition.Model.Content") <> NULL OR json_extract(partition.JsonProperties, "$.GraphicalPartition3d.Model.Content") <> NULL) AND this.HasRelatedInstance("BisCore:ModelContainsElements", "Forward", "${elementClassSpecification.schemaName}:${elementClassSpecification.className}")`,
-                        hideNodesInHierarchy: true,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
+                    createContentModelsSpecification(context),
                 ],
                 customizationRules: [
                     {
@@ -594,25 +311,7 @@ function createSearchRuleset(props) {
                 ruleType: "ChildNodes",
                 condition: `ParentNode.IsOfClass("GeometricModel3d", "BisCore")`,
                 specifications: [
-                    {
-                        specType: "RelatedInstanceNodes",
-                        relationshipPaths: [
-                            {
-                                relationship: {
-                                    schemaName: "BisCore",
-                                    className: "ModelOwnsSubModel",
-                                },
-                                direction: "Forward",
-                                targetClass: {
-                                    schemaName: "BisCore",
-                                    className: "GeometricModel3d",
-                                },
-                            },
-                        ],
-                        instanceFilter: `NOT this.IsPrivate AND this.HasRelatedInstance("BisCore:ModelContainsElements", "Forward", "${elementClassSpecification.schemaName}:${elementClassSpecification.className}")`,
-                        groupByClass: false,
-                        groupByLabel: false,
-                    },
+                    createModelSubModelsSpecification(context),
                 ],
                 customizationRules: [
                     {
@@ -628,16 +327,251 @@ function createSearchRuleset(props) {
     };
 }
 exports.createSearchRuleset = createSearchRuleset;
-/** @internal */
-function customizeModelsTreeNodeItem(item, node) {
-    item.isCheckboxVisible = true;
-    item.isCheckboxDisabled = true;
-    item.icon = getIcon(node);
+function createRootSubjectSpecification() {
+    return {
+        specType: "InstanceNodesOfSpecificClasses",
+        classes: [
+            {
+                schemaName: "BisCore",
+                classNames: [
+                    "Subject",
+                ],
+            },
+        ],
+        instanceFilter: `this.Parent = NULL`,
+        groupByClass: false,
+        groupByLabel: false,
+    };
+}
+function createRelatedHierarchySubjectSpecification() {
+    return {
+        specType: "RelatedInstanceNodes",
+        relationshipPaths: [
+            {
+                relationship: {
+                    schemaName: "BisCore",
+                    className: "SubjectOwnsSubjects",
+                },
+                direction: "Forward",
+                targetClass: {
+                    schemaName: "BisCore",
+                    className: "Subject",
+                },
+            },
+        ],
+        instanceFilter: `json_extract(this.JsonProperties, "$.Subject.Job.Bridge") <> NULL OR ifnull(json_extract(this.JsonProperties, "$.Subject.Model.Type"), "") = "Hierarchy"`,
+        hideNodesInHierarchy: true,
+        groupByClass: false,
+        groupByLabel: false,
+    };
+}
+function createRelatedNonHierarchySubjectSpecification() {
+    return {
+        specType: "RelatedInstanceNodes",
+        relationshipPaths: [
+            {
+                relationship: {
+                    schemaName: "BisCore",
+                    className: "SubjectOwnsSubjects",
+                },
+                direction: "Forward",
+                targetClass: {
+                    schemaName: "BisCore",
+                    className: "Subject",
+                },
+            },
+        ],
+        instanceFilter: `json_extract(this.JsonProperties, "$.Subject.Job.Bridge") = NULL AND ifnull(json_extract(this.JsonProperties, "$.Subject.Model.Type"), "") <> "Hierarchy"`,
+        hideIfNoChildren: true,
+        groupByClass: false,
+        groupByLabel: false,
+    };
 }
-exports.customizeModelsTreeNodeItem = customizeModelsTreeNodeItem;
-function getIcon(node) {
-    return node.key && presentation_common_1.NodeKey.isClassGroupingNodeKey(node.key)
+function createNonContentModelsSpecification({ elementClassSpecification, showEmptyModels }) {
+    const partitionFilter = `parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, "$.Subject.Model.TargetPartition") = printf("0x%x", partition.ECInstanceId)`;
+    const modelHasElements = `this.HasRelatedInstance("BisCore:ModelContainsElements", "Forward", "${elementClassSpecification.schemaName}:${elementClassSpecification.className}")`;
+    const hasNoContent = `json_extract(partition.JsonProperties, "$.PhysicalPartition.Model.Content") = NULL AND json_extract(partition.JsonProperties, "$.GraphicalPartition3d.Model.Content") = NULL`;
+    const instanceFilter = `(${partitionFilter}) AND NOT this.IsPrivate AND ${hasNoContent}${showEmptyModels ? "" : ` AND ${modelHasElements}`}`;
+    return {
+        specType: "InstanceNodesOfSpecificClasses",
+        classes: {
+            schemaName: "BisCore",
+            classNames: [
+                "GeometricModel3d",
+            ],
+            arePolymorphic: true,
+        },
+        relatedInstances: [
+            {
+                relationshipPath: {
+                    relationship: {
+                        schemaName: "BisCore",
+                        className: "ModelModelsElement",
+                    },
+                    direction: "Forward",
+                    targetClass: {
+                        schemaName: "BisCore",
+                        className: "InformationPartitionElement",
+                    },
+                },
+                alias: "partition",
+                isRequired: true,
+            },
+        ],
+        instanceFilter,
+        hasChildren: showEmptyModels ? "Unknown" : "Always",
+        groupByClass: false,
+        groupByLabel: false,
+    };
+}
+function createContentModelsSpecification({ elementClassSpecification, showEmptyModels }) {
+    const partitionFilter = `parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, "$.Subject.Model.TargetPartition") = printf("0x%x", partition.ECInstanceId)`;
+    const modelHasElements = `this.HasRelatedInstance("BisCore:ModelContainsElements", "Forward", "${elementClassSpecification.schemaName}:${elementClassSpecification.className}")`;
+    const hasContent = `json_extract(partition.JsonProperties, "$.PhysicalPartition.Model.Content") <> NULL OR json_extract(partition.JsonProperties, "$.GraphicalPartition3d.Model.Content") <> NULL`;
+    const instanceFilter = `(${partitionFilter}) AND NOT this.IsPrivate AND (${hasContent})${showEmptyModels ? "" : ` AND ${modelHasElements}`}`;
+    return {
+        specType: "InstanceNodesOfSpecificClasses",
+        classes: {
+            schemaName: "BisCore",
+            classNames: ["GeometricModel3d"],
+            arePolymorphic: true,
+        },
+        relatedInstances: [
+            {
+                relationshipPath: {
+                    relationship: {
+                        schemaName: "BisCore",
+                        className: "ModelModelsElement",
+                    },
+                    direction: "Forward",
+                    targetClass: {
+                        schemaName: "BisCore",
+                        className: "InformationPartitionElement",
+                    },
+                },
+                alias: "partition",
+                isRequired: true,
+            },
+        ],
+        instanceFilter,
+        hasChildren: showEmptyModels ? "Unknown" : "Always",
+        hideNodesInHierarchy: true,
+        groupByClass: false,
+        groupByLabel: false,
+    };
+}
+function createElementModelSpecification({ elementClassSpecification, showEmptyModels }) {
+    const hasElements = `this.HasRelatedInstance("BisCore:ModelContainsElements", "Forward", "${elementClassSpecification.schemaName}:${elementClassSpecification.className}")`;
+    const instanceFilter = `NOT this.IsPrivate${showEmptyModels ? "" : ` AND ${hasElements}`}`;
+    return {
+        specType: "RelatedInstanceNodes",
+        relationshipPaths: [
+            {
+                relationship: {
+                    schemaName: "BisCore",
+                    className: "ModelModelsElement",
+                },
+                direction: "Backward",
+            },
+        ],
+        instanceFilter,
+        hideNodesInHierarchy: true,
+        groupByClass: false,
+        groupByLabel: false,
+    };
+}
+function createModelCategoriesSpecification({ elementClassSpecification }) {
+    return {
+        specType: "RelatedInstanceNodes",
+        relationshipPaths: [
+            [
+                {
+                    relationship: {
+                        schemaName: "BisCore",
+                        className: "ModelContainsElements",
+                    },
+                    direction: "Forward",
+                    targetClass: elementClassSpecification,
+                },
+                {
+                    relationship: {
+                        schemaName: "BisCore",
+                        className: "GeometricElement3dIsInCategory",
+                    },
+                    direction: "Forward",
+                },
+            ],
+        ],
+        instanceFilter: `NOT this.IsPrivate`,
+        suppressSimilarAncestorsCheck: true,
+        hideIfNoChildren: true,
+        groupByClass: false,
+        groupByLabel: false,
+    };
+}
+function createCategoryElementsSpecification({ elementClassSpecification, groupElements }) {
+    return {
+        specType: "RelatedInstanceNodes",
+        relationshipPaths: [
+            {
+                relationship: {
+                    schemaName: "BisCore",
+                    className: "GeometricElement3dIsInCategory",
+                },
+                direction: "Backward",
+                targetClass: elementClassSpecification,
+            },
+        ],
+        instanceFilter: `this.Model.Id = parent.parent.ECInstanceId ANDALSO this.Parent = NULL`,
+        groupByClass: groupElements,
+        groupByLabel: false,
+    };
+}
+function createEementElementsSpecification({ elementClassSpecification, groupElements }) {
+    return {
+        specType: "RelatedInstanceNodes",
+        relationshipPaths: [
+            {
+                relationship: {
+                    schemaName: "BisCore",
+                    className: "ElementOwnsChildElements",
+                },
+                direction: "Forward",
+                targetClass: elementClassSpecification,
+            },
+        ],
+        groupByClass: groupElements,
+        groupByLabel: false,
+    };
+}
+function createModelSubModelsSpecification({ elementClassSpecification, showEmptyModels }) {
+    const hasElements = `this.HasRelatedInstance("BisCore:ModelContainsElements", "Forward", "${elementClassSpecification.schemaName}:${elementClassSpecification.className}")`;
+    const instanceFilter = `NOT this.IsPrivate${showEmptyModels ? "" : ` AND ${hasElements}`}`;
+    return {
+        specType: "RelatedInstanceNodes",
+        relationshipPaths: [
+            {
+                relationship: {
+                    schemaName: "BisCore",
+                    className: "ModelOwnsSubModel",
+                },
+                direction: "Forward",
+                targetClass: {
+                    schemaName: "BisCore",
+                    className: "GeometricModel3d",
+                },
+            },
+        ],
+        instanceFilter,
+        groupByClass: false,
+        groupByLabel: false,
+    };
+}
+/** @internal */
+function addModelsTreeNodeItemIcons(item, node) {
+    item.icon = node.key && presentation_common_1.NodeKey.isClassGroupingNodeKey(node.key)
         ? node.extendedData?.groupIcon
         : node.extendedData?.icon;
 }
+exports.addModelsTreeNodeItemIcons = addModelsTreeNodeItemIcons;
 //# sourceMappingURL=Utils.js.map