mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-21 04:32:23 +00:00
Andrej
63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
# Bonsai - OpenBIM Blender Add-on
|
|
# Copyright (C) 2021 Dion Moult <dion@thinkmoult.com>
|
|
#
|
|
# This file is part of Bonsai.
|
|
#
|
|
# Bonsai 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.
|
|
#
|
|
# Bonsai 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 Bonsai. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import bpy
|
|
import numpy as np
|
|
import ifcopenshell
|
|
import ifcopenshell.api
|
|
import ifcopenshell.api.root
|
|
import ifcopenshell.api.unit
|
|
import ifcopenshell.util.geolocation
|
|
|
|
import test.bim.bootstrap
|
|
import bonsai.core.tool
|
|
import bonsai.tool as tool
|
|
from bonsai.tool import Surveyor as subject
|
|
|
|
|
|
class TestImplementsTool(test.bim.bootstrap.NewFile):
|
|
def test_run(self):
|
|
assert isinstance(subject(), bonsai.core.tool.Surveyor)
|
|
|
|
|
|
class TestGetGlobalMatrix(test.bim.bootstrap.NewFile):
|
|
def test_getting_an_absolute_matrix_if_no_blender_offset(self):
|
|
props = tool.Georeference.get_georeference_props()
|
|
props.has_blender_offset = False
|
|
obj = bpy.data.objects.new("Object", None)
|
|
assert (subject.get_absolute_matrix(obj) == np.array(obj.matrix_world)).all()
|
|
|
|
def test_applying_an_object_placement_blender_offset(self):
|
|
ifc = ifcopenshell.file()
|
|
ifcopenshell.api.root.create_entity(ifc, ifc_class="IfcProject")
|
|
unit = ifcopenshell.api.unit.add_si_unit(ifc, unit_type="LENGTHUNIT", prefix="MILLI")
|
|
ifcopenshell.api.unit.assign_unit(ifc, units=[unit])
|
|
tool.Ifc.set(ifc)
|
|
props = tool.Georeference.get_georeference_props()
|
|
props.has_blender_offset = True
|
|
props.blender_offset_x = "1000"
|
|
props.blender_offset_y = "2000"
|
|
props.blender_offset_z = "3000"
|
|
props.blender_x_axis_abscissa = "0"
|
|
props.blender_x_axis_ordinate = "1"
|
|
obj = bpy.data.objects.new("Object", None)
|
|
props = tool.Blender.get_object_bim_props(obj)
|
|
props.blender_offset_type = "OBJECT_PLACEMENT"
|
|
matrix = ifcopenshell.util.geolocation.local2global(np.array(obj.matrix_world), 1.0, 2.0, 3.0, 0.0, 1.0)
|
|
assert (subject.get_absolute_matrix(obj) == matrix).all()
|