This commit is contained in:
Andrej730
2025-07-28 11:45:12 +05:00
parent 23944d6060
commit cdeeab4b31
20 changed files with 161 additions and 134 deletions
@@ -47,7 +47,11 @@ def _add_curve_segment_to_composite_curve(
composite_curve.Segments += (curve_segment,)
assert len(curve_segment.UsingCurves) == 1
else:
zero_length_segment = composite_curve.Segments[-1] if ifcopenshell.api.alignment.has_zero_length_segment(composite_curve) else None
zero_length_segment = (
composite_curve.Segments[-1]
if ifcopenshell.api.alignment.has_zero_length_segment(composite_curve)
else None
)
prev_segment = None
if zero_length_segment and 1 < len(composite_curve.Segments):
@@ -59,7 +63,10 @@ def _add_curve_segment_to_composite_curve(
segments = composite_curve.Segments[0:-1]
if zero_length_segment:
segments += (curve_segment,zero_length_segment,)
segments += (
curve_segment,
zero_length_segment,
)
composite_curve.Segments = []
composite_curve.Segments += segments
else:
@@ -163,12 +163,11 @@ def _add_segment_to_layout(file: ifcopenshell.file, layout: entity_instance, seg
end_referent.ObjectPlacement.CartesianPosition.Axis.DirectionRatios = (ax, ay, az)
end_referent.ObjectPlacement.CartesianPosition.RefDirection.DirectionRatios = (rx, ry, rz)
start_station = ifcopenshell.api.alignment.get_alignment_station(file,alignment)
start_station = ifcopenshell.api.alignment.get_alignment_station(file, alignment)
end_referent_station = start_station + start_dist_along
pset_stationing = ifcopenshell.api.pset.add_pset(file, product=end_referent, name="Pset_Stationing")
ifcopenshell.api.pset.edit_pset(file, pset=pset_stationing, properties={"Station": end_referent_station})
# create the start of segment referent
# get the previous segment. Working from the end of the basis curve, -1 is zero length segment
@@ -22,6 +22,7 @@ from ifcopenshell import entity_instance
from ifcopenshell.api.alignment._get_segment_start_point_label import _get_segment_start_point_label
def _add_zero_length_segment(file: ifcopenshell.file, layout: entity_instance) -> None:
"""
Adds a zero length segment to the end of a layout. Also adds a zero length segment to the end of the corresponding geometric curve.
@@ -32,23 +33,23 @@ def _add_zero_length_segment(file: ifcopenshell.file, layout: entity_instance) -
:param layout: An IfcAlignmentHorizontal, IfcAlignmentVertical, or IfcAlignmentCant
:return: None
"""
expected_types = ["IfcAlignmentHorizontal", "IfcAlignmentVertical", "IfcAlignmentCant"]
if not layout.is_a() in expected_types:
raise TypeError(
f"Expected layout type to be one of {[_ for _ in expected_types]}, instead received {layout.is_a()}"
)
if (not ifcopenshell.api.alignment.add_zero_length_segment(file,layout,include_referent=False)):
return # zero length segment not added, probably because it already exists
if not ifcopenshell.api.alignment.add_zero_length_segment(file, layout, include_referent=False):
return # zero length segment not added, probably because it already exists
curve = ifcopenshell.api.alignment.get_layout_curve(layout)
if curve:
ifcopenshell.api.alignment.add_zero_length_segment(file,curve)
ifcopenshell.api.alignment.add_zero_length_segment(file, curve)
segment = layout.IsNestedBy[0].RelatedObjects[-1]
alignment = ifcopenshell.api.alignment.get_alignment(layout)
station = ifcopenshell.api.alignment.get_alignment_station(file,alignment)
station = ifcopenshell.api.alignment.get_alignment_station(file, alignment)
name = f"{_get_segment_start_point_label(segment,None)} ({ifcopenshell.util.alignment.station_as_string(file,station)})"
ifcopenshell.api.alignment.add_stationing_referent(file, segment, 0.0, station, name=name)
@@ -37,7 +37,7 @@ def _map_line(file: ifcopenshell.file, design_parameters: entity_instance) -> Se
start_direction = design_parameters.StartDirection
length = design_parameters.SegmentLength
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file,'PLANEANGLEUNIT')
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file, "PLANEANGLEUNIT")
start_direction *= angle_unit_scale
transition = "DISCONTINUOUS"
@@ -80,7 +80,7 @@ def _map_circular_arc(file: ifcopenshell.file, design_parameters: entity_instanc
start_radius = design_parameters.StartRadiusOfCurvature
length = design_parameters.SegmentLength
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file,'PLANEANGLEUNIT')
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file, "PLANEANGLEUNIT")
start_direction *= angle_unit_scale
transition = "DISCONTINUOUS"
@@ -114,7 +114,7 @@ def _map_clothoid(file: ifcopenshell.file, design_parameters: entity_instance) -
end_radius = design_parameters.EndRadiusOfCurvature
length = design_parameters.SegmentLength
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file,'PLANEANGLEUNIT')
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file, "PLANEANGLEUNIT")
start_direction *= angle_unit_scale
transition = "DISCONTINUOUS"
@@ -156,7 +156,7 @@ def _map_cubic(file: ifcopenshell.file, design_parameters: entity_instance) -> S
end_radius = design_parameters.EndRadiusOfCurvature
length = design_parameters.SegmentLength
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file,'PLANEANGLEUNIT')
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file, "PLANEANGLEUNIT")
start_direction *= angle_unit_scale
transition = "DISCONTINUOUS"
@@ -209,7 +209,7 @@ def _map_helmert_curve(file: ifcopenshell.file, design_parameters: entity_instan
end_radius = design_parameters.EndRadiusOfCurvature
length = design_parameters.SegmentLength
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file,'PLANEANGLEUNIT')
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file, "PLANEANGLEUNIT")
start_direction *= angle_unit_scale
transition = "DISCONTINUOUS"
@@ -315,7 +315,7 @@ def _map_bloss_curve(file: ifcopenshell.file, design_parameters: entity_instance
start_radius = design_parameters.StartRadiusOfCurvature
length = design_parameters.SegmentLength
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file,'PLANEANGLEUNIT')
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file, "PLANEANGLEUNIT")
start_direction *= angle_unit_scale
transition = "DISCONTINUOUS"
@@ -362,7 +362,7 @@ def _map_cosine_curve(file: ifcopenshell.file, design_parameters: entity_instanc
start_radius = design_parameters.StartRadiusOfCurvature
length = design_parameters.SegmentLength
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file,'PLANEANGLEUNIT')
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file, "PLANEANGLEUNIT")
start_direction *= angle_unit_scale
transition = "DISCONTINUOUS"
@@ -405,7 +405,7 @@ def _map_sine_curve(file: ifcopenshell.file, design_parameters: entity_instance)
start_radius = design_parameters.StartRadiusOfCurvature
length = design_parameters.SegmentLength
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file,'PLANEANGLEUNIT')
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file, "PLANEANGLEUNIT")
start_direction *= angle_unit_scale
transition = "DISCONTINUOUS"
@@ -136,7 +136,7 @@ def _map_circular_arc(file: ifcopenshell.file, design_parameters: entity_instanc
start_height = design_parameters.StartHeight
start_gradient = design_parameters.StartGradient
end_gradient = design_parameters.EndGradient
#radius = design_parameters.RadiusOfCurvature
# radius = design_parameters.RadiusOfCurvature
transition = "DISCONTINUOUS"
start_angle = math.atan(start_gradient)
@@ -30,7 +30,7 @@ from ifcopenshell.api.alignment._map_alignment_horizontal_segment import _map_al
from ifcopenshell.api.alignment._update_curve_segment_transition_code import _update_curve_segment_transition_code
def add_zero_length_segment(file: ifcopenshell.file, layout: entity_instance, include_referent : bool = True) -> bool:
def add_zero_length_segment(file: ifcopenshell.file, layout: entity_instance, include_referent: bool = True) -> bool:
"""
Adds a zero length segment to the end of a layout.
@@ -40,27 +40,33 @@ def add_zero_length_segment(file: ifcopenshell.file, layout: entity_instance, in
:param include_referent: If True, an IfcReferent representing the ending point of the layout is included for IfcLinearElement layouts (i.e. business logic)
:return: True if segment is added
"""
# These are valid curve types for alignment, but don't have the zero-length segment
if layout.is_a("IfcOffsetCurveByDistances") or layout.is_a("IfcPolyline") or layout.is_a("IfcIndexedPolyCurve"):
return
expected_types = ["IfcAlignmentHorizontal", "IfcAlignmentVertical", "IfcAlignmentCant","IfcCompositeCurve","IfcGradientCurve","IfcSegmentedReferenceCurve"]
expected_types = [
"IfcAlignmentHorizontal",
"IfcAlignmentVertical",
"IfcAlignmentCant",
"IfcCompositeCurve",
"IfcGradientCurve",
"IfcSegmentedReferenceCurve",
]
if not layout.is_a() in expected_types:
raise TypeError(
f"Expected layout type to be one of {[_ for _ in expected_types]}, instead received {layout.is_a()}"
)
if ifcopenshell.api.alignment.has_zero_length_segment(layout):
return False
if layout.is_a("IfcCompositeCurve") or layout.is_a("IfcGradientCurve") or layout.is_a("IfcSegmentedReferenceCurve"):
x = 0.
y = 0.
dx = 1.
dy = 0.
segment_start = 0.
x = 0.0
y = 0.0
dx = 1.0
dy = 0.0
segment_start = 0.0
last_segment = None
if layout.Segments and 0 < len(layout.Segments):
@@ -68,15 +74,15 @@ def add_zero_length_segment(file: ifcopenshell.file, layout: entity_instance, in
# because this becomes of placement of the zero length segment
last_segment = layout.Segments[-1]
settings = ifcopenshell.geom.settings()
fn = wrapper.map_shape(settings,last_segment.wrapped_data)
eval = wrapper.function_item_evaluator(settings,fn)
fn = wrapper.map_shape(settings, last_segment.wrapped_data)
eval = wrapper.function_item_evaluator(settings, fn)
e = np.array(eval.evaluate(fn.end()))
unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file)
e[:3,3] /= unit_scale
x = float(e[0,3])
y = float(e[1,3])
dx = float(e[0,0])
dy = float(e[1,0])
e[:3, 3] /= unit_scale
x = float(e[0, 3])
y = float(e[1, 3])
dx = float(e[0, 0])
dy = float(e[1, 0])
parent_curve = file.createIfcLine(
Pnt=file.createIfcCartesianPoint(Coordinates=((0.0, 0.0))),
@@ -99,21 +105,21 @@ def add_zero_length_segment(file: ifcopenshell.file, layout: entity_instance, in
layout.Segments += (zero_length_curve_segment,)
if last_segment:
_update_curve_segment_transition_code(last_segment,zero_length_curve_segment)
_update_curve_segment_transition_code(last_segment, zero_length_curve_segment)
# add zero length segments to base curves
if layout.is_a("IfcSegmentedReferenceCurve"):
ifcopenshell.api.alignment.add_zero_length_segment(file,layout.BaseCurve)
ifcopenshell.api.alignment.add_zero_length_segment(file, layout.BaseCurve)
elif layout.is_a("IfcGradientCurve"):
ifcopenshell.api.alignment.add_zero_length_segment(file,layout.BaseCurve)
ifcopenshell.api.alignment.add_zero_length_segment(file, layout.BaseCurve)
else:
zero_length_curve_segment = None
if layout.is_a("IfcAlignmentHorizontal"):
x = 0.
y = 0.
dx = 1.
dy = 0.
x = 0.0
y = 0.0
dx = 1.0
dy = 0.0
last_segment = None
for rel in layout.IsNestedBy:
if 0 < len(rel.RelatedObjects):
@@ -121,35 +127,34 @@ def add_zero_length_segment(file: ifcopenshell.file, layout: entity_instance, in
break
if last_segment:
file.begin_transaction() # use a transaction so we can discard any temporary IFC entities created
file.begin_transaction() # use a transaction so we can discard any temporary IFC entities created
settings = ifcopenshell.geom.settings()
mapped_segments = _map_alignment_horizontal_segment(file,last_segment)
mapped_segments = _map_alignment_horizontal_segment(file, last_segment)
geometry_segment = mapped_segments[0] if mapped_segments[1] == None else mapped_segments[1]
fn = wrapper.map_shape(settings,geometry_segment.wrapped_data)
eval = wrapper.function_item_evaluator(settings,fn)
fn = wrapper.map_shape(settings, geometry_segment.wrapped_data)
eval = wrapper.function_item_evaluator(settings, fn)
e = np.array(eval.evaluate(fn.end()))
unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file)
x = float(e[0,3]) / unit_scale
y = float(e[1,3]) / unit_scale
dx = float(e[0,0])
dy = float(e[1,0])
x = float(e[0, 3]) / unit_scale
y = float(e[1, 3]) / unit_scale
dx = float(e[0, 0])
dy = float(e[1, 0])
file.discard_transaction()
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file,'PLANEANGLEUNIT')
angle_unit_scale = ifcopenshell.util.unit.calculate_unit_scale(file, "PLANEANGLEUNIT")
design_parameters = file.createIfcAlignmentHorizontalSegment(
StartPoint=file.createIfcCartesianPoint(
(x,y)
),
StartDirection=math.atan2(dy,dx) / angle_unit_scale,
StartPoint=file.createIfcCartesianPoint((x, y)),
StartDirection=math.atan2(dy, dx) / angle_unit_scale,
StartRadiusOfCurvature=0.0,
EndRadiusOfCurvature=0.0,
SegmentLength=0.0,
PredefinedType="LINE",
)
zero_length_curve_segment = file.createIfcAlignmentSegment(GlobalId=ifcopenshell.guid.new(), DesignParameters=design_parameters)
zero_length_curve_segment = file.createIfcAlignmentSegment(
GlobalId=ifcopenshell.guid.new(), DesignParameters=design_parameters
)
elif layout.is_a("IfcAlignmentVertical"):
last_segment_dist_along = 0.0
last_segment_end_gradient = 0.0
@@ -170,7 +175,9 @@ def add_zero_length_segment(file: ifcopenshell.file, layout: entity_instance, in
EndGradient=last_segment_end_gradient,
PredefinedType="CONSTANTGRADIENT",
)
zero_length_curve_segment = file.createIfcAlignmentSegment(GlobalId=ifcopenshell.guid.new(), DesignParameters=design_parameters)
zero_length_curve_segment = file.createIfcAlignmentSegment(
GlobalId=ifcopenshell.guid.new(), DesignParameters=design_parameters
)
elif layout.is_a("IfcAlignmentCant"):
last_segment_dist_along = 0.0
last_segment_cant_left = 0.0
@@ -200,14 +207,16 @@ def add_zero_length_segment(file: ifcopenshell.file, layout: entity_instance, in
StartCantRight=last_segment_cant_right,
PredefinedType="CONSTANTCANT",
)
zero_length_curve_segment = file.createIfcAlignmentSegment(GlobalId=ifcopenshell.guid.new(), DesignParameters=design_parameters)
zero_length_curve_segment = file.createIfcAlignmentSegment(
GlobalId=ifcopenshell.guid.new(), DesignParameters=design_parameters
)
ifcopenshell.api.nest.assign_object(file, related_objects=[zero_length_curve_segment], relating_object=layout)
if include_referent:
alignment = ifcopenshell.api.alignment.get_alignment(layout)
station = ifcopenshell.api.alignment.get_alignment_station(file,alignment)
station = ifcopenshell.api.alignment.get_alignment_station(file, alignment)
name = f"{_get_segment_start_point_label(zero_length_curve_segment,None)} ({ifcopenshell.util.alignment.station_as_string(file,station)})"
ifcopenshell.api.alignment.add_stationing_referent(file, zero_length_curve_segment, 0.0, station, name=name)
return True
return True
@@ -54,9 +54,9 @@ def create_as_offset_curve(
_create_offset_curve_representation(file, alignment, offsets)
# define stationing
#name = ifcopenshell.util.alignment.station_as_string(file, start_station)
#referent = ifcopenshell.api.alignment.add_stationing_referent(file, alignment, 0.0, start_station, name)
#ifcopenshell.api.nest.reorder_nesting(file, referent, -1, 0)
# name = ifcopenshell.util.alignment.station_as_string(file, start_station)
# referent = ifcopenshell.api.alignment.add_stationing_referent(file, alignment, 0.0, start_station, name)
# ifcopenshell.api.nest.reorder_nesting(file, referent, -1, 0)
# IFC 4.1.4.1.1 Alignment Aggregation To Project
project = file.by_type("IfcProject")[0]
@@ -23,6 +23,7 @@ from ifcopenshell import entity_instance
from ifcopenshell.api.alignment._create_geometric_representation import _create_geometric_representation
from ifcopenshell.api.alignment._add_segment_to_curve import _add_segment_to_curve
def create_representation(
file: ifcopenshell.file,
alignment: entity_instance,
@@ -43,11 +44,11 @@ def create_representation(
if alignment.Representation:
return
_create_geometric_representation(file,alignment)
_create_geometric_representation(file, alignment)
layouts = ifcopenshell.api.alignment.get_alignment_layouts(alignment)
for layout in layouts:
curve = ifcopenshell.api.alignment.get_layout_curve(layout)
curve = ifcopenshell.api.alignment.get_layout_curve(layout)
for segment in layout.IsNestedBy[0].RelatedObjects:
_add_segment_to_curve(file, segment, curve)
_add_segment_to_curve(file, segment, curve)
@@ -36,7 +36,7 @@ def get_alignment_station(file: ifcopenshell.file, alignment: entity_instance) -
parent_alignment = ifcopenshell.api.alignment.get_parent_alignment(alignment)
if parent_alignment:
start_station = ifcopenshell.api.alignment.get_alignment_station(file,parent_alignment)
start_station = ifcopenshell.api.alignment.get_alignment_station(file, parent_alignment)
else:
components = ifcopenshell.util.element.get_components(alignment)
for c in components:
@@ -44,5 +44,4 @@ def get_alignment_station(file: ifcopenshell.file, alignment: entity_instance) -
start_station = ifcopenshell.util.element.get_pset(c, name="Pset_Stationing", prop="Station")
break
return start_station
@@ -29,7 +29,14 @@ def has_zero_length_segment(layout: entity_instance) -> bool:
:param layout: An IfcAlignmentHorizontal, IfcAlignmentVertical, IfcAlignmentCant, IfcCompositeCurve, IfcGradientCurve, or IfcSegmentedReferenceCurve
:return: True if the zero length segment is present
"""
expected_types = ["IfcAlignmentHorizontal", "IfcAlignmentVertical", "IfcAlignmentCant","IfcCompositeCurve","IfcGradientCurve","IfcSegmentedReferenceCurve"]
expected_types = [
"IfcAlignmentHorizontal",
"IfcAlignmentVertical",
"IfcAlignmentCant",
"IfcCompositeCurve",
"IfcGradientCurve",
"IfcSegmentedReferenceCurve",
]
if not layout.is_a() in expected_types:
raise TypeError(
f"Expected entity type to be one of {[_ for _ in expected_types]}, instead received '{layout.is_a()}"
@@ -38,7 +45,11 @@ def has_zero_length_segment(layout: entity_instance) -> bool:
result = False
if layout.is_a("IfcCompositeCurve") or layout.is_a("IfcGradientCurve") or layout.is_a("IfcSegmentedReferenceCurve"):
result = True if layout.Segments and 0 < len(layout.Segments) and layout.Segments[-1].SegmentLength.wrappedValue == 0.0 else False
result = (
True
if layout.Segments and 0 < len(layout.Segments) and layout.Segments[-1].SegmentLength.wrappedValue == 0.0
else False
)
else:
for rel in layout.IsNestedBy:
if 0 < len(rel.RelatedObjects):
@@ -33,9 +33,7 @@ def update_fallback_position(file: ifcopenshell.file, lp: entity_instance):
"""
if not lp.CartesianPosition:
lp.CartesianPosition = file.createIfcAxis2Placement3D(
Location = file.createIfcCartesianPoint((0.,0.))
)
lp.CartesianPosition = file.createIfcAxis2Placement3D(Location=file.createIfcCartesianPoint((0.0, 0.0)))
p = np.array(ifcopenshell.util.placement.get_axis2placement(lp.RelativePlacement))
@@ -51,13 +49,13 @@ def update_fallback_position(file: ifcopenshell.file, lp: entity_instance):
ay = float(p[1, 2])
az = float(p[2, 2])
lp.CartesianPosition.Location.Coordinates = ((x,y,z))
lp.CartesianPosition.Location.Coordinates = (x, y, z)
if not lp.CartesianPosition.RefDirection:
lp.CartesianPosition.RefDirection = file.createIfcDirection((1.,0.,0.))
lp.CartesianPosition.RefDirection = file.createIfcDirection((1.0, 0.0, 0.0))
if not lp.CartesianPosition.Axis:
lp.CartesianPosition.Axis = file.createIfcDirection((0.,0.,1.))
lp.CartesianPosition.Axis = file.createIfcDirection((0.0, 0.0, 1.0))
lp.CartesianPosition.RefDirection.DirectionRatios = ((rx,ry,rz))
lp.CartesianPosition.Axis.DirectionRatios = ((ax,ay,az))
lp.CartesianPosition.RefDirection.DirectionRatios = (rx, ry, rz)
lp.CartesianPosition.Axis.DirectionRatios = (ax, ay, az)