# -*- coding: utf-8 -*-
###################################################################################
#
# SheetMetalBend.py
#
# Copyright 2015 Shai Seger <shaise at gmail dot com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
# MA 02110-1301, USA.
#
#
###################################################################################
from FreeCAD import Gui
from PySide import QtCore, QtGui
import FreeCAD, FreeCADGui, Part, os, math
__dir__ = os.path.dirname(__file__)
iconPath = os.path.join( __dir__, 'Resources', 'icons' )
smEpsilon = 0.0000001
# IMPORTANT: please remember to change the element map version in case of any
# changes in modeling logic
smElementMapVersion = 'sm1.'
def smWarnDialog(msg):
diag = QtGui.QMessageBox(QtGui.QMessageBox.Warning, 'Error in macro MessageBox', msg)
diag.setWindowModality(QtCore.Qt.ApplicationModal)
diag.exec_()
def smBelongToBody(item, body):
if (body is None):
return False
for obj in body.Group:
if obj.Name == item.Name:
return True
return False
def smIsPartDesign(obj):
return str(obj).find("<PartDesign::") == 0
def smIsOperationLegal(body, selobj):
#FreeCAD.Console.PrintLog(str(selobj) + " " + str(body) + " " + str(smBelongToBody(selobj, body)) + "\n")
if smIsPartDesign(selobj) and not smBelongToBody(selobj, body):
smWarnDialog("The selected geometry does not belong to the active Body.\nPlease make the container of this item active by\ndouble clicking on it.")
return False
return True
def smSolidBend(thk = 1.0, radius = 1.0, selEdgeNames = '', MainObject = None):
import BOPTools.SplitFeatures, BOPTools.JoinFeatures
BendR = thk + radius
resultSolid = MainObject
for selEdgeName in selEdgeNames:
edge = MainObject.getElement(selEdgeName)
facelist = MainObject.ancestorsOfType(edge, Part.Face)
#for face in facelist :
# Part.show(face,'face')
joinface = facelist[0].fuse(facelist[1])
#Part.show(joinface,'joinface')
filletedface = joinface.makeFillet(BendR, joinface.Edges)
#Part.show(filletedface,'filletedface')
filletedoffsetface = filletedface.makeOffsetShape(-thk, 0.0, fill = False)
if filletedoffsetface.Area > filletedface.Area :
filletedface = joinface.makeFillet(radius, joinface.Edges)
#Part.show(filletedface,'filletedface')
cutface = filletedface.cut(facelist[0])
cutface = cutface.cut(facelist[1])
#Part.show(cutface1,'cutface1')
if filletedoffsetface.Area > filletedface.Area :
offsetsolid1 = cutface.makeOffsetShape(-thk, 0.0, fill = True)
#Part.show(offsetsolid1,'offsetsolid1')
else:
offsetsolid1 = cutface.makeOffsetShape(-thk, 0.0, fill = True)
#Part.show(offsetsolid1,'offsetsolid1')
cutsolid = BOPTools.JoinAPI.cutout_legacy(resultSolid, offsetsolid1, 0.0)
#Part.show(cutsolid,'cutsolid')
offsetsolid1 = cutsolid.fuse(offsetsolid1)
#Part.show(offsetsolid1,'offsetsolid1')
resultSolid = BOPTools.JoinAPI.cutout_legacy(resultSolid, offsetsolid1, 0.0)
#Part.show(resultsolid,'resultsolid')
resultSolid = resultSolid.fuse(offsetsolid1)
#Part.show(resultsolid,'resultsolid')
return resultSolid
class SMSolidBend:
def __init__(self, obj):
'''"Add Bend to Solid" '''
selobj = Gui.Selection.getSelectionEx()[0]
obj.addProperty("App::PropertyLength","radius","Parameters","Bend Radius").radius = 1.0
obj.addProperty("App::PropertyLength","thickness","Parameters","thickness of sheetmetal").thickness = 1.0
obj.addProperty("App::PropertyLinkSub", "baseObject", "Parameters", "Base object").baseObject = (selobj.Object, selobj.SubElementNames)
obj.Proxy = self
def getElementMapVersion(self, _fp, ver, _prop, restored):
if not restored:
return smElementMapVersion + ver
def execute(self, fp):
'''"Print a short message when doing a recomputation, this method is mandatory" '''
# pass selected object shape
Main_Object = fp.baseObject[0].Shape.copy()
s = smSolidBend(thk = fp.thickness.Value, radius = fp.radius.Value, selEdgeNames = fp.baseObject[1],
MainObject = Main_Object)
fp.Shape = s
fp.baseObject[0].ViewObject.Visibility = False
class SMBendViewProviderTree:
"A View provider that nests children objects under the created one"
def __init__(self, obj):
obj.Proxy = self
self.Object = obj.Object
def attach(self, obj):
self.Object = obj.Object
return
def updateData(self, fp, prop):
return
def getDisplayModes(self,obj):
modes=[]
return modes
def setDisplayMode(self,mode):
return mode
def onChanged(self, vp, prop):
return
def __getstate__(self):
# return {'ObjectName' : self.Object.Name}
return None
def __setstate__(self,state):
if state is not None:
import FreeCAD
doc = FreeCAD.ActiveDocument #crap
self.Object = doc.getObject(state['ObjectName'])
def claimChildren(self):
objs = []
if hasattr(self.Object,"baseObject"):
objs.append(self.Object.baseObject[0])
return objs
def getIcon(self):
return os.path.join( iconPath , 'AddBend.svg')
def setEdit(self,vobj,mode):
taskd = SMBendTaskPanel()
taskd.obj = vobj.Object
taskd.update()
self.Object.ViewObject.Visibility=False
self.Object.baseObject[0].ViewObject.Visibility=True
FreeCADGui.Control.showDialog(taskd)
return True
def unsetEdit(self,vobj,mode):
FreeCADGui.Control.closeDialog()
self.Object.baseObject[0].ViewObject.Visibility=False
self.Object.ViewObject.Visibility=True
return False
class SMBendViewProviderFlat:
"A View provider that nests children objects under the created one"
def __init__(self, obj):
obj.Proxy = self
self.Object = obj.Object
def attach(self, obj):
self.Object = obj.Object
return
def updateData(self, fp, prop):
return
def getDisplayModes(self,obj):
modes=[]
return modes
def setDisplayMode(self,mode):
return mode
def onChanged(self, vp, prop):
return
def __getstate__(self):
# return {'ObjectName' : self.Object.Name}
return None
def __setstate__(self,state):
if state is not None:
import FreeCAD
doc = FreeCAD.ActiveDocument #crap
self.Object = doc.getObject(state['ObjectName'])
def claimChildren(self):
return []
def getIcon(self):
return os.path.join( iconPath , 'AddBend.svg')
def setEdit(self,vobj,mode):
taskd = SMBendTaskPanel()
taskd.obj = vobj.Object
taskd.update()
self.Object.ViewObject.Visibility=False
self.Object.baseObject[0].ViewObject.Visibility=True
FreeCADGui.Control.showDialog(taskd)
return True
def unsetEdit(self,vobj,mode):
FreeCADGui.Control.closeDialog()
self.Object.baseObject[0].ViewObject.Visibility=False
self.Object.ViewObject.Visibility=True
return False
class SMBendTaskPanel:
'''A TaskPanel for the Sheetmetal'''
def __init__(self):
self.obj = None
self.form = QtGui.QWidget()
self.form.setObjectName("SMBendTaskPanel")
self.form.setWindowTitle("Binded faces/edges list")
self.grid = QtGui.QGridLayout(self.form)
self.grid.setObjectName("grid")
self.title = QtGui.QLabel(self.form)
self.grid.addWidget(self.title, 0, 0, 1, 2)
self.title.setText("Select new face(s)/Edge(s) and press Update")
# tree
self.tree = QtGui.QTreeWidget(self.form)
self.grid.addWidget(self.tree, 1, 0, 1, 2)
self.tree.setColumnCount(2)
self.tree.setHeaderLabels(["Name","Subelement"])
# buttons
self.addButton = QtGui.QPushButton(self.form)
self.addButton.setObjectName("addButton")
self.addButton.setIcon(QtGui.QIcon(os.path.join( iconPath , 'SMUpdate.svg')))
self.grid.addWidget(self.addButton, 3, 0, 1, 2)
QtCore.QObject.connect(self.addButton, QtCore.SIGNAL("clicked()"), self.updateElement)
self.update()
def isAllowedAlterSelection(self):
return True
def isAllowedAlterView(self):
return True
def getStandardButtons(self):
return int(QtGui.QDialogButtonBox.Ok)
def update(self):
'fills the treewidget'
self.tree.clear()
if self.obj:
f = self.obj.baseObject
if isinstance(f[1],list):
for subf in f[1]:
#FreeCAD.Console.PrintLog("item: " + subf + "\n")
item = QtGui.QTreeWidgetItem(self.tree)
item.setText(0,f[0].Name)
item.setIcon(0,QtGui.QIcon(":/icons/Tree_Part.svg"))
item.setText(1,subf)
else:
item = QtGui.QTreeWidgetItem(self.tree)
item.setText(0,f[0].Name)
item.setIcon(0,QtGui.QIcon(":/icons/Tree_Part.svg"))
item.setText(1,f[1][0])
self.retranslateUi(self.form)
def updateElement(self):
if self.obj:
sel = FreeCADGui.Selection.getSelectionEx()[0]
if sel.HasSubObjects:
obj = sel.Object
for elt in sel.SubElementNames:
if "Face" in elt or "Edge" in elt:
face = self.obj.baseObject
found = False
if (face[0] == obj.Name):
if isinstance(face[1],tuple):
for subf in face[1]:
if subf == elt:
found = True
else:
if (face[1][0] == elt):
found = True
if not found:
self.obj.baseObject = (sel.Object, sel.SubElementNames)
self.update()
def accept(self):
FreeCAD.ActiveDocument.recompute()
FreeCADGui.ActiveDocument.resetEdit()
#self.obj.ViewObject.Visibility=True
return True
def retranslateUi(self, TaskPanel):
#TaskPanel.setWindowTitle(QtGui.QApplication.translate("draft", "Faces", None))
self.addButton.setText(QtGui.QApplication.translate("draft", "Update", None))
class AddBendCommandClass():
"""Add Bend command"""
def GetResources(self):
return {'Pixmap' : os.path.join( iconPath , 'AddBend.svg'), # the name of a svg file available in the resources
'MenuText': QtCore.QT_TRANSLATE_NOOP('SheetMetal','Make Bend'),
'ToolTip' : QtCore.QT_TRANSLATE_NOOP('SheetMetal','Create Bend on solids')}
def Activated(self):
doc = FreeCAD.ActiveDocument
view = Gui.ActiveDocument.ActiveView
activeBody = None
selobj = Gui.Selection.getSelectionEx()[0].Object
if hasattr(view,'getActiveObject'):
activeBody = view.getActiveObject('pdbody')
if not smIsOperationLegal(activeBody, selobj):
return
doc.openTransaction("Add Bend")
if activeBody is None or not smIsPartDesign(selobj):
a = doc.addObject("Part::FeaturePython","SolidBend")
SMSolidBend(a)
SMBendViewProviderTree(a.ViewObject)
else:
#FreeCAD.Console.PrintLog("found active body: " + activeBody.Name)
a = doc.addObject("PartDesign::FeaturePython","SolidBend")
SMSolidBend(a)
SMBendViewProviderFlat(a.ViewObject)
activeBody.addObject(a)
FreeCADGui.Selection.clearSelection()
doc.recompute()
doc.commitTransaction()
return
def IsActive(self):
if len(Gui.Selection.getSelection()) < 1 or len(Gui.Selection.getSelectionEx()[0].SubElementNames) < 1:
return False
selobj = Gui.Selection.getSelection()[0]
for selFace in Gui.Selection.getSelectionEx()[0].SubObjects:
if type(selFace) != Part.Edge :
return False
return True
Gui.addCommand('SMMakeBend',AddBendCommandClass())
