diff --git a/src/ifcopenshell-python/ifcopenshell/api/geometry/add_axis_representation.py b/src/ifcopenshell-python/ifcopenshell/api/geometry/add_axis_representation.py new file mode 100644 index 0000000000..ecb6870345 --- /dev/null +++ b/src/ifcopenshell-python/ifcopenshell/api/geometry/add_axis_representation.py @@ -0,0 +1,49 @@ +# IfcOpenShell - IFC toolkit and geometry engine +# Copyright (C) 2022 Dion Moult +# +# This file is part of IfcOpenShell. +# +# IfcOpenShell is free software: you can redistribute it and/or modify +# it under the terms of the GNU Lesser General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# IfcOpenShell 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 Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public License +# along with IfcOpenShell. If not, see . + +import ifcopenshell.util.unit + + +class Usecase: + def __init__(self, file, **settings): + self.file = file + self.settings = { + "context": None, # IfcGeometricRepresentationContext + "axis": [], # A list of ordered coordinates for the axis + } + for key, value in settings.items(): + self.settings[key] = value + + def execute(self): + self.settings["unit_scale"] = ifcopenshell.util.unit.calculate_unit_scale(self.file) + points = [self.convert_si_to_unit(p) for p in self.settings["axis"]] + if self.file.schema == "IFC2X3": + curve = self.file.createIfcPolyline([self.file.createIfcCartesianPoint(p) for p in points]) + else: + curve = self.file.createIfcIndexedPolyCurve(self.file.createIfcCartesianPointList3D(points)) + return self.file.createIfcShapeRepresentation( + self.settings["context"], + self.settings["context"].ContextIdentifier, + "Curve2D" if len(self.settings["axis"][0]) == 2 else "Curve3D", + [curve], + ) + + def convert_si_to_unit(self, co): + if isinstance(co, (tuple, list)): + return [self.convert_si_to_unit(o) for o in co] + return co / self.settings["unit_scale"]