New IfcPatch recipe to offset and rotate an entire model

2026-08-10 09:48:32 +00:00 · 2020-04-21 18:55:28 +10:00
parent 048966bda2
commit 2bb57aebb5
1 changed files with 62 additions and 0 deletions
@@ -0,0 +1,62 @@
+import math
+
+class Patcher:
+    def __init__(self, file, logger, args=None):
+        self.file = file
+        self.logger = logger
+        self.args = args
+
+    def patch(self):
+        absolute_placements = []
+
+        for product in self.file.by_type('IfcProduct'):
+            if not product.ObjectPlacement:
+                continue
+            absolute_placement = self.get_absolute_placement(product.ObjectPlacement)
+            if absolute_placement.is_a('IfcLocalPlacement'):
+                absolute_placements.append(absolute_placement)
+        absolute_placements = set(absolute_placements)
+
+        for placement in absolute_placements:
+            offset_location = (
+                placement.RelativePlacement.Location.Coordinates[0] + float(self.args[0]),
+                placement.RelativePlacement.Location.Coordinates[1] + float(self.args[1]),
+                placement.RelativePlacement.Location.Coordinates[2] + float(self.args[2])
+            )
+
+            angle = float(self.args[3])
+            if not angle:
+                placement.RelativePlacement.Location.Coordinates = offset_location
+                continue
+
+            rotation_matrix = self.z_rotation_matrix(math.radians(angle))
+            placement.RelativePlacement.Location.Coordinates = self.multiply_by_matrix(offset_location, rotation_matrix)
+
+            if placement.RelativePlacement.Axis:
+                z_axis = placement.RelativePlacement.Axis.DirectionRatios
+                placement.RelativePlacement.Axis.DirectionRatios = self.multiply_by_matrix(z_axis, rotation_matrix)
+
+            if placement.RelativePlacement.RefDirection:
+                x_axis = placement.RelativePlacement.RefDirection.DirectionRatios
+                placement.RelativePlacement.RefDirection.DirectionRatios = self.multiply_by_matrix(x_axis, rotation_matrix)
+            else:
+                x_axis = self.file.createIfcDirection(self.multiply_by_matrix((1., 0., 0.), rotation_matrix))
+
+    def get_absolute_placement(self, object_placement):
+        if object_placement.PlacementRelTo:
+            return self.get_absolute_placement(object_placement.PlacementRelTo)
+        return object_placement
+
+    def z_rotation_matrix(self, angle):
+        return [
+            [math.cos(angle), -math.sin(angle), 0.],
+            [math.sin(angle), math.cos(angle), 0.],
+            [0., 0., 1.]
+        ]
+
+    def multiply_by_matrix(self, v, m):
+        return [
+            v[0]*m[0][0] + v[1]*m[0][1] + v[2]*m[0][2],
+            v[0]*m[1][0] + v[1]*m[1][1] + v[2]*m[1][2],
+            v[0]*m[2][0] + v[1]*m[2][1] + v[2]*m[2][2]
+        ]