mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-17 02:49:12 +00:00
Dion Moult
188 lines
7.4 KiB
Python
188 lines
7.4 KiB
Python
# BlenderBIM Add-on - OpenBIM Blender Add-on
|
|
# Copyright (C) 2021 Dion Moult <dion@thinkmoult.com>
|
|
#
|
|
# This file is part of BlenderBIM Add-on.
|
|
#
|
|
# BlenderBIM Add-on is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# BlenderBIM Add-on is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with BlenderBIM Add-on. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import bpy
|
|
import math
|
|
import numpy
|
|
import ifcopenshell
|
|
import test.bim.bootstrap
|
|
import blenderbim.core.tool
|
|
import blenderbim.tool as tool
|
|
from blenderbim.tool.misc import Misc as subject
|
|
from blenderbim.bim.ifc import IfcStore
|
|
|
|
|
|
class TestImplementsTool(test.bim.bootstrap.NewFile):
|
|
def test_run(self):
|
|
assert isinstance(subject(), blenderbim.core.tool.Misc)
|
|
|
|
|
|
class TestGetObjectStorey(test.bim.bootstrap.NewFile):
|
|
def test_run(self):
|
|
ifc = ifcopenshell.file()
|
|
tool.Ifc.set(ifc)
|
|
obj = bpy.data.objects.new("Object", None)
|
|
wall = ifc.createIfcWall()
|
|
tool.Ifc.link(wall, obj)
|
|
storey = ifc.createIfcBuildingStorey()
|
|
ifcopenshell.api.run("spatial.assign_container", ifc, product=wall, relating_structure=storey)
|
|
assert subject.get_object_storey(obj) == storey
|
|
|
|
def test_only_returning_a_building_storey(self):
|
|
ifc = ifcopenshell.file()
|
|
tool.Ifc.set(ifc)
|
|
obj = bpy.data.objects.new("Object", None)
|
|
wall = ifc.createIfcWall()
|
|
tool.Ifc.link(wall, obj)
|
|
building = ifc.createIfcBuilding()
|
|
ifcopenshell.api.run("spatial.assign_container", ifc, product=wall, relating_structure=building)
|
|
assert subject.get_object_storey(obj) is None
|
|
|
|
def test_returning_nothing_if_uncontained(self):
|
|
ifc = ifcopenshell.file()
|
|
tool.Ifc.set(ifc)
|
|
obj = bpy.data.objects.new("Object", None)
|
|
wall = ifc.createIfcWall()
|
|
tool.Ifc.link(wall, obj)
|
|
assert subject.get_object_storey(obj) is None
|
|
|
|
|
|
class TestGetStoreyElevation(test.bim.bootstrap.NewFile):
|
|
def test_run(self):
|
|
ifc = ifcopenshell.file()
|
|
ifc.createIfcProject()
|
|
unit = ifcopenshell.api.run("unit.add_si_unit", ifc, prefix="MILLI")
|
|
ifcopenshell.api.run("unit.assign_unit", ifc, units=[unit])
|
|
storey = ifc.createIfcBuildingStorey()
|
|
storey.Elevation = 3000
|
|
tool.Ifc.set(ifc)
|
|
assert subject.get_storey_elevation_in_si(storey) == 3.0
|
|
|
|
|
|
class TestGetStoreyHeight(test.bim.bootstrap.NewFile):
|
|
def test_run(self):
|
|
ifc = ifcopenshell.file()
|
|
ifc.createIfcProject()
|
|
unit = ifcopenshell.api.run("unit.add_si_unit", ifc, prefix="MILLI")
|
|
ifcopenshell.api.run("unit.assign_unit", ifc, units=[unit])
|
|
tool.Ifc.set(ifc)
|
|
building = ifc.createIfcBuilding()
|
|
storey = ifc.createIfcBuildingStorey()
|
|
storey.Elevation = 3000
|
|
storey2 = ifc.createIfcBuildingStorey()
|
|
storey2.Elevation = 5000
|
|
ifcopenshell.api.run("aggregate.assign_object", ifc, product=storey, relating_object=building)
|
|
ifcopenshell.api.run("aggregate.assign_object", ifc, product=storey2, relating_object=building)
|
|
assert subject.get_storey_height_in_si(storey, 1) == 2.0
|
|
|
|
def test_getting_a_double_storey_height(self):
|
|
ifc = ifcopenshell.file()
|
|
ifc.createIfcProject()
|
|
unit = ifcopenshell.api.run("unit.add_si_unit", ifc, prefix="MILLI")
|
|
ifcopenshell.api.run("unit.assign_unit", ifc, units=[unit])
|
|
tool.Ifc.set(ifc)
|
|
building = ifc.createIfcBuilding()
|
|
storey = ifc.createIfcBuildingStorey()
|
|
storey.Elevation = 3000
|
|
storey2 = ifc.createIfcBuildingStorey()
|
|
storey2.Elevation = 5000
|
|
storey3 = ifc.createIfcBuildingStorey()
|
|
storey3.Elevation = 9000
|
|
ifcopenshell.api.run("aggregate.assign_object", ifc, product=storey, relating_object=building)
|
|
ifcopenshell.api.run("aggregate.assign_object", ifc, product=storey2, relating_object=building)
|
|
ifcopenshell.api.run("aggregate.assign_object", ifc, product=storey3, relating_object=building)
|
|
assert subject.get_storey_height_in_si(storey, 2) == 6.0
|
|
|
|
def test_only_considering_storeys_in_the_same_building(self):
|
|
ifc = ifcopenshell.file()
|
|
tool.Ifc.set(ifc)
|
|
building = ifc.createIfcBuilding()
|
|
storey = ifc.createIfcBuildingStorey()
|
|
storey.Elevation = 3000
|
|
storey2 = ifc.createIfcBuildingStorey()
|
|
storey2.Elevation = 5000
|
|
ifcopenshell.api.run("aggregate.assign_object", ifc, product=storey, relating_object=building)
|
|
assert subject.get_storey_height_in_si(storey, 1) is None
|
|
|
|
def test_returning_none_if_the_storey_height_is_undefined(self):
|
|
ifc = ifcopenshell.file()
|
|
tool.Ifc.set(ifc)
|
|
building = ifc.createIfcBuilding()
|
|
storey = ifc.createIfcBuildingStorey()
|
|
ifcopenshell.api.run("aggregate.assign_object", ifc, product=storey, relating_object=building)
|
|
assert subject.get_storey_height_in_si(storey, 1) is None
|
|
|
|
|
|
class TestSetObjectOriginToBottom(test.bim.bootstrap.NewFile):
|
|
def test_run(self):
|
|
bpy.ops.mesh.primitive_cube_add()
|
|
obj = bpy.data.objects.get("Cube")
|
|
subject.set_object_origin_to_bottom(obj)
|
|
assert list(obj.matrix_world.translation) == [0.0, 0.0, -1.0]
|
|
|
|
def test_moving_the_origin_of_a_rotated_object(self):
|
|
bpy.ops.mesh.primitive_cube_add()
|
|
obj = bpy.data.objects.get("Cube")
|
|
obj.rotation_euler[1] = math.pi / 2
|
|
bpy.context.view_layer.update()
|
|
subject.set_object_origin_to_bottom(obj)
|
|
assert list(obj.matrix_world.translation) == [0.0, 0.0, -1.0]
|
|
|
|
|
|
class TestMoveObjectToElevation(test.bim.bootstrap.NewFile):
|
|
def test_run(self):
|
|
bpy.ops.mesh.primitive_cube_add()
|
|
obj = bpy.data.objects.get("Cube")
|
|
subject.move_object_to_elevation(obj, 3)
|
|
assert list(obj.matrix_world.translation) == [0.0, 0.0, 3.0]
|
|
|
|
|
|
class TestRunRootCopyClass(test.bim.bootstrap.NewFile):
|
|
def test_nothing(self):
|
|
pass
|
|
|
|
|
|
class TestScaleObjectToHeight(test.bim.bootstrap.NewFile):
|
|
def test_run(self):
|
|
bpy.ops.mesh.primitive_cube_add()
|
|
obj = bpy.data.objects.get("Cube")
|
|
subject.set_object_origin_to_bottom(obj)
|
|
subject.scale_object_to_height(obj, 3)
|
|
assert obj.dimensions[2] == 3.0
|
|
assert list(obj.scale) == [1.0, 1.0, 1.0]
|
|
|
|
|
|
class TestMarkObjectAsEdited(test.bim.bootstrap.NewFile):
|
|
def test_run(self):
|
|
obj = bpy.data.objects.new("Cube", None)
|
|
subject.mark_object_as_edited(obj)
|
|
assert obj in IfcStore.edited_objs
|
|
|
|
|
|
class TestSplitObjectsWithCutter(test.bim.bootstrap.NewFile):
|
|
def test_run(self):
|
|
bpy.ops.mesh.primitive_cube_add()
|
|
bpy.ops.mesh.primitive_plane_add(size=4)
|
|
obj = bpy.data.objects.get("Cube")
|
|
cutter = bpy.data.objects.get("Plane")
|
|
assert obj.dimensions[2] == 2
|
|
new_objs = subject.split_objects_with_cutter([obj], cutter)
|
|
assert new_objs[0].name == "Cube.001"
|
|
assert obj.dimensions[2] == 1
|
|
assert new_objs[0].dimensions[2] == 1
|