@itwin/core-backend 5.1.0-dev.9 → 5.2.0-dev.2

package/lib/esm/test/standalone/GeometryStream.test.js

@@ -6,7 +6,7 @@ import { assert, expect } from "chai";
 import { BentleyStatus, Id64, IModelStatus } from "@itwin/core-bentley";
 import { Angle, AngleSweep, Arc3d, Box, ClipMaskXYZRangePlanes, ClipPlane, ClipPlaneContainment, ClipPrimitive, ClipShape, ClipVector, ConvexClipPlaneSet, CurveCollection, CurvePrimitive, Geometry, IndexedPolyface, InterpolationCurve3d, InterpolationCurve3dOptions, LineSegment3d, LineString3d, Loop, Matrix3d, Plane3dByOriginAndUnitNormal, Point2d, Point3d, Point3dArray, PointString3d, PolyfaceBuilder, Range2d, Range3d, SolidPrimitive, Sphere, StrokeOptions, Transform, Vector3d, YawPitchRollAngles, } from "@itwin/core-geometry";
 import { AreaPattern, BackgroundFill, BRepGeometryOperation, Code, ColorByName, ColorDef, ElementGeometry, ElementGeometryOpcode, FillDisplay, GeometryClass, GeometryParams, GeometryStreamBuilder, GeometryStreamFlags, GeometryStreamIterator, Gradient, ImageGraphicCorners, IModel, LinePixels, LineStyle, MassPropertiesOperation, Placement3d, TextString, ThematicGradientMode, ThematicGradientSettings, ViewFlags, } from "@itwin/core-common";
-import { _nativeDb, DefinitionModel, deleteElementTree, GeometryPart, LineStyleDefinition, PhysicalObject, Subject } from "../../core-backend";
+import { _nativeDb, DefinitionModel, deleteElementTree, GeometryPart, LineStyleDefinition, PhysicalObject, SubCategory, Subject } from "../../core-backend";
 import { createBRepDataProps } from "../GeometryTestUtil";
 import { IModelTestUtils } from "../IModelTestUtils";
 import { Timer } from "../TestUtils";
@@ -933,6 +933,54 @@ describe("GeometryStream", () => {
         assert.isTrue(usageInfo.geometryPartIds.includes(partId));
         assert.isTrue(usageInfo.usedIds.includes(partId));
     });
+    it("create GeometricElement3d with material and sub-category", async () => {
+        const seedElement = imodel.elements.getElement("0x1d");
+        assert.exists(seedElement);
+        const subCategory = SubCategory.create(imodel, seedElement.category, "testSubCat", { weight: 2 });
+        const subCategoryId = imodel.elements.insertElement(subCategory.toJSON());
+        imodel.saveChanges();
+        const params = new GeometryParams(seedElement.category);
+        params.subCategoryId = subCategoryId;
+        params.materialId = "0x5";
+        const shape = Loop.create(LineString3d.create(Point3d.create(0, 0, 0), Point3d.create(1, 0, 0), Point3d.create(1, 1, 0), Point3d.create(0, 1, 0), Point3d.create(0, 0, 0)));
+        // Test flatbuffer roundtrip...
+        const builderF = new ElementGeometry.Builder();
+        builderF.appendGeometryParamsChange(params);
+        builderF.appendGeometryQuery(shape);
+        const elementPropsF = createPhysicalElementProps(seedElement);
+        elementPropsF.elementGeometryBuilderParams = { entryArray: builderF.entries };
+        const newIdF = imodel.elements.insertElement(elementPropsF);
+        assert.isTrue(Id64.isValidId64(newIdF));
+        imodel.saveChanges();
+        const onGeometry = (info) => {
+            assert.isTrue(undefined !== info.entryArray);
+            const it = new ElementGeometry.Iterator(info);
+            for (const entry of it) {
+                if (ElementGeometry.isGeometryQueryEntry(entry.value)) {
+                    assert.isTrue(params.subCategoryId === entry.geomParams.subCategoryId && params.materialId === entry.geomParams.materialId);
+                }
+            }
+            ;
+        };
+        const status = imodel.elementGeometryRequest({ elementId: newIdF, onGeometry });
+        assert(IModelStatus.Success === status);
+        // Test JSON roundtrip...
+        const builderJ = new GeometryStreamBuilder();
+        builderJ.appendGeometryParamsChange(params);
+        builderJ.appendGeometry(shape);
+        const elementPropsJ = createPhysicalElementProps(seedElement);
+        elementPropsJ.geom = builderJ.geometryStream;
+        const newIdJ = imodel.elements.insertElement(elementPropsJ);
+        assert.isTrue(Id64.isValidId64(newIdJ));
+        imodel.saveChanges();
+        const value = imodel.elements.getElementProps({ id: newIdJ, wantGeometry: true });
+        assert.isDefined(value.geom);
+        const itLocal = new GeometryStreamIterator(value.geom, value.category);
+        for (const entry of itLocal) {
+            assert.equal(entry.primitive.type, "geometryQuery");
+            assert.isTrue(params.subCategoryId === entry.geomParams.subCategoryId && params.materialId === entry.geomParams.materialId);
+        }
+    });
     it("create GeometricElement3d from world coordinate text using a newly added font", async () => {
         // Set up element to be placed in iModel
         const seedElement = imodel.elements.getElement("0x1d");