WIP refactor of geometry for stabilisation and optimisation. Switch representation now done.

src/blenderbim/test/tool/test_geometry.py

@@ -33,6 +33,55 @@ class TestImplementsTool(test.bim.bootstrap.NewFile):
         assert isinstance(subject(), blenderbim.core.tool.Geometry)
 
 
+class TestChangeObjectData(test.bim.bootstrap.NewFile):
+    def test_change_single_object_data(self):
+        data1 = bpy.data.meshes.new("Mesh")
+        data2 = bpy.data.meshes.new("Mesh")
+        obj1 = bpy.data.objects.new("Object", data1)
+        obj2 = bpy.data.objects.new("Object", data1)
+        subject.change_object_data(obj1, data2, is_global=False)
+        assert obj1.data == data2
+        assert obj2.data == data1
+
+    def test_change_object_data_globally(self):
+        data1 = bpy.data.meshes.new("Mesh")
+        data2 = bpy.data.meshes.new("Mesh")
+        obj1 = bpy.data.objects.new("Object", data1)
+        obj2 = bpy.data.objects.new("Object", data1)
+        subject.change_object_data(obj1, data2, is_global=True)
+        assert obj1.data == data2
+        assert obj2.data == data2
+
+
+class TestClearDynamicVoids(test.bim.bootstrap.NewFile):
+    def test_run(self):
+        obj = bpy.data.objects.new("Object", bpy.data.meshes.new("Mesh"))
+        obj.modifiers.new("IfcOpeningElement", "BOOLEAN")
+        subject.clear_dynamic_voids(obj)
+        assert len(obj.modifiers) == 0
+
+
+class TestCreateDynamicVoids(test.bim.bootstrap.NewFile):
+    def test_run(self):
+        ifc = ifcopenshell.file()
+        tool.Ifc.set(ifc)
+        wall = ifc.createIfcWall()
+        opening = ifc.createIfcOpeningElement()
+        ifcopenshell.api.run("void.add_opening", ifc, opening=opening, element=wall)
+        wall_obj = bpy.data.objects.new("Object", bpy.data.meshes.new("Mesh"))
+        opening_obj = bpy.data.objects.new("Object", bpy.data.meshes.new("Mesh"))
+        tool.Ifc.link(wall, wall_obj)
+        tool.Ifc.link(opening, opening_obj)
+        subject.create_dynamic_voids(wall_obj)
+        modifier = wall_obj.modifiers[0]
+        assert modifier.type == "BOOLEAN"
+        assert modifier.name == "IfcOpeningElement"
+        assert modifier.operation == "DIFFERENCE"
+        assert modifier.object == opening_obj
+        assert modifier.solver == "EXACT"
+        assert modifier.use_self is True
+
+
 class TestDoesObjectHaveMeshWithFaces(test.bim.bootstrap.NewFile):
     def test_empties_return_false(self):
         obj = bpy.data.objects.new("Object", None)
@@ -81,12 +130,23 @@ class TestGetObjectMaterialsWithoutStyles(test.bim.bootstrap.NewFile):
         assert subject.get_object_materials_without_styles(obj) == [material1, material2]
 
 
+class TestGetRepresentationData(test.bim.bootstrap.NewFile):
+    def test_run(self):
+        ifc = ifcopenshell.file()
+        context = ifc.createIfcGeometricRepresentationContext()
+        representation = ifc.createIfcShapeRepresentation()
+        representation.ContextOfItems = context
+        data = bpy.data.meshes.new("1/2")
+        assert subject.get_representation_data(representation) == data
+
+
 class TestGetRepresentationName(test.bim.bootstrap.NewFile):
     def test_run(self):
         ifc = ifcopenshell.file()
         context = ifc.createIfcGeometricRepresentationContext()
         representation = ifc.createIfcShapeRepresentation()
-        assert subject.get_representation_name(context, representation) == f"{context.id()}/{representation.id()}"
+        representation.ContextOfItems = context
+        assert subject.get_representation_name(representation) == f"{context.id()}/{representation.id()}"
 
 
 class TestGetCartesianPointCoordinateOffset(test.bim.bootstrap.NewFile):
@@ -127,6 +187,44 @@ class TestGetTotalRepresentationItems(test.bim.bootstrap.NewFile):
         assert subject.get_total_representation_items(obj) == 2
 
 
+class TestImportRepresentation(test.bim.bootstrap.NewFile):
+    def test_importing_a_normal_shape(self):
+        ifc = ifcopenshell.open("test/files/basic.ifc")
+        tool.Ifc.set(ifc)
+        obj = bpy.data.objects.new("Object", bpy.data.meshes.new("Mesh"))
+        material = bpy.data.materials.new("Material")
+        material.BIMMaterialProperties.ifc_style_id = 101
+        element = ifc.by_type("IfcWall")[0]
+        tool.Ifc.link(element, obj)
+        representation = element.Representation.Representations[0]
+        mesh = subject.import_representation(obj, representation, enable_dynamic_voids=False)
+        assert len(mesh.polygons) == 12
+        assert mesh.materials[0] == material
+
+    def test_importing_non_body_curves(self):
+        ifc = ifcopenshell.open("test/files/annotation.ifc")
+        tool.Ifc.set(ifc)
+        obj = bpy.data.objects.new("Object", bpy.data.meshes.new("Mesh"))
+        element = ifc.by_type("IfcWall")[0]
+        tool.Ifc.link(element, obj)
+        representation = element.Representation.Representations[0]
+        mesh = subject.import_representation(obj, representation, enable_dynamic_voids=False)
+        assert len(mesh.polygons) == 0
+        assert len(mesh.edges) == 4
+
+
+class TestIsBodyRepresentation(test.bim.bootstrap.NewFile):
+    def test_run(self):
+        ifc = ifcopenshell.file()
+        context = ifc.createIfcGeometricRepresentationContext()
+        representation = ifc.createIfcShapeRepresentation()
+        representation.ContextOfItems = context
+        context.ContextIdentifier = "Body"
+        assert subject.is_body_representation(representation) is True
+        context.ContextIdentifier = "Clearance"
+        assert subject.is_body_representation(representation) is False
+
+
 class TestLink(test.bim.bootstrap.NewFile):
     def test_run(self):
         ifc = ifcopenshell.file()
@@ -139,10 +237,27 @@ class TestLink(test.bim.bootstrap.NewFile):
 class TestRenameObjectData(test.bim.bootstrap.NewFile):
     def test_run(self):
         obj = bpy.data.meshes.new("Mesh")
-        subject.rename_object_data(obj, "name")
+        subject.rename_object(obj, "name")
         assert obj.name == "name"
 
 
+class TestResolveMappedRepresentation(test.bim.bootstrap.NewFile):
+    def test_run(self):
+        ifc = ifcopenshell.file()
+        mapped_representation = ifc.createIfcShapeRepresentation()
+        representation = ifc.createIfcShapeRepresentation()
+        mapped_representation.RepresentationType = "MappedRepresentation"
+        mapped_representation.Items = [
+            ifc.createIfcMappedItem(MappingSource=ifc.createIfcRepresentationMap(MappedRepresentation=representation))
+        ]
+        assert subject.resolve_mapped_representation(mapped_representation) == representation
+
+    def test_return_a_representation_if_not_mapped(self):
+        ifc = ifcopenshell.file()
+        representation = ifc.createIfcShapeRepresentation()
+        assert subject.resolve_mapped_representation(representation) == representation
+
+
 class TestShouldForceFacetedBrep(test.bim.bootstrap.NewFile):
     def test_run(self):
         result = bpy.context.scene.BIMGeometryProperties.should_force_faceted_brep