d8889c3ca4
Fixes #21364. `self.Type` should be set in `__init__` and `loads`, and not in `onDocumentRestored`. Additionally: fixed mistake in `loads` in ifc_objects.py.
463 lines
20 KiB
Python
463 lines
20 KiB
Python
# SPDX-License-Identifier: LGPL-2.1-or-later
|
|
|
|
# ***************************************************************************
|
|
# * *
|
|
# * Copyright (c) 2016 Yorik van Havre <yorik@uncreated.net> *
|
|
# * *
|
|
# * This file is part of FreeCAD. *
|
|
# * *
|
|
# * FreeCAD 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 2.1 of the *
|
|
# * License, or (at your option) any later version. *
|
|
# * *
|
|
# * FreeCAD 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 FreeCAD. If not, see *
|
|
# * <https://www.gnu.org/licenses/>. *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
__title__ = "Arch Pipe tools"
|
|
__author__ = "Yorik van Havre"
|
|
__url__ = "https://www.freecad.org"
|
|
|
|
## @package ArchPipe
|
|
# \ingroup ARCH
|
|
# \brief The Pipe object and tools
|
|
#
|
|
# This module provides tools to build Pipe and Pipe connector objects.
|
|
# Pipes are tubular objects extruded along a base line.
|
|
|
|
import FreeCAD
|
|
import ArchComponent
|
|
import ArchIFC
|
|
|
|
from draftutils import params
|
|
|
|
if FreeCAD.GuiUp:
|
|
from PySide.QtCore import QT_TRANSLATE_NOOP
|
|
import FreeCADGui
|
|
import Arch_rc
|
|
from draftutils.translate import translate
|
|
else:
|
|
# \cond
|
|
def translate(ctxt,txt):
|
|
return txt
|
|
def QT_TRANSLATE_NOOP(ctxt,txt):
|
|
return txt
|
|
# \endcond
|
|
|
|
|
|
class _ArchPipe(ArchComponent.Component):
|
|
|
|
|
|
"the Arch Pipe object"
|
|
|
|
def __init__(self,obj):
|
|
|
|
ArchComponent.Component.__init__(self,obj)
|
|
self.Type = "Pipe"
|
|
self.setProperties(obj)
|
|
# IfcPipeSegment is new in IFC4
|
|
from ArchIFC import IfcTypes
|
|
if "Pipe Segment" in IfcTypes:
|
|
obj.IfcType = "Pipe Segment"
|
|
else:
|
|
# IFC2x3 does not know a Pipe Segment
|
|
obj.IfcType = "Building Element Proxy"
|
|
|
|
def setProperties(self,obj):
|
|
|
|
pl = obj.PropertiesList
|
|
if not "Diameter" in pl:
|
|
obj.addProperty("App::PropertyLength", "Diameter", "Pipe", QT_TRANSLATE_NOOP("App::Property","The diameter of this pipe, if not based on a profile"), locked=True)
|
|
if not "Width" in pl:
|
|
obj.addProperty("App::PropertyLength", "Width", "Pipe", QT_TRANSLATE_NOOP("App::Property","The width of this pipe, if not based on a profile"), locked=True)
|
|
obj.setPropertyStatus("Width", "Hidden")
|
|
if not "Height" in pl:
|
|
obj.addProperty("App::PropertyLength", "Height", "Pipe", QT_TRANSLATE_NOOP("App::Property","The height of this pipe, if not based on a profile"), locked=True)
|
|
obj.setPropertyStatus("Height", "Hidden")
|
|
if not "Length" in pl:
|
|
obj.addProperty("App::PropertyLength", "Length", "Pipe", QT_TRANSLATE_NOOP("App::Property","The length of this pipe, if not based on an edge"), locked=True)
|
|
if not "Profile" in pl:
|
|
obj.addProperty("App::PropertyLink", "Profile", "Pipe", QT_TRANSLATE_NOOP("App::Property","An optional closed profile to base this pipe on"), locked=True)
|
|
if not "OffsetStart" in pl:
|
|
obj.addProperty("App::PropertyLength", "OffsetStart", "Pipe", QT_TRANSLATE_NOOP("App::Property","Offset from the start point"), locked=True)
|
|
if not "OffsetEnd" in pl:
|
|
obj.addProperty("App::PropertyLength", "OffsetEnd", "Pipe", QT_TRANSLATE_NOOP("App::Property","Offset from the end point"), locked=True)
|
|
if not "WallThickness" in pl:
|
|
obj.addProperty("App::PropertyLength", "WallThickness","Pipe", QT_TRANSLATE_NOOP("App::Property","The wall thickness of this pipe, if not based on a profile"), locked=True)
|
|
if not "ProfileType" in pl:
|
|
obj.addProperty("App::PropertyEnumeration", "ProfileType", "Pipe", QT_TRANSLATE_NOOP("App::Property","If not based on a profile, this controls the profile of this pipe"), locked=True)
|
|
obj.ProfileType = ["Circle", "Square", "Rectangle"]
|
|
|
|
def onDocumentRestored(self,obj):
|
|
|
|
ArchComponent.Component.onDocumentRestored(self,obj)
|
|
self.setProperties(obj)
|
|
|
|
def loads(self,state):
|
|
|
|
self.Type = "Pipe"
|
|
|
|
def onChanged(self, obj, prop):
|
|
if prop == "IfcType":
|
|
root = ArchIFC.IfcProduct()
|
|
root.setupIfcAttributes(obj)
|
|
root.setupIfcComplexAttributes(obj)
|
|
elif prop == "ProfileType":
|
|
if obj.ProfileType == "Square":
|
|
obj.setPropertyStatus("Height", "Hidden")
|
|
obj.setPropertyStatus("Diameter", "Hidden")
|
|
obj.setPropertyStatus("Width", "-Hidden")
|
|
elif obj.ProfileType == "Rectangle":
|
|
obj.setPropertyStatus("Height", "-Hidden")
|
|
obj.setPropertyStatus("Diameter", "Hidden")
|
|
obj.setPropertyStatus("Width", "-Hidden")
|
|
else:
|
|
obj.setPropertyStatus("Height", "Hidden")
|
|
obj.setPropertyStatus("Diameter", "-Hidden")
|
|
obj.setPropertyStatus("Width", "Hidden")
|
|
|
|
def execute(self,obj):
|
|
|
|
import math
|
|
import Part
|
|
import DraftGeomUtils
|
|
if self.clone(obj):
|
|
return
|
|
pl = obj.Placement
|
|
w = self.getWire(obj)
|
|
if not w:
|
|
FreeCAD.Console.PrintError(translate("Arch","Unable to build the base path")+"\n")
|
|
return
|
|
if obj.OffsetStart.Value:
|
|
e = w.Edges[0]
|
|
v = e.Vertexes[-1].Point.sub(e.Vertexes[0].Point).normalize()
|
|
v.multiply(obj.OffsetStart.Value)
|
|
e = Part.LineSegment(e.Vertexes[0].Point.add(v),e.Vertexes[-1].Point).toShape()
|
|
w = Part.Wire([e]+w.Edges[1:])
|
|
if obj.OffsetEnd.Value:
|
|
e = w.Edges[-1]
|
|
v = e.Vertexes[0].Point.sub(e.Vertexes[-1].Point).normalize()
|
|
v.multiply(obj.OffsetEnd.Value)
|
|
e = Part.LineSegment(e.Vertexes[-1].Point.add(v),e.Vertexes[0].Point).toShape()
|
|
w = Part.Wire(w.Edges[:-1]+[e])
|
|
p = self.getProfile(obj)
|
|
if not p:
|
|
FreeCAD.Console.PrintError(translate("Arch","Unable to build the profile")+"\n")
|
|
return
|
|
# move and rotate the profile to the first point
|
|
if hasattr(p,"CenterOfMass"):
|
|
c = p.CenterOfMass
|
|
else:
|
|
c = p.BoundBox.Center
|
|
delta = w.Vertexes[0].Point-c
|
|
p.translate(delta)
|
|
import Draft
|
|
if Draft.getType(obj.Base) == "BezCurve":
|
|
v1 = obj.Base.Placement.multVec(obj.Base.Points[1])-w.Vertexes[0].Point
|
|
else:
|
|
v1 = w.Vertexes[1].Point-w.Vertexes[0].Point
|
|
v2 = DraftGeomUtils.getNormal(p)
|
|
#rot = FreeCAD.Rotation(v2,v1)
|
|
# rotate keeping up vector
|
|
if v1.getAngle(FreeCAD.Vector(0,0,1)) > 0.01:
|
|
up = FreeCAD.Vector(0,0,1)
|
|
else:
|
|
up = FreeCAD.Vector(0,1,0)
|
|
v1y = up.cross(v1)
|
|
v1x = v1.cross(v1y)
|
|
rot = FreeCAD.Rotation(v1x,v1y,v1,"ZYX")
|
|
p.rotate(w.Vertexes[0].Point,rot.Axis,math.degrees(rot.Angle))
|
|
shapes = []
|
|
try:
|
|
if p.Faces:
|
|
for f in p.Faces:
|
|
sh = w.makePipeShell([f.OuterWire],True,False,2)
|
|
for shw in f.Wires:
|
|
if shw.hashCode() != f.OuterWire.hashCode():
|
|
sh2 = w.makePipeShell([shw],True,False,2)
|
|
sh = sh.cut(sh2)
|
|
shapes.append(sh)
|
|
elif p.Wires:
|
|
for pw in p.Wires:
|
|
sh = w.makePipeShell([pw],True,False,2)
|
|
shapes.append(sh)
|
|
except Exception:
|
|
FreeCAD.Console.PrintError(translate("Arch","Unable to build the pipe")+"\n")
|
|
else:
|
|
if len(shapes) == 0:
|
|
return
|
|
elif len(shapes) == 1:
|
|
sh = shapes[0]
|
|
else:
|
|
sh = Part.makeCompound(shapes)
|
|
obj.Shape = self.processSubShapes(obj,sh,pl)
|
|
if obj.Base:
|
|
obj.Length = w.Length
|
|
else:
|
|
obj.Placement = pl
|
|
|
|
def getWire(self,obj):
|
|
|
|
import Part
|
|
if obj.Base:
|
|
if not hasattr(obj.Base,'Shape'):
|
|
FreeCAD.Console.PrintError(translate("Arch","The base object is not a Part")+"\n")
|
|
return
|
|
if len(obj.Base.Shape.Wires) != 1:
|
|
FreeCAD.Console.PrintError(translate("Arch","Too many wires in the base shape")+"\n")
|
|
return
|
|
if obj.Base.Shape.Wires[0].isClosed():
|
|
FreeCAD.Console.PrintError(translate("Arch","The base wire is closed")+"\n")
|
|
return
|
|
w = obj.Base.Shape.Wires[0]
|
|
else:
|
|
if obj.Length.Value == 0:
|
|
return
|
|
w = Part.Wire([Part.LineSegment(FreeCAD.Vector(0,0,0),FreeCAD.Vector(0,0,obj.Length.Value)).toShape()])
|
|
return w
|
|
|
|
def getProfile(self,obj):
|
|
|
|
import Part
|
|
if obj.Profile:
|
|
if not obj.Profile.getLinkedObject().isDerivedFrom("Part::Part2DObject"):
|
|
FreeCAD.Console.PrintError(translate("Arch","The profile is not a 2D Part")+"\n")
|
|
return
|
|
if not obj.Profile.Shape.Wires[0].isClosed():
|
|
FreeCAD.Console.PrintError(translate("Arch","The profile is not closed")+"\n")
|
|
return
|
|
p = obj.Profile.Shape.Wires[0]
|
|
else:
|
|
if obj.ProfileType == "Square":
|
|
if obj.Width.Value == 0:
|
|
return
|
|
p = Part.makePlane(obj.Width.Value, obj.Width.Value,FreeCAD.Vector(-obj.Width.Value/2,-obj.Width.Value/2,0))
|
|
if obj.WallThickness.Value and (obj.WallThickness.Value < obj.Width.Value/2):
|
|
p2 = Part.makePlane(obj.Width.Value-obj.WallThickness.Value*2,obj.Width.Value-obj.WallThickness.Value*2,FreeCAD.Vector(obj.WallThickness.Value-obj.Width.Value/2,obj.WallThickness.Value-obj.Width.Value/2,0))
|
|
p = p.cut(p2)
|
|
elif obj.ProfileType == "Rectangle":
|
|
if obj.Width.Value == 0:
|
|
return
|
|
if obj.Height.Value == 0:
|
|
return
|
|
p = Part.makePlane(obj.Width.Value, obj.Height.Value,FreeCAD.Vector(-obj.Width.Value/2,-obj.Height.Value/2,0))
|
|
if obj.WallThickness.Value and (obj.WallThickness.Value < obj.Height.Value/2) and (obj.WallThickness.Value < obj.Width.Value/2):
|
|
p2 = Part.makePlane(obj.Width.Value-obj.WallThickness.Value*2,obj.Height.Value-obj.WallThickness.Value*2,FreeCAD.Vector(obj.WallThickness.Value-obj.Width.Value/2,obj.WallThickness.Value-obj.Height.Value/2,0))
|
|
p = p.cut(p2)
|
|
else:
|
|
if obj.Diameter.Value == 0:
|
|
return
|
|
p = Part.Wire([Part.Circle(FreeCAD.Vector(0,0,0),FreeCAD.Vector(0,0,1),obj.Diameter.Value/2).toShape()])
|
|
if obj.WallThickness.Value and (obj.WallThickness.Value < obj.Diameter.Value/2):
|
|
p2 = Part.Wire([Part.Circle(FreeCAD.Vector(0,0,0),FreeCAD.Vector(0,0,1),(obj.Diameter.Value/2-obj.WallThickness.Value)).toShape()])
|
|
p = Part.Face(p)
|
|
p2 = Part.Face(p2)
|
|
p = p.cut(p2)
|
|
return p
|
|
|
|
|
|
class _ViewProviderPipe(ArchComponent.ViewProviderComponent):
|
|
|
|
|
|
"A View Provider for the Pipe object"
|
|
|
|
def __init__(self,vobj):
|
|
|
|
ArchComponent.ViewProviderComponent.__init__(self,vobj)
|
|
|
|
def getIcon(self):
|
|
|
|
import Arch_rc
|
|
return ":/icons/Arch_Pipe_Tree.svg"
|
|
|
|
|
|
class _ArchPipeConnector(ArchComponent.Component):
|
|
|
|
|
|
"the Arch Pipe Connector object"
|
|
|
|
def __init__(self,obj):
|
|
|
|
ArchComponent.Component.__init__(self,obj)
|
|
self.setProperties(obj)
|
|
obj.IfcType = "Pipe Fitting"
|
|
|
|
def setProperties(self,obj):
|
|
|
|
pl = obj.PropertiesList
|
|
if not "Radius" in pl:
|
|
obj.addProperty("App::PropertyLength", "Radius", "PipeConnector", QT_TRANSLATE_NOOP("App::Property","The curvature radius of this connector"), locked=True)
|
|
if not "Pipes" in pl:
|
|
obj.addProperty("App::PropertyLinkList", "Pipes", "PipeConnector", QT_TRANSLATE_NOOP("App::Property","The pipes linked by this connector"), locked=True)
|
|
if not "ConnectorType" in pl:
|
|
obj.addProperty("App::PropertyEnumeration", "ConnectorType", "PipeConnector", QT_TRANSLATE_NOOP("App::Property","The type of this connector"), locked=True)
|
|
obj.ConnectorType = ["Corner","Tee"]
|
|
obj.setEditorMode("ConnectorType",1)
|
|
self.Type = "PipeConnector"
|
|
|
|
def onDocumentRestored(self,obj):
|
|
|
|
ArchComponent.Component.onDocumentRestored(self,obj)
|
|
self.setProperties(obj)
|
|
|
|
def execute(self,obj):
|
|
|
|
if self.clone(obj):
|
|
return
|
|
|
|
tol = 1 # tolerance for alignment. This is only visual, we can keep it low...
|
|
ptol = 0.001 # tolerance for coincident points
|
|
|
|
import math
|
|
import Part
|
|
import DraftGeomUtils
|
|
import ArchCommands
|
|
if len(obj.Pipes) < 2:
|
|
return
|
|
if len(obj.Pipes) > 3:
|
|
FreeCAD.Console.PrintWarning(translate("Arch","Only the 3 first wires will be connected")+"\n")
|
|
if obj.Radius.Value == 0:
|
|
return
|
|
wires = []
|
|
order = []
|
|
for o in obj.Pipes:
|
|
wires.append(o.Proxy.getWire(o))
|
|
if wires[0].Vertexes[0].Point.sub(wires[1].Vertexes[0].Point).Length <= ptol:
|
|
order = ["start","start"]
|
|
point = wires[0].Vertexes[0].Point
|
|
elif wires[0].Vertexes[0].Point.sub(wires[1].Vertexes[-1].Point).Length <= ptol:
|
|
order = ["start","end"]
|
|
point = wires[0].Vertexes[0].Point
|
|
elif wires[0].Vertexes[-1].Point.sub(wires[1].Vertexes[-1].Point).Length <= ptol:
|
|
order = ["end","end"]
|
|
point = wires[0].Vertexes[-1].Point
|
|
elif wires[0].Vertexes[-1].Point.sub(wires[1].Vertexes[0].Point).Length <= ptol:
|
|
order = ["end","start"]
|
|
point = wires[0].Vertexes[-1].Point
|
|
else:
|
|
FreeCAD.Console.PrintError(translate("Arch","Common vertex not found")+"\n")
|
|
return
|
|
if order[0] == "start":
|
|
v1 = wires[0].Vertexes[1].Point.sub(wires[0].Vertexes[0].Point).normalize()
|
|
else:
|
|
v1 = wires[0].Vertexes[-2].Point.sub(wires[0].Vertexes[-1].Point).normalize()
|
|
if order[1] == "start":
|
|
v2 = wires[1].Vertexes[1].Point.sub(wires[1].Vertexes[0].Point).normalize()
|
|
else:
|
|
v2 = wires[1].Vertexes[-2].Point.sub(wires[1].Vertexes[-1].Point).normalize()
|
|
p = obj.Pipes[0].Proxy.getProfile(obj.Pipes[0])
|
|
if not p:
|
|
return
|
|
# If the pipe has a non-zero WallThickness p is a shape instead of a wire:
|
|
if p.ShapeType != "Wire":
|
|
p = p.Wires
|
|
p = Part.Face(p)
|
|
if len(obj.Pipes) == 2:
|
|
if obj.ConnectorType != "Corner":
|
|
obj.ConnectorType = "Corner"
|
|
if round(v1.getAngle(v2),tol) in [0,round(math.pi,tol)]:
|
|
FreeCAD.Console.PrintError(translate("Arch","Pipes are already aligned")+"\n")
|
|
return
|
|
normal = v2.cross(v1)
|
|
offset = math.tan(math.pi/2-v1.getAngle(v2)/2)*obj.Radius.Value
|
|
v1.multiply(offset)
|
|
v2.multiply(offset)
|
|
self.setOffset(obj.Pipes[0],order[0],offset)
|
|
self.setOffset(obj.Pipes[1],order[1],offset)
|
|
# find center
|
|
perp = v1.cross(normal).normalize()
|
|
perp.multiply(obj.Radius.Value)
|
|
center = point.add(v1).add(perp)
|
|
# move and rotate the profile to the first point
|
|
delta = point.add(v1)-p.CenterOfMass
|
|
p.translate(delta)
|
|
vp = DraftGeomUtils.getNormal(p)
|
|
#rot = FreeCAD.Rotation(vp,v1)
|
|
# rotate keeping up vector
|
|
if v1.getAngle(FreeCAD.Vector(0,0,1)) > 0.01:
|
|
up = FreeCAD.Vector(0,0,1)
|
|
else:
|
|
up = FreeCAD.Vector(0,1,0)
|
|
v1y = up.cross(v1)
|
|
v1x = v1.cross(v1y)
|
|
rot = FreeCAD.Rotation(v1x,v1y,v1,"ZYX")
|
|
p.rotate(p.CenterOfMass,rot.Axis,math.degrees(rot.Angle))
|
|
try:
|
|
sh = p.revolve(center,normal,math.degrees(math.pi-v1.getAngle(v2)))
|
|
except:
|
|
FreeCAD.Console.PrintError(translate("Arch","Unable to revolve this connector")+"\n")
|
|
return
|
|
#sh = Part.makeCompound([sh]+[Part.Vertex(point),Part.Vertex(point.add(v1)),Part.Vertex(center),Part.Vertex(point.add(v2))])
|
|
else:
|
|
if obj.ConnectorType != "Tee":
|
|
obj.ConnectorType = "Tee"
|
|
if wires[2].Vertexes[0].Point == point:
|
|
order.append("start")
|
|
elif wires[0].Vertexes[-1].Point == point:
|
|
order.append("end")
|
|
else:
|
|
FreeCAD.Console.PrintError(translate("Arch","Common vertex not found")+"\n")
|
|
if order[2] == "start":
|
|
v3 = wires[2].Vertexes[1].Point.sub(wires[2].Vertexes[0].Point).normalize()
|
|
else:
|
|
v3 = wires[2].Vertexes[-2].Point.sub(wires[2].Vertexes[-1].Point).normalize()
|
|
if round(v1.getAngle(v2),tol) in [0,round(math.pi,tol)]:
|
|
pair = [v1,v2,v3]
|
|
elif round(v1.getAngle(v3),tol) in [0,round(math.pi,tol)]:
|
|
pair = [v1,v3,v2]
|
|
elif round(v2.getAngle(v3),tol) in [0,round(math.pi,tol)]:
|
|
pair = [v2,v3,v1]
|
|
else:
|
|
FreeCAD.Console.PrintError(translate("Arch","At least 2 pipes must align")+"\n")
|
|
return
|
|
offset = obj.Radius.Value
|
|
v1.multiply(offset)
|
|
v2.multiply(offset)
|
|
v3.multiply(offset)
|
|
self.setOffset(obj.Pipes[0],order[0],offset)
|
|
self.setOffset(obj.Pipes[1],order[1],offset)
|
|
self.setOffset(obj.Pipes[2],order[2],offset)
|
|
normal = pair[0].cross(pair[2])
|
|
# move and rotate the profile to the first point
|
|
delta = point.add(pair[0])-p.CenterOfMass
|
|
p.translate(delta)
|
|
vp = DraftGeomUtils.getNormal(p)
|
|
rot = FreeCAD.Rotation(vp,pair[0])
|
|
p.rotate(p.CenterOfMass,rot.Axis,math.degrees(rot.Angle))
|
|
t1 = p.extrude(pair[1].multiply(2))
|
|
# move and rotate the profile to the second point
|
|
delta = point.add(pair[2])-p.CenterOfMass
|
|
p.translate(delta)
|
|
vp = DraftGeomUtils.getNormal(p)
|
|
rot = FreeCAD.Rotation(vp,pair[2])
|
|
p.rotate(p.CenterOfMass,rot.Axis,math.degrees(rot.Angle))
|
|
t2 = p.extrude(pair[2].negative().multiply(2))
|
|
# create a cut plane
|
|
cp = Part.makePolygon([point,point.add(pair[0]),point.add(normal),point])
|
|
cp = Part.Face(cp)
|
|
if cp.normalAt(0,0).getAngle(pair[2]) < math.pi/2:
|
|
cp.reverse()
|
|
cf, cv, invcv = ArchCommands.getCutVolume(cp,t2)
|
|
t2 = t2.cut(cv)
|
|
sh = t1.fuse(t2)
|
|
obj.Shape = sh
|
|
|
|
def setOffset(self,pipe,pos,offset):
|
|
|
|
if pos == "start":
|
|
if pipe.OffsetStart != offset:
|
|
pipe.OffsetStart = offset
|
|
pipe.Proxy.execute(pipe)
|
|
else:
|
|
if pipe.OffsetEnd != offset:
|
|
pipe.OffsetEnd = offset
|
|
pipe.Proxy.execute(pipe)
|