def placeTheElbow(c, v1=None, v2=None, P=None):
'''
placeTheElbow(c,v1,v2,P=None)
Puts the curve C between vectors v1 and v2.
If point P is given, translates it in there.
NOTE: v1 and v2 oriented in the same direction along the path!
'''
if not (v1 and v2):
v1, v2 = [e.tangentAt(0) for e in frameCmd.edges()]
try:
P = frameCmd.intersectionCLines(*frameCmd.edges())
except:
pass
if hasattr(c, 'PType') and hasattr(c, 'BendAngle') and v1 and v2:
v1.normalize()
v2.normalize()
ortho = rounded(frameCmd.ortho(v1, v2))
bisect = rounded(v2 - v1)
ang = degrees(v1.getAngle(v2))
c.BendAngle = ang
rot1 = FreeCAD.Rotation(
rounded(frameCmd.beamAx(c, FreeCAD.Vector(0, 0, 1))), ortho)
c.Placement.Rotation = rot1.multiply(c.Placement.Rotation)
rot2 = FreeCAD.Rotation(
rounded(frameCmd.beamAx(c, FreeCAD.Vector(1, 1, 0))), bisect)
c.Placement.Rotation = rot2.multiply(c.Placement.Rotation)
if not P:
P = c.Placement.Base
c.Placement.Base = P
def placeoTherElbow(c, v1=None, v2=None, P=None):
'''
Like placeTheElbow() but with more math.
'''
if not (v1 and v2):
v1, v2 = [e.tangentAt(0) for e in frameCmd.edges()]
try:
P = frameCmd.intersectionCLines(*frameCmd.edges())
except:
pass
if hasattr(c, 'PType') and hasattr(c, 'BendAngle') and v1 and v2:
v1.normalize()
v2.normalize()
ortho = rounded(frameCmd.ortho(v1, v2))
bisect = rounded(v2 - v1)
cBisect = rounded(c.Ports[1] + c.Ports[0]) # math
cZ = c.Ports[0].cross(c.Ports[1]) # more math
ang = degrees(v1.getAngle(v2))
c.BendAngle = ang
rot1 = FreeCAD.Rotation(rounded(frameCmd.beamAx(c, cZ)), ortho)
c.Placement.Rotation = rot1.multiply(c.Placement.Rotation)
rot2 = FreeCAD.Rotation(rounded(frameCmd.beamAx(c, cBisect)), bisect)
c.Placement.Rotation = rot2.multiply(c.Placement.Rotation)
if not P:
P = c.Placement.Base
c.Placement.Base = P
def Activated (self):
import frameCmd, pipeCmd
if len(frameCmd.beams())>1:
p1,p2=frameCmd.beams()[:2]
try:
P=frameCmd.intersectionCLines(p1,p2)
com1=p1.Shape.Solids[0].CenterOfMass
com2=p2.Shape.Solids[0].CenterOfMass
v1=P-com1
v2=com2-P
curves=[e for e in FreeCADGui.Selection.getSelection() if hasattr(e,'PType') and hasattr(e,'BendAngle')]
if curves:
FreeCAD.ActiveDocument.openTransaction('Place one curve')
pipeCmd.placeoTherElbow(curves[0],v1,v2,P)
FreeCAD.ActiveDocument.recompute() # recompute for the elbow
port1,port2=pipeCmd.portsPos(curves[0])
if (com1-port1).Length<(com1-port2).Length:
frameCmd.extendTheBeam(p1,port1)
frameCmd.extendTheBeam(p2,port2)
else:
frameCmd.extendTheBeam(p1,port2)
frameCmd.extendTheBeam(p2,port1)
FreeCAD.ActiveDocument.recompute() # recompute for the pipes
FreeCAD.ActiveDocument.commitTransaction()
else:
FreeCAD.Console.PrintError('Select at least one elbow')
except:
FreeCAD.Console.PrintError('Intersection point not found\n')
else:
FreeCAD.Console.PrintError('Select two intersecting pipes\n')
def makeW():
edges = frameCmd.edges()
if len(edges) > 1:
# patch for FC 0.17:
first = edges[0]
points = list()
while len(edges) > 1:
points.append(frameCmd.intersectionCLines(edges.pop(0), edges[0]))
if edges[0].valueAt(0) == points[-1]:
points.append(edges[0].valueAt(edges[0].LastParameter))
else:
points.append(edges[0].valueAt(0))
if first.valueAt(0) == points[0]:
points.insert(0, first.valueAt(first.LastParameter))
else:
points.insert(0, first.valueAt(0)) # END
from Draft import makeWire
try:
p = makeWire(points)
except:
FreeCAD.Console.PrintError('Missing intersection\n')
return None
p.Label = 'Path'
drawAsCenterLine(p)
return p
elif FreeCADGui.Selection.getSelection():
obj = FreeCADGui.Selection.getSelection()[0]
if hasattr(obj, 'Shape') and type(obj.Shape) == Part.Wire:
drawAsCenterLine(obj)
return obj
else:
return None
def Activated(self):
import frameCmd, pipeCmd
if len(frameCmd.beams()) > 1:
p1, p2 = frameCmd.beams()[:2]
try:
P = frameCmd.intersectionCLines(p1, p2)
com1 = p1.Shape.Solids[0].CenterOfMass
com2 = p2.Shape.Solids[0].CenterOfMass
v1 = P - com1
v2 = com2 - P
curves = [
e for e in FreeCADGui.Selection.getSelection()
if hasattr(e, 'PType') and hasattr(e, 'BendAngle')
]
if curves:
FreeCAD.ActiveDocument.openTransaction('Place one curve')
pipeCmd.placeoTherElbow(curves[0], v1, v2, P)
FreeCAD.ActiveDocument.recompute(
) # recompute for the elbow
port1, port2 = pipeCmd.portsPos(curves[0])
if (com1 - port1).Length < (com1 - port2).Length:
frameCmd.extendTheBeam(p1, port1)
frameCmd.extendTheBeam(p2, port2)
else:
frameCmd.extendTheBeam(p1, port2)
frameCmd.extendTheBeam(p2, port1)
FreeCAD.ActiveDocument.recompute(
) # recompute for the pipes
FreeCAD.ActiveDocument.commitTransaction()
else:
FreeCAD.Console.PrintError('Select at least one elbow')
except:
FreeCAD.Console.PrintError('Intersection point not found\n')
else:
FreeCAD.Console.PrintError('Select two intersecting pipes\n')
def flattenTheTube(obj=None, v1=None, v2=None, X=None):
'''
flattenTheTube(obj=None,v1=None,v2=None, X=None)
Put obj in the same plane defined by vectors v1 and v2 and move it to X.
obj: the object to be rotated
v1, v2: the vectors of the plane
X: the Placement.Base
If no parameter is defined: v1, v2 are the axis of the first two beams
in the selections set, X is their intersection and obj is the first other
object in the selection set.
'''
if None in [obj, v1, v2]:
try:
sel = FreeCADGui.Selection.getSelection()
t1, t2 = frameCmd.beams()[:2]
v1 = frameCmd.beamAx(t1)
v2 = frameCmd.beamAx(t2)
sel.remove(t1)
sel.remove(t2)
obj = sel[0]
X = frameCmd.intersectionCLines(t1, t2)
except:
FreeCAD.Console.PrintError('Not enough arguments\n')
return
obj.Placement.Rotation = FreeCAD.Rotation(
frameCmd.beamAx(obj), v1.cross(v2)).multiply(obj.Placement.Rotation)
obj.Placement.Base = X
def makeElbowBetweenThings(thing1=None, thing2=None, propList=None):
'''
makeElbowBetweenThings(thing1, thing2, propList=None):
Place one elbow at the intersection of thing1 and thing2.
Things can be any combination of intersecting beams, pipes or edges.
If nothing is passed as argument, the function attempts to take the
first two edges selected in the view.
propList is one optional list with 5 elements:
DN (string): nominal diameter
OD (float): outside diameter
thk (float): shell thickness
BA (float): bend angle - that will be recalculated! -
BR (float): bend radius
Default is "DN50 (SCH-STD)"
Remember: property PRating must be defined afterwards
'''
if not (thing1 and thing2):
thing1, thing2 = frameCmd.edges()[:2]
P = frameCmd.intersectionCLines(thing1, thing2)
directions = list()
try:
for thing in [thing1, thing2]:
if frameCmd.beams([thing]):
directions.append(
rounded(
(frameCmd.beamAx(thing).multiply(thing.Height / 2) +
thing.Placement.Base) - P))
elif hasattr(thing, 'ShapeType') and thing.ShapeType == 'Edge':
directions.append(rounded(thing.CenterOfMass - P))
except:
return None
ang = 180 - degrees(directions[0].getAngle(directions[1]))
if ang == 0 or ang == 180: return None
if not propList:
propList = ["DN50", 60.3, 3.91, ang, 45.24]
else:
propList[3] = ang
elb = makeElbow(propList, P,
directions[0].negative().cross(directions[1].negative()))
# mate the elbow ends with the pipes or edges
b = frameCmd.bisect(directions[0], directions[1])
elbBisect = rounded(frameCmd.beamAx(elb, FreeCAD.Vector(
1, 1, 0))) #if not rounded, fail in plane xz
rot = FreeCAD.Rotation(elbBisect, b)
elb.Placement.Rotation = rot.multiply(elb.Placement.Rotation)
# trim the adjacent tubes
#FreeCAD.activeDocument().recompute() # may delete this row?
portA = elb.Placement.multVec(elb.Ports[0])
portB = elb.Placement.multVec(elb.Ports[1])
for tube in [t for t in [thing1, thing2] if frameCmd.beams([t])]:
vectA = tube.Shape.Solids[0].CenterOfMass - portA
vectB = tube.Shape.Solids[0].CenterOfMass - portB
if frameCmd.isParallel(vectA, frameCmd.beamAx(tube)):
frameCmd.extendTheBeam(tube, portA)
else:
frameCmd.extendTheBeam(tube, portB)
return elb
def makeW():
edges = frameCmd.edges()
if len(edges) > 1:
# patch for FC 0.17:
first = edges[0]
points = list()
while len(edges) > 1:
points.append(frameCmd.intersectionCLines(edges.pop(0), edges[0]))
if edges[0].valueAt(0) == points[-1]:
points.append(edges[0].valueAt(edges[0].LastParameter))
else:
points.append(edges[0].valueAt(0))
if first.valueAt(0) == points[0]:
points.insert(0, first.valueAt(first.LastParameter))
else:
points.insert(0, first.valueAt(0)) # END
#P0=edges[0].valueAt(0)
#P1=edges[0].valueAt(edges[0].LastParameter)
#Pint=frameCmd.intersectionCLines(edges[0],edges[1])
#d0=Pint-P0
#d1=Pint-P1
#if d1.Length>d0.Length:
#P0=P1
#eds=list()
#for i in range(len(edges)-1):
#P1=frameCmd.intersectionCLines(edges[i],edges[i+1])
#eds.append(Part.Edge(Part.Line(P0,P1)))
#P0=P1
#P1=edges[-1].valueAt(edges[-1].LastParameter)
#P2=edges[-1].valueAt(0)
#d1=P1-P0
#d2=P2-P0
#if d1.Length<d2.Length:
#P1=P2
#eds.append(Part.Edge(Part.Line(P0,P1)))
#for e in eds:
#print(type(e))
#w=Part.Wire(eds)
#points=[e.valueAt(0) for e in w.Edges]
#last=e.Edges[-1]
#points.append(last.valueAt(last.LastParameter))
from Draft import makeWire
try:
p = makeWire(points)
except:
FreeCAD.Console.PrintError('Missing intersection\n')
return None
p.Label = 'Path'
drawAsCenterLine(p)
return p
else:
FreeCAD.Console.PrintError('Not enough edges/n')
return None
def extendTheTubes2intersection(pipe1=None,pipe2=None,both=True):
'''
Does what it says; also with beams.
If arguments are None, it picks the first 2 selected beams().
'''
if not (pipe1 and pipe2):
try:
pipe1,pipe2=frameCmd.beams()[:2]
except:
FreeCAD.Console.PrintError('Insufficient arguments for extendTheTubes2intersection\n')
P=frameCmd.intersectionCLines(pipe1,pipe2)
if P!=None:
frameCmd.extendTheBeam(pipe1,P)
if both:
frameCmd.extendTheBeam(pipe2,P)
def setWP():
'function to change working plane'
import FreeCAD, FreeCADGui, frameCmd
normal = point = None
curves = []
straight = []
Z = FreeCAD.Vector(0, 0, 1)
for edge in frameCmd.edges():
if edge.curvatureAt(0) != 0:
curves.append(edge)
else:
straight.append(edge)
# define normal: 1st from face->2nd from curve->3rd from straight edges
if frameCmd.faces():
normal = frameCmd.faces()[0].normalAt(0, 0)
elif curves:
normal = curves[0].tangentAt(0).cross(curves[0].normalAt(0))
elif len(straight) > 1:
if straight and not frameCmd.isParallel(straight[0].tangentAt(0),
straight[1].tangentAt(0)):
normal = straight[0].tangentAt(0).cross(straight[1].tangentAt(0))
elif FreeCADGui.Selection.getSelection():
normal = FreeCAD.DraftWorkingPlane.getRotation().multVec(Z)
else:
normal = Z
# define point: 1st from vertex->2nd from centerOfCurvature->3rd from intersection->4th from center of edge
points = [
v.Point for sx in FreeCADGui.Selection.getSelectionEx()
for v in sx.SubObjects if v.ShapeType == 'Vertex'
]
if not points:
points = [edge.centerOfCurvatureAt(0) for edge in curves]
if not points and len(straight) > 1:
inters = frameCmd.intersectionCLines(straight[0], straight[1])
if inters:
points.append(inters)
if not points and len(straight):
points.append(straight[0].CenterOfMass)
if points:
point = points[0]
else:
point = FreeCAD.Vector()
# move the draft WP
FreeCAD.DraftWorkingPlane.alignToPointAndAxis(point, normal)
FreeCADGui.Snapper.setGrid()
def setWP():
'function to change working plane'
import FreeCAD, FreeCADGui, frameCmd
normal=point=None
curves=[]
straight=[]
Z=FreeCAD.Vector(0,0,1)
for edge in frameCmd.edges():
if edge.curvatureAt(0)!=0:
curves.append(edge)
else:
straight.append(edge)
# define normal: 1st from face->2nd from curve->3rd from straight edges
if frameCmd.faces():
normal=frameCmd.faces()[0].normalAt(0,0)
elif curves:
normal=curves[0].tangentAt(0).cross(curves[0].normalAt(0))
elif len(straight)>1:
if straight and not frameCmd.isParallel(straight[0].tangentAt(0),straight[1].tangentAt(0)):
normal=straight[0].tangentAt(0).cross(straight[1].tangentAt(0))
elif FreeCADGui.Selection.getSelection():
normal=FreeCAD.DraftWorkingPlane.getRotation().multVec(Z)
else:
normal=Z
# define point: 1st from vertex->2nd from centerOfCurvature->3rd from intersection->4th from center of edge
points=[v.Point for sx in FreeCADGui.Selection.getSelectionEx() for v in sx.SubObjects if v.ShapeType=='Vertex']
if not points:
points=[edge.centerOfCurvatureAt(0) for edge in curves]
if not points and len(straight)>1:
inters=frameCmd.intersectionCLines(straight[0],straight[1])
if inters:
points.append(inters)
if not points and len(straight):
points.append(straight[0].CenterOfMass)
if points:
point=points[0]
else:
point=FreeCAD.Vector()
# move the draft WP
FreeCAD.DraftWorkingPlane.alignToPointAndAxis(point,normal)
FreeCADGui.Snapper.setGrid()
