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

lint with black

Browse Source
This commit is contained in:
civilx64
2024-10-26 10:16:24 -04:00
committed by Thomas Krijnen
parent 8d001d0120
commit 8ff4cef7ec
1 changed files with 89 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/alignment.py
+89 -81
View File
@@ -39,7 +39,9 @@ def evaluate_representation(shape_rep: entity_instance, dist_along: float) -> np
supported_rep_types = ["IFCCOMPOSITECURVE", "IFCGRADIENTCURVE", "IFCSEGMENTEDREFERENCECURVE"] supported_rep_types = ["IFCCOMPOSITECURVE", "IFCGRADIENTCURVE", "IFCSEGMENTEDREFERENCECURVE"]
shape_rep_type = shape_rep.is_a().upper() shape_rep_type = shape_rep.is_a().upper()
if not shape_rep_type in supported_rep_types: if not shape_rep_type in supported_rep_types:
raise NotImplementedError(f"Expected entity type to be one of {[_ for _ in supported_rep_types]}, got '{shape_rep_type}") raise NotImplementedError(
f"Expected entity type to be one of {[_ for _ in supported_rep_types]}, got '{shape_rep_type}"
)
# TODO: confirm point is not beyond limits of alignment # TODO: confirm point is not beyond limits of alignment
@@ -124,15 +126,16 @@ class IfcAlignmentHelper:
# TODO: add missing functionality noted in the docstring # TODO: add missing functionality noted in the docstring
def __init__(self, def __init__(
file: ifcopenshell.file = None, self,
filename: str = None, file: ifcopenshell.file = None,
creator: str = None, filename: str = None,
organization: str = None, creator: str = None,
application: str = None, organization: str = None,
project_globalid=None, application: str = None,
project_name: str = None project_globalid=None,
): project_name: str = None,
):
""" """
@param file: An existing model that the alignment will be added to @param file: An existing model that the alignment will be added to
@param filename: Name for a new model to be created that will contain the alignment @param filename: Name for a new model to be created that will contain the alignment
@@ -157,27 +160,25 @@ class IfcAlignmentHelper:
self._geom_context = self._file.by_type("IfcGeometricRepresentationContext")[0] self._geom_context = self._file.by_type("IfcGeometricRepresentationContext")[0]
self._axis_geom_subcontext = self._file.createIfcGeometricRepresentationSubContext( self._axis_geom_subcontext = self._file.createIfcGeometricRepresentationSubContext(
ContextIdentifier="Axis", ContextIdentifier="Axis", ContextType="Model", ParentContext=self._geom_context, TargetView="GRAPH_VIEW"
ContextType="Model",
ParentContext=self._geom_context,
TargetView="GRAPH_VIEW"
) )
def _create_segment_representations(self, global_placement: entity_instance, def _create_segment_representations(
curve_segments: Sequence[entity_instance], segments: Sequence[entity_instance]): self,
global_placement: entity_instance,
curve_segments: Sequence[entity_instance],
segments: Sequence[entity_instance],
):
for curve_segment, alignment_segment in zip(curve_segments, segments): for curve_segment, alignment_segment in zip(curve_segments, segments):
axis_representation = self._file.create_entity( axis_representation = self._file.create_entity(
type="IfcShapeRepresentation", type="IfcShapeRepresentation",
ContextOfItems=self._axis_geom_subcontext, ContextOfItems=self._axis_geom_subcontext,
RepresentationIdentifier="Axis", RepresentationIdentifier="Axis",
RepresentationType="Segment", RepresentationType="Segment",
Items=(curve_segment,) Items=(curve_segment,),
) )
product = self._file.create_entity( product = self._file.create_entity(
type="IfcProductDefinitionShape", type="IfcProductDefinitionShape", Name=None, Description=None, Representations=(axis_representation,)
Name=None,
Description=None,
Representations=(axis_representation,)
) )
alignment_segment.ObjectPlacement = global_placement alignment_segment.ObjectPlacement = global_placement
alignment_segment.Representation = product alignment_segment.Representation = product
@@ -206,16 +207,16 @@ class IfcAlignmentHelper:
type="IfcLine", type="IfcLine",
Pnt=self._file.create_entity( Pnt=self._file.create_entity(
type="IfcCartesianPoint", type="IfcCartesianPoint",
Coordinates=(0., 0.), Coordinates=(0.0, 0.0),
), ),
Dir=self._file.create_entity( Dir=self._file.create_entity(
type="IfcVector", type="IfcVector",
Orientation=self._file.create_entity( Orientation=self._file.create_entity(
type="IfcDirection", type="IfcDirection",
DirectionRatios=(1., 0.), DirectionRatios=(1.0, 0.0),
), ),
Magnitude=1., Magnitude=1.0,
) ),
) )
curve_segment = self._file.create_entity( curve_segment = self._file.create_entity(
type="IfcCurveSegment", type="IfcCurveSegment",
@@ -227,7 +228,7 @@ class IfcAlignmentHelper:
(math.cos(start_direction), math.sin(start_direction)), (math.cos(start_direction), math.sin(start_direction)),
), ),
), ),
SegmentStart=self._file.createIfcLengthMeasure(0.), SegmentStart=self._file.createIfcLengthMeasure(0.0),
SegmentLength=self._file.createIfcLengthMeasure(length), SegmentLength=self._file.createIfcLengthMeasure(length),
ParentCurve=parent_curve, ParentCurve=parent_curve,
) )
@@ -235,9 +236,10 @@ class IfcAlignmentHelper:
case "CIRCULARARC": case "CIRCULARARC":
parent_curve = self._file.createIfcCircle( parent_curve = self._file.createIfcCircle(
Position=self._file.createIfcAxis2Placement2D( Position=self._file.createIfcAxis2Placement2D(
Location=self._file.createIfcCartesianPoint(Coordinates=(0., 0.)), Location=self._file.createIfcCartesianPoint(Coordinates=(0.0, 0.0)),
RefDirection=self._file.createIfcDirection( RefDirection=self._file.createIfcDirection(
(math.cos(start_direction), math.sin(start_direction))), (math.cos(start_direction), math.sin(start_direction))
),
), ),
Radius=abs(start_radius), Radius=abs(start_radius),
) )
@@ -249,8 +251,10 @@ class IfcAlignmentHelper:
type="IfcAxis2Placement2D", type="IfcAxis2Placement2D",
Location=start_point, Location=start_point,
RefDirection=self._file.createIfcDirection( RefDirection=self._file.createIfcDirection(
(math.cos(start_direction), math.sin(start_direction)))), (math.cos(start_direction), math.sin(start_direction))
SegmentStart=self._file.createIfcLengthMeasure(0.), ),
),
SegmentStart=self._file.createIfcLengthMeasure(0.0),
SegmentLength=self._file.createIfcLengthMeasure(length * start_radius / abs(start_radius)), SegmentLength=self._file.createIfcLengthMeasure(length * start_radius / abs(start_radius)),
ParentCurve=parent_curve, ParentCurve=parent_curve,
) )
@@ -262,8 +266,12 @@ class IfcAlignmentHelper:
return result return result
def _create_horizontal_alignment( def _create_horizontal_alignment(
self, name: str, description: str, points: Sequence[Sequence[float]], radii: Sequence[float], self,
include_geometry: bool = True name: str,
description: str,
points: Sequence[Sequence[float]],
radii: Sequence[float],
include_geometry: bool = True,
): ):
""" """
Create a horizontal alignment using the PI layout method. Create a horizontal alignment using the PI layout method.
@@ -343,9 +351,7 @@ class IfcAlignmentHelper:
horizontal_segments.append(alignment_segment) horizontal_segments.append(alignment_segment)
if include_geometry: if include_geometry:
horizontal_curve_segments.append( horizontal_curve_segments.append(self._map_alignment_horizontal_segment(design_parameters)[0])
self._map_alignment_horizontal_segment(design_parameters)[0]
)
# create circular curve # create circular curve
pc = self._file.create_entity( pc = self._file.create_entity(
@@ -378,9 +384,7 @@ class IfcAlignmentHelper:
horizontal_segments.append(alignment_segment) horizontal_segments.append(alignment_segment)
if include_geometry: if include_geometry:
horizontal_curve_segments.append( horizontal_curve_segments.append(self._map_alignment_horizontal_segment(design_parameters)[0])
self._map_alignment_horizontal_segment(design_parameters)[0]
)
xBT = xPT xBT = xPT
yBT = yPT yBT = yPT
@@ -393,10 +397,7 @@ class IfcAlignmentHelper:
dy = yPI - yBT dy = yPI - yBT
angleBT = math.atan2(dy, dx) angleBT = math.atan2(dy, dx)
tangent_run = math.sqrt(dx * dx + dy * dy) tangent_run = math.sqrt(dx * dx + dy * dy)
pt = self._file.create_entity( pt = self._file.create_entity(type="IfcCartesianPoint", Coordinates=(xBT, yBT))
type="IfcCartesianPoint",
Coordinates=(xBT, yBT)
)
design_parameters = self._file.create_entity( design_parameters = self._file.create_entity(
type="IfcAlignmentHorizontalSegment", type="IfcAlignmentHorizontalSegment",
@@ -423,15 +424,10 @@ class IfcAlignmentHelper:
) )
horizontal_segments.append(alignment_segment) horizontal_segments.append(alignment_segment)
if include_geometry: if include_geometry:
horizontal_curve_segments.append( horizontal_curve_segments.append(self._map_alignment_horizontal_segment(design_parameters)[0])
self._map_alignment_horizontal_segment(design_parameters)[0]
)
# create zero length terminator segment # create zero length terminator segment
poe = self._file.create_entity( poe = self._file.create_entity(type="IfcCartesianPoint", Coordinates=(xPI, yPI))
type="IfcCartesianPoint",
Coordinates=(xPI, yPI)
)
design_parameters = self._file.create_entity( design_parameters = self._file.create_entity(
type="IfcAlignmentHorizontalSegment", type="IfcAlignmentHorizontalSegment",
@@ -458,9 +454,7 @@ class IfcAlignmentHelper:
) )
horizontal_segments.append(alignment_segment) horizontal_segments.append(alignment_segment)
if include_geometry: if include_geometry:
horizontal_curve_segments.append( horizontal_curve_segments.append(self._map_alignment_horizontal_segment(design_parameters)[0])
self._map_alignment_horizontal_segment(design_parameters)[0]
)
if include_geometry: if include_geometry:
composite_curve = self._file.create_entity( composite_curve = self._file.create_entity(
@@ -474,11 +468,20 @@ class IfcAlignmentHelper:
return horizontal_segments, horizontal_curve_segments, composite_curve return horizontal_segments, horizontal_curve_segments, composite_curve
def _add_horizontal_alignment( def _add_horizontal_alignment(
self, alignment_name: str, points: Sequence[Sequence[float]], radii: Sequence[float], self,
include_geometry: bool = True, alignment_description: str = None, start_station : float = 1000., alignment_name: str,
points: Sequence[Sequence[float]],
radii: Sequence[float],
include_geometry: bool = True,
alignment_description: str = None,
start_station: float = 1000.0,
): ):
horizontal_segments, horizontal_curve_segments, composite_curve = self._create_horizontal_alignment( horizontal_segments, horizontal_curve_segments, composite_curve = self._create_horizontal_alignment(
alignment_name, alignment_description, points, radii, include_geometry, alignment_name,
alignment_description,
points,
radii,
include_geometry,
) )
name_segments(prefix="H", segments=horizontal_segments) name_segments(prefix="H", segments=horizontal_segments)
@@ -507,10 +510,8 @@ class IfcAlignmentHelper:
placement = self._file.createIfcLocalPlacement( placement = self._file.createIfcLocalPlacement(
PlacementRelTo=None, PlacementRelTo=None,
RelativePlacement=self._file.createIfcAxis2Placement2D( RelativePlacement=self._file.createIfcAxis2Placement2D(
Location=self._file.createIfcCartesianPoint( Location=self._file.createIfcCartesianPoint(Coordinates=(0.0, 0.0))
Coordinates=(0., 0.) ),
)
)
) )
# create the alignment # create the alignment
@@ -534,7 +535,7 @@ class IfcAlignmentHelper:
ContextOfItems=self._axis_geom_subcontext, ContextOfItems=self._axis_geom_subcontext,
RepresentationIdentifier="FootPrint", RepresentationIdentifier="FootPrint",
RepresentationType="Curve2D", RepresentationType="Curve2D",
Items=(composite_curve,) Items=(composite_curve,),
) )
# create the alignment product definition # create the alignment product definition
@@ -542,18 +543,15 @@ class IfcAlignmentHelper:
type="IfcProductDefinitionShape", type="IfcProductDefinitionShape",
Name="Alignment Product Definition Shape", Name="Alignment Product Definition Shape",
Description=None, Description=None,
Representations=(footprint_shape_representation,) Representations=(footprint_shape_representation,),
) )
# create representations for each segment # create representations for each segment
self._create_segment_representations( self._create_segment_representations(placement, horizontal_curve_segments, horizontal_segments)
placement, horizontal_curve_segments, horizontal_segments)
# add the representation to the alignment # add the representation to the alignment
alignment.Representation = product_definition_shape alignment.Representation = product_definition_shape
# create referent for start station # create referent for start station
start_referent = self._file.createIfcReferent( start_referent = self._file.createIfcReferent(
GlobalId=ifcopenshell.guid.new(), GlobalId=ifcopenshell.guid.new(),
@@ -564,7 +562,7 @@ class IfcAlignmentHelper:
ObjectPlacement=self._file.createIfcLinearPlacement( ObjectPlacement=self._file.createIfcLinearPlacement(
RelativePlacement=self._file.createIfcAxis2PlacementLinear( RelativePlacement=self._file.createIfcAxis2PlacementLinear(
Location=self._file.createIfcPointByDistanceExpression( Location=self._file.createIfcPointByDistanceExpression(
DistanceAlong=self._file.createIfcLengthMeasure(0.), DistanceAlong=self._file.createIfcLengthMeasure(0.0),
OffsetLateral=None, OffsetLateral=None,
OffsetVertical=None, OffsetVertical=None,
OffsetLongitudinal=None, OffsetLongitudinal=None,
@@ -584,7 +582,7 @@ class IfcAlignmentHelper:
OwnerHistory=None, OwnerHistory=None,
Name="Nests horizontal alignment and referents under overall alignment", Name="Nests horizontal alignment and referents under overall alignment",
RelatingObject=alignment, RelatingObject=alignment,
RelatedObjects=(horizontal_alignment, start_referent) RelatedObjects=(horizontal_alignment, start_referent),
) )
# aggregate the horizontal under the project # aggregate the horizontal under the project
@@ -594,14 +592,18 @@ class IfcAlignmentHelper:
OwnerHistory=None, OwnerHistory=None,
Name="Aggregates alignment under the project", Name="Aggregates alignment under the project",
RelatingObject=project, RelatingObject=project,
RelatedObjects=(alignment, ) RelatedObjects=(alignment,),
) )
return alignment return alignment
def add_vertical_alignment( def add_vertical_alignment(
self, name: str, description: str, vpoints: Sequence[Sequence[float]], vclengths: Sequence[Sequence[float]], self,
include_geometry: bool = True name: str,
description: str,
vpoints: Sequence[Sequence[float]],
vclengths: Sequence[Sequence[float]],
include_geometry: bool = True,
): ):
""" """
Create a vertical alignment using the PI layout method. Create a vertical alignment using the PI layout method.
@@ -614,20 +616,26 @@ class IfcAlignmentHelper:
""" """
pass pass
def add_alignment(
def add_alignment(self, self,
name: str, hpoints: Sequence[Sequence[float]], name: str,
radii: Sequence[float], hpoints: Sequence[Sequence[float]],
include_geometry: bool = True, radii: Sequence[float],
description: str = None, include_geometry: bool = True,
start_station: float = 1000., description: str = None,
): start_station: float = 1000.0,
):
""" """
Create a new alignment with a horizontal alignment using the PI layout method Create a new alignment with a horizontal alignment using the PI layout method
""" """
self._add_horizontal_alignment(alignment_name=name, points=hpoints, radii=radii, self._add_horizontal_alignment(
include_geometry=include_geometry, alignment_description=description, alignment_name=name,
start_station=start_station) points=hpoints,
radii=radii,
include_geometry=include_geometry,
alignment_description=description,
start_station=start_station,
)
def save_file(self, filename) -> None: def save_file(self, filename) -> None:
self._file.write(filename) self._file.write(filename)