80dbfd5814
Removes unused imports as reported by LGTM. There are exceptions: `import Arch_rc` is shown as an alert, but has side effects. It's not clear what the best thing to do in those cases is, so I've left them for now.
273 lines
12 KiB
Python
273 lines
12 KiB
Python
#***************************************************************************
|
|
#* Copyright (c) 2013 Yorik van Havre <yorik@uncreated.net> *
|
|
#* *
|
|
#* This program is free software; you can redistribute it and/or modify *
|
|
#* it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
#* as published by the Free Software Foundation; either version 2 of *
|
|
#* the License, or (at your option) any later version. *
|
|
#* for detail see the LICENCE text file. *
|
|
#* *
|
|
#* This program 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 Library General Public License for more details. *
|
|
#* *
|
|
#* You should have received a copy of the GNU Library General Public *
|
|
#* License along with this program; if not, write to the Free Software *
|
|
#* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
#* USA *
|
|
#* *
|
|
#***************************************************************************
|
|
|
|
import FreeCAD,Draft,ArchComponent,DraftVecUtils
|
|
if FreeCAD.GuiUp:
|
|
import FreeCADGui
|
|
from DraftTools import translate
|
|
from PySide.QtCore import QT_TRANSLATE_NOOP
|
|
else:
|
|
# \cond
|
|
def translate(ctxt,txt):
|
|
return txt
|
|
def QT_TRANSLATE_NOOP(ctxt,txt):
|
|
return txt
|
|
# \endcond
|
|
|
|
## @package ArchFrame
|
|
# \ingroup ARCH
|
|
# \brief The Frame object and tools
|
|
#
|
|
# This module provides tools to build Frame objects.
|
|
# Frames are objects made of a profile and an object with
|
|
# edges along which the profile gets extruded
|
|
|
|
__title__ = "FreeCAD Arch Frame"
|
|
__author__ = "Yorik van Havre"
|
|
__url__ = "https://www.freecadweb.org"
|
|
|
|
|
|
def makeFrame(baseobj,profile,name=translate("Arch","Frame")):
|
|
|
|
"""makeFrame(baseobj,profile,[name]): creates a frame object from a base sketch (or any other object
|
|
containing wires) and a profile object (an extrudable 2D object containing faces or closed wires)"""
|
|
|
|
if not FreeCAD.ActiveDocument:
|
|
FreeCAD.Console.PrintError("No active document. Aborting\n")
|
|
return
|
|
obj = FreeCAD.ActiveDocument.addObject("Part::FeaturePython","Frame")
|
|
obj.Label = translate("Arch",name)
|
|
_Frame(obj)
|
|
if FreeCAD.GuiUp:
|
|
_ViewProviderFrame(obj.ViewObject)
|
|
if baseobj:
|
|
obj.Base = baseobj
|
|
if profile:
|
|
obj.Profile = profile
|
|
if FreeCAD.GuiUp:
|
|
profile.ViewObject.hide()
|
|
return obj
|
|
|
|
|
|
class _CommandFrame:
|
|
|
|
"the Arch Frame command definition"
|
|
|
|
def GetResources(self):
|
|
|
|
return {'Pixmap' : 'Arch_Frame',
|
|
'MenuText': QT_TRANSLATE_NOOP("Arch_Frame","Frame"),
|
|
'Accel': "F, R",
|
|
'ToolTip': QT_TRANSLATE_NOOP("Arch_Frame","Creates a frame object from a planar 2D object (the extrusion path(s)) and a profile. Make sure objects are selected in that order.")}
|
|
|
|
def IsActive(self):
|
|
|
|
return not FreeCAD.ActiveDocument is None
|
|
|
|
def Activated(self):
|
|
|
|
s = FreeCADGui.Selection.getSelection()
|
|
if len(s) == 2:
|
|
FreeCAD.ActiveDocument.openTransaction(translate("Arch","Create Frame"))
|
|
FreeCADGui.addModule("Arch")
|
|
FreeCADGui.doCommand("obj = Arch.makeFrame(FreeCAD.ActiveDocument."+s[0].Name+",FreeCAD.ActiveDocument."+s[1].Name+")")
|
|
FreeCADGui.addModule("Draft")
|
|
FreeCADGui.doCommand("Draft.autogroup(obj)")
|
|
FreeCAD.ActiveDocument.commitTransaction()
|
|
FreeCAD.ActiveDocument.recompute()
|
|
|
|
|
|
class _Frame(ArchComponent.Component):
|
|
|
|
"A parametric frame object"
|
|
|
|
def __init__(self,obj):
|
|
ArchComponent.Component.__init__(self,obj)
|
|
self.setProperties(obj)
|
|
obj.IfcType = "Railing"
|
|
|
|
def setProperties(self,obj):
|
|
|
|
pl = obj.PropertiesList
|
|
if not "Profile" in pl:
|
|
obj.addProperty("App::PropertyLink","Profile","Frame",QT_TRANSLATE_NOOP("App::Property","The profile used to build this frame"))
|
|
if not "Align" in pl:
|
|
obj.addProperty("App::PropertyBool","Align","Frame",QT_TRANSLATE_NOOP("App::Property","Specifies if the profile must be aligned with the extrusion wires"))
|
|
obj.Align = True
|
|
if not "Offset" in pl:
|
|
obj.addProperty("App::PropertyVectorDistance","Offset","Frame",QT_TRANSLATE_NOOP("App::Property","An offset vector between the base sketch and the frame"))
|
|
if not "BasePoint" in pl:
|
|
obj.addProperty("App::PropertyInteger","BasePoint","Frame",QT_TRANSLATE_NOOP("App::Property","Crossing point of the path on the profile."))
|
|
if not "ProfilePlacement" in pl:
|
|
obj.addProperty("App::PropertyPlacement","ProfilePlacement","Frame",QT_TRANSLATE_NOOP("App::Property","An optional additional placement to add to the profile before extruding it"))
|
|
if not "Rotation" in pl:
|
|
obj.addProperty("App::PropertyAngle","Rotation","Frame",QT_TRANSLATE_NOOP("App::Property","The rotation of the profile around its extrusion axis"))
|
|
if not "Edges" in pl:
|
|
obj.addProperty("App::PropertyEnumeration","Edges","Frame",QT_TRANSLATE_NOOP("App::Property","The type of edges to consider"))
|
|
obj.Edges = ["All edges","Vertical edges","Horizontal edges","Bottom horizontal edges","Top horizontal edges"]
|
|
if not "Fuse" in pl:
|
|
obj.addProperty("App::PropertyBool","Fuse","Frame",QT_TRANSLATE_NOOP("App::Property","If true, geometry is fused, otherwise a compound"))
|
|
self.Type = "Frame"
|
|
|
|
def onDocumentRestored(self,obj):
|
|
|
|
ArchComponent.Component.onDocumentRestored(self,obj)
|
|
self.setProperties(obj)
|
|
|
|
def execute(self,obj):
|
|
|
|
if self.clone(obj):
|
|
return
|
|
|
|
if not obj.Base:
|
|
return
|
|
if not obj.Base.Shape:
|
|
return
|
|
if not obj.Base.Shape.Wires:
|
|
return
|
|
|
|
pl = obj.Placement
|
|
if obj.Base.Shape.Solids:
|
|
obj.Shape = obj.Base.Shape.copy()
|
|
if not pl.isNull():
|
|
obj.Placement = obj.Shape.Placement.multiply(pl)
|
|
else:
|
|
if not obj.Profile:
|
|
return
|
|
if not obj.Profile.isDerivedFrom("Part::Part2DObject"):
|
|
return
|
|
if not obj.Profile.Shape:
|
|
return
|
|
if not obj.Profile.Shape.Wires:
|
|
return
|
|
if not obj.Profile.Shape.Faces:
|
|
for w in obj.Profile.Shape.Wires:
|
|
if not w.isClosed():
|
|
return
|
|
import DraftGeomUtils, Part, math
|
|
baseprofile = obj.Profile.Shape.copy()
|
|
if hasattr(obj,"ProfilePlacement"):
|
|
if not obj.ProfilePlacement.isNull():
|
|
baseprofile.Placement = obj.ProfilePlacement.multiply(baseprofile.Placement)
|
|
if not baseprofile.Faces:
|
|
f = []
|
|
for w in baseprofile.Wires:
|
|
f.append(Part.Face(w))
|
|
if len(f) == 1:
|
|
baseprofile = f[0]
|
|
else:
|
|
baseprofile = Part.makeCompound(f)
|
|
shapes = []
|
|
normal = DraftGeomUtils.getNormal(obj.Base.Shape)
|
|
#for wire in obj.Base.Shape.Wires:
|
|
edges = obj.Base.Shape.Edges
|
|
if hasattr(obj,"Edges"):
|
|
if obj.Edges == "Vertical edges":
|
|
rv = obj.Base.Placement.Rotation.multVec(FreeCAD.Vector(0,1,0))
|
|
edges = [e for e in edges if round(rv.getAngle(e.tangentAt(e.FirstParameter)),4) in [0,3.1416]]
|
|
elif obj.Edges == "Horizontal edges":
|
|
rv = obj.Base.Placement.Rotation.multVec(FreeCAD.Vector(1,0,0))
|
|
edges = [e for e in edges if round(rv.getAngle(e.tangentAt(e.FirstParameter)),4) in [0,3.1416]]
|
|
elif obj.Edges == "Top Horizontal edges":
|
|
rv = obj.Base.Placement.Rotation.multVec(FreeCAD.Vector(1,0,0))
|
|
edges = [e for e in edges if round(rv.getAngle(e.tangentAt(e.FirstParameter)),4) in [0,3.1416]]
|
|
edges = sorted(edges,key=lambda x: x.CenterOfMass.z,reverse=True)
|
|
z = edges[0].CenterOfMass.z
|
|
edges = [e for e in edges if abs(e.CenterOfMass.z-z) < 0.00001]
|
|
elif obj.Edges == "Bottom Horizontal edges":
|
|
rv = obj.Base.Placement.Rotation.multVec(FreeCAD.Vector(1,0,0))
|
|
edges = [e for e in edges if round(rv.getAngle(e.tangentAt(e.FirstParameter)),4) in [0,3.1416]]
|
|
edges = sorted(edges,key=lambda x: x.CenterOfMass.z)
|
|
z = edges[0].CenterOfMass.z
|
|
edges = [e for e in edges if abs(e.CenterOfMass.z-z) < 0.00001]
|
|
for e in edges:
|
|
#e = wire.Edges[0]
|
|
bvec = DraftGeomUtils.vec(e)
|
|
bpoint = e.Vertexes[0].Point
|
|
profile = baseprofile.copy()
|
|
#basepoint = profile.Placement.Base
|
|
if hasattr(obj,"BasePoint"):
|
|
edges = Part.__sortEdges__(profile.Edges)
|
|
basepointliste = [profile.CenterOfMass]
|
|
for edge in edges:
|
|
basepointliste.append(DraftGeomUtils.findMidpoint(edge))
|
|
basepointliste.append(edge.Vertexes[-1].Point)
|
|
try:
|
|
basepoint = basepointliste[obj.BasePoint]
|
|
except IndexError:
|
|
FreeCAD.Console.PrintMessage(translate("Arch","Crossing point not found in profile.")+"\n")
|
|
basepoint = basepointliste[0]
|
|
else :
|
|
basepoint = profile.CenterOfMass
|
|
profile.translate(bpoint.sub(basepoint))
|
|
if obj.Align:
|
|
axis = profile.Placement.Rotation.multVec(FreeCAD.Vector(0,0,1))
|
|
angle = bvec.getAngle(axis)
|
|
if round(angle,Draft.precision()) != 0:
|
|
if round(angle,Draft.precision()) != round(math.pi,Draft.precision()):
|
|
rotaxis = axis.cross(bvec)
|
|
profile.rotate(DraftVecUtils.tup(bpoint), DraftVecUtils.tup(rotaxis), math.degrees(angle))
|
|
if obj.Rotation:
|
|
profile.rotate(DraftVecUtils.tup(bpoint), DraftVecUtils.tup(FreeCAD.Vector(bvec).normalize()), obj.Rotation)
|
|
#profile = wire.makePipeShell([profile],True,False,2) TODO buggy
|
|
profile = profile.extrude(bvec)
|
|
if obj.Offset:
|
|
if not DraftVecUtils.isNull(obj.Offset):
|
|
profile.translate(obj.Offset)
|
|
shapes.append(profile)
|
|
if shapes:
|
|
if hasattr(obj,"Fuse"):
|
|
if obj.Fuse:
|
|
if len(shapes) > 1:
|
|
s = shapes[0].multiFuse(shapes[1:])
|
|
s = s.removeSplitter()
|
|
obj.Shape = s
|
|
obj.Placement = pl
|
|
return
|
|
obj.Shape = Part.makeCompound(shapes)
|
|
obj.Placement = pl
|
|
|
|
|
|
class _ViewProviderFrame(ArchComponent.ViewProviderComponent):
|
|
|
|
"A View Provider for the Frame object"
|
|
|
|
def __init__(self,vobj):
|
|
|
|
ArchComponent.ViewProviderComponent.__init__(self,vobj)
|
|
|
|
def getIcon(self):
|
|
|
|
import Arch_rc
|
|
return ":/icons/Arch_Frame_Tree.svg"
|
|
|
|
def claimChildren(self):
|
|
|
|
p = []
|
|
if hasattr(self,"Object"):
|
|
if self.Object.Profile:
|
|
p = [self.Object.Profile]
|
|
return ArchComponent.ViewProviderComponent.claimChildren(self)+p
|
|
|
|
|
|
if FreeCAD.GuiUp:
|
|
FreeCADGui.addCommand('Arch_Frame',_CommandFrame())
|