#*************************************************************************** #* Copyright (c) 2020 Adam Spontarelli * #* * #* 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, Part from fcsprocket import fcsprocket from fcsprocket import sprocket if FreeCAD.GuiUp: import FreeCADGui from PySide import QtCore, QtGui from FreeCADGui import PySideUic as uic __title__="PartDesign SprocketObject management" __author__ = "Adam Spontarelli" __url__ = "http://www.freecadweb.org" def makeSprocket(name): """ makeSprocket(name): makes a Sprocket """ obj = FreeCAD.ActiveDocument.addObject("Part::Part2DObjectPython",name) Sprocket(obj) if FreeCAD.GuiUp: ViewProviderSprocket(obj.ViewObject) #FreeCAD.ActiveDocument.recompute() if FreeCAD.GuiUp: body=FreeCADGui.ActiveDocument.ActiveView.getActiveObject("pdbody") part=FreeCADGui.ActiveDocument.ActiveView.getActiveObject("part") if body: body.Group=body.Group+[obj] elif part: part.Group=part.Group+[obj] return obj class CommandSprocket: """ the Fem Sprocket command definition """ def GetResources(self): return {'Pixmap' : 'PartDesign_Sprocket', 'MenuText': QtCore.QT_TRANSLATE_NOOP("PartDesign_Sprocket","Sprocket..."), 'Accel': "", 'ToolTip': QtCore.QT_TRANSLATE_NOOP("PartDesign_Sprocket","Creates or edit the sprocket definition.")} def Activated(self): FreeCAD.ActiveDocument.openTransaction("Create Sprocket") FreeCADGui.addModule("SprocketFeature") FreeCADGui.doCommand("SprocketFeature.makeSprocket('Sprocket')") FreeCADGui.doCommand("Gui.activeDocument().setEdit(App.ActiveDocument.ActiveObject.Name,0)") def IsActive(self): if FreeCAD.ActiveDocument: return True else: return False class Sprocket: """ The Sprocket object """ def __init__(self,obj): self.Type = "Sprocket" obj.addProperty("App::PropertyInteger","NumberOfTeeth","Sprocket","Number of gear teeth") obj.addProperty("App::PropertyLength","Pitch","Sprocket","Chain Pitch") obj.addProperty("App::PropertyLength","RollerDiameter","Sprocket","Roller Diameter") obj.addProperty("App::PropertyString","ANSISize","Sprocket","ANSI Size") obj.NumberOfTeeth = 50 obj.Pitch = "0.375 in" obj.RollerDiameter = "0.20 in" obj.ANSISize = "35" obj.Proxy = self def execute(self,obj): w = fcsprocket.FCWireBuilder() sprocket.CreateSprocket(w, obj.Pitch.Value, obj.NumberOfTeeth, obj.RollerDiameter.Value) sprocketw = Part.Wire([o.toShape() for o in w.wire]) obj.Shape = sprocketw obj.positionBySupport(); return class ViewProviderSprocket: """ A View Provider for the Sprocket object """ def __init__(self,vobj): vobj.Proxy = self def getIcon(self): return ":/icons/PartDesign_Sprocket.svg" def attach(self, vobj): self.ViewObject = vobj self.Object = vobj.Object def setEdit(self,vobj,mode): taskd = SprocketTaskPanel(self.Object,mode) taskd.obj = vobj.Object taskd.update() FreeCADGui.Control.showDialog(taskd) return True def unsetEdit(self,vobj,mode): FreeCADGui.Control.closeDialog() return def __getstate__(self): return None def __setstate__(self,state): return None class SprocketTaskPanel: """ The editmode TaskPanel for Sprocket objects """ def __init__(self,obj,mode): self.obj = obj self.form=FreeCADGui.PySideUic.loadUi(FreeCAD.getHomePath() + "Mod/PartDesign/SprocketFeature.ui") self.form.setWindowIcon(QtGui.QIcon(":/icons/PartDesign_Sprocket.svg")) QtCore.QObject.connect(self.form.Quantity_Pitch, QtCore.SIGNAL("valueChanged(double)"), self.pitchChanged) QtCore.QObject.connect(self.form.Quantity_RollerDiameter, QtCore.SIGNAL("valueChanged(double)"), self.rollerDiameterChanged) QtCore.QObject.connect(self.form.spinBox_NumberOfTeeth, QtCore.SIGNAL("valueChanged(int)"), self.numTeethChanged) QtCore.QObject.connect(self.form.comboBox_ANSISize, QtCore.SIGNAL("currentTextChanged(const QString)"), self.ANSISizeChanged) self.update() if mode == 0: # fresh created self.obj.Proxy.execute(self.obj) # calculate once FreeCAD.Gui.SendMsgToActiveView("ViewFit") def transferTo(self): """ Transfer from the dialog to the object """ self.obj.NumberOfTeeth = self.form.spinBox_NumberOfTeeth.value() self.obj.Pitch = self.form.Quantity_Pitch.text() self.obj.RollerDiameter = self.form.Quantity_RollerDiameter.text() self.obj.ANSISize = self.form.comboBox_ANSISize.currentText() def transferFrom(self): """ Transfer from the object to the dialog """ self.form.spinBox_NumberOfTeeth.setValue(self.obj.NumberOfTeeth) self.form.Quantity_Pitch.setText(self.obj.Pitch.UserString) self.form.Quantity_RollerDiameter.setText(self.obj.RollerDiameter.UserString) self.form.comboBox_ANSISize.setCurrentText(self.obj.ANSISize) def pitchChanged(self, value): self.obj.Pitch = value self.obj.Proxy.execute(self.obj) FreeCAD.Gui.SendMsgToActiveView("ViewFit") def ANSISizeChanged(self, size): """ ANSI B29.1-2011 standard roller chain sizes in USCS units (inches) {size: [Pitch, Roller Diameter]} """ ANSIRollerTable = {"25": [0.250, 0.130], "35": [0.375, 0.200], "41": [0.500, 0.306], "40": [0.500, 0.312], "50": [0.625, 0.400], "60": [0.750, 0.469], "80": [1.000, 0.625], "100":[1.250, 0.750], "120":[1.500, 0.875], "140":[1.750, 1.000], "160":[2.000, 1.125], "180":[2.250, 1.460], "200":[2.500, 1.562], "240":[3.000, 1.875]} self.obj.Pitch = str(ANSIRollerTable[size][0]) + " in" self.obj.RollerDiameter = str(ANSIRollerTable[size][1]) + " in" self.form.Quantity_Pitch.setText(self.obj.Pitch.UserString) self.form.Quantity_RollerDiameter.setText(self.obj.RollerDiameter.UserString) self.obj.Proxy.execute(self.obj) FreeCAD.Gui.SendMsgToActiveView("ViewFit") def rollerDiameterChanged(self, value): self.obj.RollerDiameter = value self.obj.Proxy.execute(self.obj) def numTeethChanged(self, value): self.obj.NumberOfTeeth = value self.obj.Proxy.execute(self.obj) FreeCAD.Gui.SendMsgToActiveView("ViewFit") def getStandardButtons(self): return int(QtGui.QDialogButtonBox.Ok) | int(QtGui.QDialogButtonBox.Cancel)| int(QtGui.QDialogButtonBox.Apply) def clicked(self,button): if button == QtGui.QDialogButtonBox.Apply: self.transferTo() self.obj.Proxy.execute(self.obj) def update(self): self.transferFrom() def accept(self): self.transferTo() FreeCAD.ActiveDocument.recompute() FreeCADGui.ActiveDocument.resetEdit() def reject(self): FreeCADGui.ActiveDocument.resetEdit() FreeCAD.ActiveDocument.abortTransaction() if FreeCAD.GuiUp: FreeCADGui.addCommand('PartDesign_Sprocket', CommandSprocket())