Skip to content

Commit

Permalink
Improve runing the Calculix process
Browse files Browse the repository at this point in the history
  • Loading branch information
@jriegel
jriegel committed Feb 18, 2014
1 parent e5eecc4 commit bb98b5a
Show file tree
Hide file tree
Showing 2 changed files with 94 additions and 24 deletions.
86 changes: 65 additions & 21 deletions src/Mod/Fem/MechanicalAnalysis.py
View file
Expand Up @@ -21,7 +21,7 @@
#***************************************************************************

import FreeCAD, Fem, FemLib, CalculixLib
import os,sys,string,math,shutil,glob,subprocess,tempfile
import os,sys,string,math,shutil,glob,subprocess,tempfile,time

if FreeCAD.GuiUp:
import FreeCADGui,FemGui
Expand Down Expand Up @@ -217,13 +217,22 @@ def __init__(self,object):
# 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.CalculixBinary = FreeCAD.getHomePath() +'bin/ccx.exe'
self.TempDir = FreeCAD.ActiveDocument.TransientDir.replace('\\','/') + '/FemAnl_'+ object.Uid[-4:]
if not os.path.isdir(self.TempDir):
os.mkdir(self.TempDir)

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/Fem")
self.Calculix = QtCore.QProcess()
self.Timer = QtCore.QTimer()
self.Timer.start(300)

self.OutStr = ''

#Connect Signals and Slots
QtCore.QObject.connect(self.formUi.toolButton_chooseOutputDir, QtCore.SIGNAL("clicked()"), self.chooseOutputDir)
Expand All @@ -232,9 +241,20 @@ def __init__(self,object):
QtCore.QObject.connect(self.Calculix, QtCore.SIGNAL("started()"), self.calculixStarted)
QtCore.QObject.connect(self.Calculix, QtCore.SIGNAL("finished(int)"), self.calculixFinished)

QtCore.QObject.connect(self.Timer, QtCore.SIGNAL("timeout()"), self.UpdateText)

self.update()


def UpdateText(self):
if(self.Calculix.state() == QtCore.QProcess.ProcessState.Running):
out = self.Calculix.readAllStandardOutput()
#print out
if out:
self.OutStr = self.OutStr + unicode(out).replace('\n','<br>')
self.formUi.textEdit_Output.setText(self.OutStr)
self.formUi.label_Time.setText('Time: {0:4.1f}: '.format(time.time() - self.Start) )

def calculixError(self,error):
print "Error()",error

Expand All @@ -247,11 +267,29 @@ def calculixStarted(self):
def calculixFinished(self,exitCode):
print "calculixFinished()",exitCode
print self.Calculix.state()
out = self.Calculix.readAllStandardOutput()
print out
if out:
self.OutStr = self.OutStr + unicode(out).replace('\n','<br>')
self.formUi.textEdit_Output.setText(self.OutStr)

self.Timer.stop()

self.OutStr = self.OutStr + '<font color="#0000FF">{0:4.1f}:</font> '.format(time.time() - self.Start) + '<font color="#00FF00">Calculix done!</font><br>'
self.formUi.textEdit_Output.setText(self.OutStr)

self.formUi.pushButton_generate.setText("Re-run Calculix")
print "Loading results...."
self.OutStr = self.OutStr + '<font color="#0000FF">{0:4.1f}:</font> '.format(time.time() - self.Start) + 'Loading result sets...<br>'
self.formUi.textEdit_Output.setText(self.OutStr)
self.formUi.label_Time.setText('Time: {0:4.1f}: '.format(time.time() - self.Start) )

QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
CalculixLib.importFrd('c:/users/jriegel/appdata/local/temp/Pocket_Mesh.frd',FemGui.getActiveAnalysis() )
CalculixLib.importFrd(self.Basename + '.frd',FemGui.getActiveAnalysis() )
QApplication.restoreOverrideCursor()
self.OutStr = self.OutStr + '<font color="#0000FF">{0:4.1f}:</font> '.format(time.time() - self.Start) + '<font color="#00FF00">Loading results done!</font><br>'
self.formUi.textEdit_Output.setText(self.OutStr)
self.formUi.label_Time.setText('Time: {0:4.1f}: '.format(time.time() - self.Start) )

def getStandardButtons(self):
return int(QtGui.QDialogButtonBox.Close)
Expand All @@ -276,10 +314,14 @@ def chooseOutputDir(self):
self.formUi.lineEdit_outputDir.setText(dirname)

def run(self):
self.Start = time.time()

dirName = self.formUi.lineEdit_outputDir.text()
#dirName = self.formUi.lineEdit_outputDir.text()
dirName = self.TempDir
print 'run() dir:',dirName

self.OutStr = self.OutStr + '<font color="#0000FF">{0:4.1f}:</font> '.format(time.time() - self.Start) + 'Check dependencies...<br>'
self.formUi.textEdit_Output.setText(self.OutStr)
self.formUi.label_Time.setText('Time: {0:4.1f}: '.format(time.time() - self.Start) )
MeshObject = None
if FemGui.getActiveAnalysis():
for i in FemGui.getActiveAnalysis().Member:
Expand Down Expand Up @@ -320,13 +362,23 @@ def run(self):

QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))

filename = dirName + '/' + MeshObject.Name + '.inp'
self.Basename = self.TempDir + '/' + MeshObject.Name
filename = self.Basename + '.inp'

self.OutStr = self.OutStr + '<font color="#0000FF">{0:4.1f}:</font> '.format(time.time() - self.Start) + self.Basename + '<br>'
self.formUi.textEdit_Output.setText(self.OutStr)

self.OutStr = self.OutStr + '<font color="#0000FF">{0:4.1f}:</font> '.format(time.time() - self.Start) + 'Write mesh...<br>'
self.formUi.textEdit_Output.setText(self.OutStr)

MeshObject.FemMesh.writeABAQUS(filename)
# reopen file with "append" and add the analysis definition
inpfile = open(filename,'a')
inpfile.write('\n\n')

self.OutStr = self.OutStr + '<font color="#0000FF">{0:4.1f}:</font> '.format(time.time() - self.Start) + 'Write loads & Co...<br>'
self.formUi.textEdit_Output.setText(self.OutStr)

# write the fixed node set
NodeSetName = FixedObject.Name
inpfile.write('*NSET,NSET=' + NodeSetName + '\n')
Expand Down Expand Up @@ -391,26 +443,18 @@ def run(self):
inpfile.write('S \n')
inpfile.write('*END STEP \n')

self.OutStr = self.OutStr + '<font color="#0000FF">{0:4.1f}:</font> '.format(time.time() - self.Start) + self.CalculixBinary + '<br>'
self.formUi.textEdit_Output.setText(self.OutStr)

self.OutStr = self.OutStr + '<font color="#0000FF">{0:4.1f}:</font> '.format(time.time() - self.Start) + 'Run Calculix...<br>'
self.formUi.textEdit_Output.setText(self.OutStr)

# run Claculix
self.Calculix.start('C:/Tools/Calculix4Win/c4w/programs/ccx/ccx.exe', ['-i','c:/users/jriegel/appdata/local/temp/Pocket_Mesh'])
print 'run Calclulix at:', self.CalculixBinary , ' with: ', self.Basename
self.Calculix.start(self.CalculixBinary, ['-i',self.Basename])


QApplication.restoreOverrideCursor()
#import subprocess,FreeCADGui
#FreeCADGui.updateGui()
#p1 = subprocess.Popen(['C:/Tools/Calculix4Win/c4w/programs/ccx/ccx.exe', '-i','c:/users/jriegel/appdata/local/temp/Pocket_Mesh'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
#ret = None
#while ret == None:
# print p1.communicate()
# FreeCADGui.updateGui()
# ret = p1.poll()

#ret = subprocess.call(['C:/Tools/Calculix4Win/c4w/programs/ccx/ccx.exe', '-i','c:/users/jriegel/appdata/local/temp/Pocket_Mesh'],shell=True)
#print "Calculix terminated with code:" , str(ret)
#print "Read Result:"
#FreeCADGui.updateGui()

#CalculixLib.importFrd('c:/users/jriegel/appdata/local/temp/Pocket_Mesh.frd',FemGui.getActiveAnalysis() )

class _ResultControlTaskPanel:
'''The control for the displacement post-processing'''
Expand Down
32 changes: 29 additions & 3 deletions src/Mod/Fem/MechanicalAnalysis.ui
View file
Expand Up @@ -7,7 +7,7 @@
<x>0</x>
<y>0</y>
<width>193</width>
<height>354</height>
<height>384</height>
</rect>
</property>
<property name="windowTitle">
Expand All @@ -17,10 +17,20 @@
<item>
<layout class="QHBoxLayout" name="horizontalLayout">
<item>
<widget class="QLineEdit" name="lineEdit_outputDir"/>
<widget class="QLineEdit" name="lineEdit_outputDir">
<property name="enabled">
<bool>false</bool>
</property>
<property name="readOnly">
<bool>true</bool>
</property>
</widget>
</item>
<item>
<widget class="QToolButton" name="toolButton_chooseOutputDir">
<property name="enabled">
<bool>false</bool>
</property>
<property name="text">
<string>...</string>
</property>
Expand All @@ -36,7 +46,23 @@
</widget>
</item>
<item>
<widget class="QTextEdit" name="textEdit_Output"/>
<widget class="QTextEdit" name="textEdit_Output">
<property name="lineWrapMode">
<enum>QTextEdit::NoWrap</enum>
</property>
</widget>
</item>
<item>
<widget class="QLabel" name="label_Time">
<property name="font">
<font>
<pointsize>12</pointsize>
</font>
</property>
<property name="text">
<string>Time:</string>
</property>
</widget>
</item>
</layout>
</widget>
Expand Down

0 comments on commit bb98b5a

Please sign in to comment.