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

Fix bug with missing system arrows in visualising offset models and sink/source mixed flows.

Browse Source
This commit is contained in:
Dion Moult
2021-12-09 13:40:23 +11:00
parent a23c88a53e
commit 763a38bc9c
1 changed files with 39 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/blenderbim/blenderbim/bim/module/misc/operator.py
+39 -4
View File
@@ -17,6 +17,7 @@
# along with BlenderBIM Add-on. If not, see <http://www.gnu.org/licenses/>. # along with BlenderBIM Add-on. If not, see <http://www.gnu.org/licenses/>.
import bpy import bpy
import numpy as np
import ifcopenshell import ifcopenshell
import blenderbim.bim.handler import blenderbim.bim.handler
import blenderbim.tool as tool import blenderbim.tool as tool
@@ -211,17 +212,51 @@ class DrawSystemArrows(bpy.types.Operator, Operator):
for obj in bpy.context.selected_objects: for obj in bpy.context.selected_objects:
if not obj.BIMObjectProperties.ifc_definition_id: if not obj.BIMObjectProperties.ifc_definition_id:
continue continue
e = tool.Ifc.get().by_id(obj.BIMObjectProperties.ifc_definition_id) element = tool.Ifc.get_entity(obj)
sources = [] sources = []
sinks = [] sinks = []
for rel in getattr(e, "HasPorts", []) or []: for rel in getattr(element, "HasPorts", []) or []:
if rel.RelatingPort.FlowDirection == "SOURCE": if rel.RelatingPort.FlowDirection == "SOURCE":
sources.append(ifcopenshell.util.placement.get_local_placement(rel.RelatingPort.ObjectPlacement)) sources.append(
self.get_absolute_matrix(
ifcopenshell.util.placement.get_local_placement(rel.RelatingPort.ObjectPlacement)
)
)
elif rel.RelatingPort.FlowDirection == "SINK": elif rel.RelatingPort.FlowDirection == "SINK":
sinks.append(ifcopenshell.util.placement.get_local_placement(rel.RelatingPort.ObjectPlacement)) sinks.append(
self.get_absolute_matrix(
ifcopenshell.util.placement.get_local_placement(rel.RelatingPort.ObjectPlacement)
)
)
else:
sources.append(
self.get_absolute_matrix(
ifcopenshell.util.placement.get_local_placement(rel.RelatingPort.ObjectPlacement)
)
)
sinks.append(
self.get_absolute_matrix(
ifcopenshell.util.placement.get_local_placement(rel.RelatingPort.ObjectPlacement)
)
)
for sink in sinks: for sink in sinks:
for source in sources: for source in sources:
polyline = curve.data.splines.new("POLY") polyline = curve.data.splines.new("POLY")
polyline.points.add(1) polyline.points.add(1)
polyline.points[0].co = (Matrix(sink).translation * unit_scale).to_4d() polyline.points[0].co = (Matrix(sink).translation * unit_scale).to_4d()
polyline.points[1].co = (Matrix(source).translation * unit_scale).to_4d() polyline.points[1].co = (Matrix(source).translation * unit_scale).to_4d()
def get_absolute_matrix(self, matrix):
props = bpy.context.scene.BIMGeoreferenceProperties
if props.has_blender_offset:
matrix = np.array(
ifcopenshell.util.geolocation.global2local(
matrix,
float(props.blender_eastings),
float(props.blender_northings),
float(props.blender_orthogonal_height),
float(props.blender_x_axis_abscissa),
float(props.blender_x_axis_ordinate),
)
)
return matrix