Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-21 14:23:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

mahtutils deprecation - translate #5192

Browse Source
This commit is contained in:
Andrej730
2024-12-16 16:53:00 +05:00
parent c1ec42c95b
commit 8250f02866
1 changed files with 31 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/util/shape_builder.py
+31 -19
View File
@@ -393,41 +393,47 @@ class ShapeBuilder:
def translate( def translate(
self, self,
curve_or_item: Union[ifcopenshell.entity_instance, list[ifcopenshell.entity_instance]], curve_or_item: Union[ifcopenshell.entity_instance, Sequence[ifcopenshell.entity_instance]],
translation: Vector, translation: VectorType,
create_copy: bool = False, create_copy: bool = False,
) -> Union[ifcopenshell.entity_instance, list[ifcopenshell.entity_instance]]: ) -> Union[ifcopenshell.entity_instance, list[ifcopenshell.entity_instance]]:
# > curve_or_item - could be a list of curves or items or representations """
# < returns translated object Translate curve/representaiton item/representation.
:param curve_or_item: A single item to translate or a sequence of them.
:param translation: Translation vector.
:param create_copy: Whether to translate the provided item or it's copy.
:return: Translated curve/item/representation or a sequence of them.
"""
multiple_objects = isinstance(curve_or_item, collections.abc.Iterable) multiple_objects = isinstance(curve_or_item, collections.abc.Iterable)
if not multiple_objects: if not multiple_objects:
curve_or_item = [curve_or_item] curve_or_item = [curve_or_item]
processed_objects = [] processed_objects: list[ifcopenshell.entity_instance] = []
for c in curve_or_item: for c in curve_or_item:
if create_copy: if create_copy:
c = ifcopenshell.util.element.copy_deep(self.file, c) c = ifcopenshell.util.element.copy_deep(self.file, c)
if c.is_a() in ("IfcIndexedPolyCurve", "IfcPolyline"): if c.is_a() in ("IfcIndexedPolyCurve", "IfcPolyline"):
coords = self.get_polyline_coords(c) coords = self.get_polyline_coords(c)
coords = [Vector(co) + translation for co in coords] coords += translation
self.set_polyline_coords(c, coords) self.set_polyline_coords(c, coords)
elif c.is_a("IfcCircle") or c.is_a("IfcExtrudedAreaSolid") or c.is_a("IfcEllipse"): elif c.is_a("IfcCircle") or c.is_a("IfcExtrudedAreaSolid") or c.is_a("IfcEllipse"):
base_position = Vector(c.Position.Location.Coordinates) base_position = np.array(c.Position.Location.Coordinates)
c.Position.Location.Coordinates = base_position + translation c.Position.Location.Coordinates = ifc_safe_vector_type(base_position + translation)
elif c.is_a("IfcShapeRepresentation"): elif c.is_a("IfcShapeRepresentation"):
for item in c.Items: for item in c.Items:
self.translate(item, translation) self.translate(item, translation)
elif c.is_a("IfcTrimmedCurve"): elif c.is_a("IfcTrimmedCurve"):
base_position = Vector(c.Trim1[0].Coordinates) base_position = np.array(c.Trim1[0].Coordinates)
c.Trim1[0].Coordinates = base_position + translation c.Trim1[0].Coordinates = ifc_safe_vector_type(base_position + translation)
base_position = Vector(c.Trim2[0].Coordinates) base_position = np.array(c.Trim2[0].Coordinates)
c.Trim2[0].Coordinates = base_position + translation c.Trim2[0].Coordinates = ifc_safe_vector_type(base_position + translation)
self.translate(c.BasisCurve, translation) self.translate(c.BasisCurve, translation)
@@ -874,22 +880,28 @@ class ShapeBuilder:
kwargs["position_z_axis"].rotate(rot) kwargs["position_z_axis"].rotate(rot)
return kwargs return kwargs
def get_polyline_coords(self, polyline: ifcopenshell.entity_instance) -> list[Vector]: def get_polyline_coords(self, polyline: ifcopenshell.entity_instance) -> np.ndarray:
"""polyline should be either `IfcIndexedPolyCurve` or `IfcPolyline`""" """polyline should be either `IfcIndexedPolyCurve` or `IfcPolyline`"""
coords = None coords = None
if polyline.is_a("IfcIndexedPolyCurve"): if polyline.is_a("IfcIndexedPolyCurve"):
coords = polyline.Points.CoordList coords = np.array(polyline.Points.CoordList)
elif polyline.is_a("IfcPolyline"): elif polyline.is_a("IfcPolyline"):
coords = [p.Coordinates for p in polyline.Points] coords = np.array(p.Coordinates for p in polyline.Points)
else:
raise Exception(f"Unsupported polyline type: {polyline.is_a()}")
return coords return coords
def set_polyline_coords(self, polyline: ifcopenshell.entity_instance, coords: list[Vector]) -> None: def set_polyline_coords(self, polyline: ifcopenshell.entity_instance, coords: SequenceOfVectors) -> None:
"""polyline should be either `IfcIndexedPolyCurve` or `IfcPolyline`""" """polyline should be either `IfcIndexedPolyCurve` or `IfcPolyline`"""
if polyline.is_a("IfcIndexedPolyCurve"): if polyline.is_a("IfcIndexedPolyCurve"):
polyline.Points.CoordList = coords polyline.Points.CoordList = ifc_safe_vector_type(coords)
elif polyline.is_a("IfcPolyline"): elif polyline.is_a("IfcPolyline"):
for i, co in enumerate(coords): ifc_points: list[ifcopenshell.entity_instance] = polyline.Points
polyline.Points[i].Coordinates = co assert len(ifc_points) == len(coords)
for point, co in zip(ifc_points, ifc_safe_vector_type(coords)):
point.Coordinates = co
else:
raise Exception(f"Unsupported polyline type: {polyline.is_a()}")
def get_simple_2dcurve_data( def get_simple_2dcurve_data(
self, self,