#*************************************************************************** #* * #* Copyright (c) 2013 - Juergen Riegel * #* * #* 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, Fem, os,sys,string,math,shutil,glob,subprocess,tempfile from ApplyingBC_IC import ApplyingBC_IC if FreeCAD.GuiUp: import FreeCADGui,FemGui from FreeCAD import Vector from PyQt4 import QtCore, QtGui from pivy import coin import PyQt4.uic as uic __title__="Machine-Distortion Analysis managment" __author__ = "Juergen Riegel" __url__ = "http://www.freecadweb.org" def makeMechanicalAnalysis(name): '''makeFemAnalysis(name): makes a Fem Analysis object''' obj = FreeCAD.ActiveDocument.addObject("Fem::FemAnalysisPython",name) _FemAnalysis(obj) _ViewProviderFemAnalysis(obj.ViewObject) #FreeCAD.ActiveDocument.recompute() return obj class _CommandNewMechanicalAnalysis: "the Fem Analysis command definition" def GetResources(self): return {'Pixmap' : 'Fem_Analysis', 'MenuText': QtCore.QT_TRANSLATE_NOOP("Fem_Analysis","New mechanical analysis"), 'Accel': "A", 'ToolTip': QtCore.QT_TRANSLATE_NOOP("Fem_Analysis","Create a new mechanical analysis")} def Activated(self): FreeCAD.ActiveDocument.openTransaction("Create Analysis") FreeCADGui.addModule("FemGui") FreeCADGui.addModule("MechanicalAnalysis") #FreeCADGui.doCommand("FreeCADGui.ActiveDocument.ActiveView.setAxisCross(True)") FreeCADGui.doCommand("MechanicalAnalysis.makeMechanicalAnalysis('MechanicalAnalysis')") FreeCADGui.doCommand("FemGui.setActiveAnalysis(App.activeDocument().ActiveObject)") sel = FreeCADGui.Selection.getSelection() if (len(sel) == 1): if(sel[0].isDerivedFrom("Fem::FemMeshObject")): FreeCADGui.doCommand("App.activeDocument().ActiveObject.Member = App.activeDocument().ActiveObject.Member + [App.activeDocument()."+sel[0].Name+"]") if(sel[0].isDerivedFrom("Part::Feature")): FreeCADGui.doCommand("App.activeDocument().addObject('Fem::FemMeshShapeNetgenObject','"+sel[0].Name +"_Mesh')") FreeCADGui.doCommand("App.activeDocument().ActiveObject.Shape = App.activeDocument()."+sel[0].Name) FreeCADGui.doCommand("FemGui.getActiveAnalysis().Member = FemGui.getActiveAnalysis().Member + [App.activeDocument().ActiveObject]") FreeCADGui.doCommand("Gui.activeDocument().hide('"+sel[0].Name+"')") #FreeCADGui.doCommand("App.activeDocument().ActiveObject.touch()") #FreeCADGui.doCommand("App.activeDocument().recompute()") FreeCADGui.doCommand("Gui.activeDocument().setEdit(App.ActiveDocument.ActiveObject.Name)") FreeCAD.ActiveDocument.commitTransaction() FreeCADGui.Selection.clearSelection() def IsActive(self): import FemGui return FreeCADGui.ActiveDocument != None and FemGui.getActiveAnalysis() == None class _CommandMechanicalJobControl: "the Fem JobControl command definition" def GetResources(self): return {'Pixmap' : 'Fem_NewAnalysis', 'MenuText': QtCore.QT_TRANSLATE_NOOP("Fem_JobControl","Start calculation"), 'Accel': "A", 'ToolTip': QtCore.QT_TRANSLATE_NOOP("Fem_Analysis","Dialog to start the calculation of the machanical anlysis")} def Activated(self): import FemGui taskd = _JobControlTaskPanel(FemGui.getActiveAnalysis()) #taskd.obj = vobj.Object taskd.update() FreeCADGui.Control.showDialog(taskd) def IsActive(self): import FemGui return FreeCADGui.ActiveDocument != None and FemGui.getActiveAnalysis() != None class _FemAnalysis: "The Material object" def __init__(self,obj): self.Type = "FemAnalysis" obj.Proxy = self #obj.Material = StartMat obj.addProperty("App::PropertyString","OutputDir","Base","Directory where the jobs get generated") obj.addProperty("App::PropertyFloat","PlateThikness","Base","Thikness of the plate") def execute(self,obj): return def onChanged(self,obj,prop): if prop in ["MaterialName"]: return def __getstate__(self): return self.Type def __setstate__(self,state): if state: self.Type = state class _ViewProviderFemAnalysis: "A View Provider for the Material object" def __init__(self,vobj): #vobj.addProperty("App::PropertyLength","BubbleSize","Base", str(translate("Fem","The size of the axis bubbles"))) vobj.Proxy = self def getIcon(self): return ":/icons/Fem_FemMesh.svg" def attach(self, vobj): self.ViewObject = vobj self.Object = vobj.Object self.bubbles = None def updateData(self, obj, prop): return def onChanged(self, vobj, prop): return def doubleClicked(self,vobj): taskd = _JobControlTaskPanel(self.Object) taskd.obj = vobj.Object taskd.update() FreeCADGui.Control.showDialog(taskd) return True def __getstate__(self): return None def __setstate__(self,state): return None class _JobControlTaskPanel: '''The editmode TaskPanel for Material objects''' def __init__(self,object): # the panel has a tree widget that contains categories # for the subcomponents, such as additions, subtractions. # the categories are shown only if they are not empty. form_class, base_class = uic.loadUiType(FreeCAD.getHomePath() + "Mod/Fem/MechanicalAnalysis.ui") self.obj = object self.formUi = form_class() self.form = QtGui.QWidget() self.formUi.setupUi(self.form) self.params = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Machining_Distortion") #Connect Signals and Slots QtCore.QObject.connect(self.formUi.toolButton_chooseOutputDir, QtCore.SIGNAL("clicked()"), self.chooseOutputDir) QtCore.QObject.connect(self.formUi.pushButton_generate, QtCore.SIGNAL("clicked()"), self.run) self.update() def getStandardButtons(self): return int(QtGui.QDialogButtonBox.Close) def update(self): 'fills the widgets' self.formUi.lineEdit_outputDir.setText(self.params.GetString("JobDir",'/')) return def accept(self): FreeCADGui.Control.closeDialog() def reject(self): FreeCADGui.Control.closeDialog() def chooseOutputDir(self): print "chooseOutputDir" dirname = QtGui.QFileDialog.getExistingDirectory(None, 'Choose material directory',self.params.GetString("JobDir",'/')) if(dirname): self.params.SetString("JobDir",str(dirname)) self.formUi.lineEdit_outputDir.setText(dirname) def run(self): dirName = self.formUi.lineEdit_outputDir.text() MeshObject = None if FemGui.getActiveAnalysis(): for i in FemGui.getActiveAnalysis().Member: if i.isDerivedFrom("Fem::FemMeshObject"): MeshObject = i else: QtGui.QMessageBox.critical(None, "Missing prerequisit","No active Analysis") return if not MeshObject: QtGui.QMessageBox.critical(None, "Missing prerequisit","No mesh object in the Analysis") return MathObject = None for i in FemGui.getActiveAnalysis().Member: if i.isDerivedFrom("App::MaterialObjectPython"): MathObject = i if not MathObject: QtGui.QMessageBox.critical(None, "Missing prerequisit","No material object in the Analysis") return matmap = MathObject.Material IsoNodeObject = None for i in FemGui.getActiveAnalysis().Member: if i.isDerivedFrom("Fem::FemSetNodesObject"): IsoNodeObject = i if not IsoNodeObject: QtGui.QMessageBox.critical(None, "Missing prerequisit","No Isostatic nodes defined in the Analysis") return IsoNodes = IsoNodeObject.Nodes filename_without_suffix = MeshObject.Name #current_file_name young_modulus = float(matmap['FEM_youngsmodulus']) poisson_ratio = float(matmap['PartDist_poissonratio']) FreeCADGui.addCommand('Fem_NewMechanicalAnalysis',_CommandNewMechanicalAnalysis()) FreeCADGui.addCommand('Fem_MechanicalJobControl',_CommandMechanicalJobControl())