2010-10-26 20:32:20 +05:30
|
|
|
#!/usr/bin/env python
|
|
|
|
|
2011-11-09 23:34:51 +05:30
|
|
|
# Makes postprocessing routines acessible from everywhere.
|
2010-10-26 20:32:20 +05:30
|
|
|
|
2012-01-04 16:36:24 +05:30
|
|
|
import os,sys,glob,string
|
|
|
|
from damask import Environment
|
2011-12-06 22:28:17 +05:30
|
|
|
from optparse import OptionParser, Option
|
2010-10-26 20:32:20 +05:30
|
|
|
|
2011-11-09 21:50:52 +05:30
|
|
|
|
2011-12-06 22:28:17 +05:30
|
|
|
# -----------------------------
|
|
|
|
class extendableOption(Option):
|
|
|
|
# -----------------------------
|
|
|
|
# used for definition of new option parser action 'extend', which enables to take multiple option arguments
|
|
|
|
# taken from online tutorial http://docs.python.org/library/optparse.html
|
|
|
|
|
|
|
|
ACTIONS = Option.ACTIONS + ("extend",)
|
|
|
|
STORE_ACTIONS = Option.STORE_ACTIONS + ("extend",)
|
|
|
|
TYPED_ACTIONS = Option.TYPED_ACTIONS + ("extend",)
|
|
|
|
ALWAYS_TYPED_ACTIONS = Option.ALWAYS_TYPED_ACTIONS + ("extend",)
|
|
|
|
|
|
|
|
def take_action(self, action, dest, opt, value, values, parser):
|
|
|
|
if action == "extend":
|
|
|
|
lvalue = value.split(",")
|
|
|
|
values.ensure_value(dest, []).extend(lvalue)
|
|
|
|
else:
|
|
|
|
Option.take_action(self, action, dest, opt, value, values, parser)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
########################################################
|
|
|
|
# MAIN
|
|
|
|
########################################################
|
|
|
|
|
|
|
|
parser = OptionParser(option_class=extendableOption, usage='%prog options', description = """
|
|
|
|
Sets up the pre and post processing tools of DAMASK
|
|
|
|
|
2012-01-12 01:48:47 +05:30
|
|
|
""" + string.replace('$Id$','\n','\\n')
|
2011-12-06 22:28:17 +05:30
|
|
|
)
|
|
|
|
|
2011-12-22 16:06:59 +05:30
|
|
|
compilers = ['intel','ifort','intel32','gfortran','gnu95']
|
|
|
|
|
2012-01-20 02:11:08 +05:30
|
|
|
parser.add_option('--F90', '--f90', dest='compiler', type='string', \
|
2011-12-22 16:06:59 +05:30
|
|
|
help='name of F90 compiler')
|
2011-12-06 22:28:17 +05:30
|
|
|
|
|
|
|
(options,filenames) = parser.parse_args()
|
|
|
|
|
2011-12-22 16:06:59 +05:30
|
|
|
if options.compiler not in compilers:
|
2012-01-04 16:36:24 +05:30
|
|
|
parser.error('compiler switch "--F90" has to be one out of: %s'%(', '.join(compilers)))
|
2011-12-07 00:08:03 +05:30
|
|
|
|
2011-12-22 16:06:59 +05:30
|
|
|
f2py_compiler = {
|
2012-02-09 19:00:59 +05:30
|
|
|
'gfortran': 'gnu95 --f90flags="-fno-range-check -xf95-cpp-input -std=f2008"',
|
|
|
|
'gnu95': 'gnu95 --f90flags="-fno-range-check -xf95-cpp-input -std=f2008"',
|
2012-03-30 01:24:31 +05:30
|
|
|
'intel32': 'intel --f90flags="-fpp -stand f03 -diag-disable 5268"',
|
|
|
|
'intel': 'intelem --f90flags="-fpp -stand f03 -diag-disable 5268"',
|
|
|
|
'ifort': 'intelem --f90flags="-fpp -stand f03 -diag-disable 5268"',
|
2011-12-22 16:06:59 +05:30
|
|
|
}[options.compiler]
|
2012-01-06 15:30:22 +05:30
|
|
|
compiler = {
|
|
|
|
'gfortran': 'gfortran',
|
|
|
|
'gnu95': 'gfortran',
|
|
|
|
'intel32': 'ifort',
|
|
|
|
'intel': 'ifort',
|
|
|
|
'ifort': 'ifort',
|
|
|
|
}[options.compiler]
|
2011-12-06 22:28:17 +05:30
|
|
|
|
2012-01-04 16:36:24 +05:30
|
|
|
damaskEnv = Environment()
|
2011-12-15 14:23:18 +05:30
|
|
|
baseDir = damaskEnv.relPath('processing/')
|
|
|
|
codeDir = damaskEnv.relPath('code/')
|
2011-12-06 22:28:17 +05:30
|
|
|
|
2011-12-22 16:06:59 +05:30
|
|
|
if 'ikml' in damaskEnv.pathInfo and damaskEnv.pathInfo['ikml'] != '':
|
|
|
|
lib_lapack = '' # TODO!!
|
|
|
|
elif 'acml' in damaskEnv.pathInfo and damaskEnv.pathInfo['acml'] != '':
|
2012-01-06 16:11:01 +05:30
|
|
|
lib_lapack = '-L%s/%s64/lib -lacml'%(os.path.join(damaskEnv.pathInfo['acml']),compiler) # can we use linker flag?
|
2011-12-22 16:06:59 +05:30
|
|
|
elif 'lapack' in damaskEnv.pathInfo and damaskEnv.pathInfo['lapack'] != '':
|
|
|
|
lib_lapack = '-L%s -llapack'%(damaskEnv.pathInfo['lapack']) # see http://cens.ioc.ee/pipermail/f2py-users/2003-December/000621.html
|
|
|
|
|
2011-12-06 22:28:17 +05:30
|
|
|
#define ToDo list
|
2011-01-26 20:47:03 +05:30
|
|
|
bin_link = { \
|
2011-11-09 23:21:00 +05:30
|
|
|
'pre' : [
|
|
|
|
'marc_addUserOutput.py',
|
|
|
|
'mentat_pbcOnBoxMesh.py',
|
|
|
|
'mentat_spectralBox.py',
|
|
|
|
'patchFromReconstructedBoundaries.py',
|
|
|
|
'spectral_geomCheck.py',
|
2011-12-20 19:03:16 +05:30
|
|
|
'spectral_geomScale.py',
|
|
|
|
'spectral_geomCrop.py',
|
2011-11-09 23:21:00 +05:30
|
|
|
'spectral_minimalSurface.py',
|
|
|
|
'spectral_vicinityOffset.py',
|
2012-03-30 01:24:31 +05:30
|
|
|
'spectral_ang2geom.py',
|
|
|
|
'spectral_randomSeeding.py',
|
2012-04-18 17:12:57 +05:30
|
|
|
'spectral_layeredGeom.py',
|
2012-04-23 18:16:38 +05:30
|
|
|
'spectral_geomPack.py',
|
2011-12-06 22:28:17 +05:30
|
|
|
'voronoi_tessellation.exe',
|
2012-03-30 01:24:31 +05:30
|
|
|
'OIMang_hex2cub',
|
2011-11-09 23:21:00 +05:30
|
|
|
],
|
|
|
|
'post' : [
|
|
|
|
'3Dvisualize.py',
|
|
|
|
'addCauchy.py',
|
2011-12-02 20:45:36 +05:30
|
|
|
'addCalculation.py',
|
2011-11-09 23:21:00 +05:30
|
|
|
'addDeterminant.py',
|
|
|
|
'addDivergence.py',
|
2011-12-21 22:55:31 +05:30
|
|
|
'addCurl.py',
|
2011-11-09 23:21:00 +05:30
|
|
|
'addMises.py',
|
|
|
|
'addNorm.py',
|
|
|
|
'addStrainTensors.py',
|
|
|
|
'addCompatibilityMismatch.py',
|
2012-04-18 17:12:57 +05:30
|
|
|
'addDeformedConfiguration.py',
|
2011-11-09 23:21:00 +05:30
|
|
|
'averageDown.py',
|
2012-02-14 17:34:37 +05:30
|
|
|
'deleteColumn.py',
|
2012-02-23 19:26:50 +05:30
|
|
|
'deleteInfo.py',
|
2012-02-16 14:06:35 +05:30
|
|
|
'filterTable.py',
|
2011-11-09 23:21:00 +05:30
|
|
|
'mentat_colorMap.py',
|
|
|
|
'postResults.py',
|
|
|
|
'spectral_iterationCount.py',
|
2012-02-16 20:45:43 +05:30
|
|
|
'spectral_parseLog.py',
|
2012-04-24 20:39:20 +05:30
|
|
|
'spectral_buildElements.py',
|
2012-02-24 14:23:19 +05:30
|
|
|
'tagLabel.py',
|
2011-11-09 23:21:00 +05:30
|
|
|
],
|
|
|
|
}
|
2010-10-26 20:32:20 +05:30
|
|
|
|
2011-01-26 20:47:03 +05:30
|
|
|
compile = { \
|
2011-11-09 23:21:00 +05:30
|
|
|
'pre' : [
|
|
|
|
'voronoi_tessellation.f90',
|
|
|
|
],
|
|
|
|
'post' : [
|
|
|
|
],
|
|
|
|
}
|
2011-12-06 22:28:17 +05:30
|
|
|
|
2011-01-26 20:47:03 +05:30
|
|
|
|
2011-03-11 14:50:37 +05:30
|
|
|
execute = { \
|
2011-12-06 22:28:17 +05:30
|
|
|
'postMath' : [
|
2011-12-22 16:06:59 +05:30
|
|
|
'make tidy',
|
2011-12-06 22:28:17 +05:30
|
|
|
# The following command is used to compile math.f90 and make the functions defined in DAMASK_math.pyf
|
2011-12-07 00:08:03 +05:30
|
|
|
# available for python in the module DAMASK_math.so
|
2011-12-06 22:28:17 +05:30
|
|
|
# It uses the fortran wrapper f2py that is included in the numpy package to construct the
|
|
|
|
# module postprocessingMath.so out of the fortran code postprocessingMath.f90
|
|
|
|
# for the generation of the pyf file:
|
2012-03-30 01:24:31 +05:30
|
|
|
# f2py -m DAMASK -h DAMASK.pyf --overwrite-signature ../../code/math.f90 \
|
2011-12-22 16:06:59 +05:30
|
|
|
'f2py %s'%(os.path.join(codeDir,'damask.core.pyf')) +\
|
|
|
|
' -c --fcompiler=%s'%(f2py_compiler) +\
|
2012-03-30 01:24:31 +05:30
|
|
|
' %s'%(os.path.join(codeDir,'prec.f90'))+\
|
2012-04-16 20:31:29 +05:30
|
|
|
' %s'%(os.path.join(codeDir,'DAMASK_spectral_interface.f90'))+\
|
2012-03-30 01:24:31 +05:30
|
|
|
' %s'%(os.path.join(codeDir,'IO.f90'))+\
|
|
|
|
' %s'%(os.path.join(codeDir,'numerics.f90'))+\
|
|
|
|
' %s'%(os.path.join(codeDir,'debug.f90'))+\
|
2011-12-22 16:06:59 +05:30
|
|
|
' %s'%(os.path.join(codeDir,'math.f90'))+\
|
2012-04-10 20:45:46 +05:30
|
|
|
' %s'%(os.path.join(codeDir,'FEsolving.f90'))+\
|
|
|
|
' %s'%(os.path.join(codeDir,'mesh.f90'))+\
|
2012-04-16 20:31:29 +05:30
|
|
|
' %s'%(os.path.join(codeDir,'spectral_quit.f90'))+\
|
2012-01-06 16:11:01 +05:30
|
|
|
' -L%s/lib -lfftw3'%(damaskEnv.pathInfo['fftw'])+\
|
2011-12-22 16:06:59 +05:30
|
|
|
' %s'%lib_lapack,
|
2012-01-20 02:11:08 +05:30
|
|
|
'mv %s `readlink -f %s`' %(os.path.join(codeDir,'core.so'),os.path.join(damaskEnv.relPath('lib/damask'),'core.so')),
|
2011-12-06 22:28:17 +05:30
|
|
|
]
|
2011-11-09 23:21:00 +05:30
|
|
|
}
|
2011-03-11 14:50:37 +05:30
|
|
|
|
2011-11-09 21:07:45 +05:30
|
|
|
|
2011-12-06 22:28:17 +05:30
|
|
|
for dir in compile:
|
|
|
|
for file in compile[dir]:
|
|
|
|
src = os.path.abspath(os.path.join(baseDir,dir,file))
|
|
|
|
if os.path.isfile(src):
|
|
|
|
try:
|
|
|
|
os.system('rm %s.exe'%(os.path.splitext(src)[0]))
|
|
|
|
print 'removing %s.exe '%(os.path.splitext(src)[0])
|
|
|
|
except:
|
|
|
|
pass
|
2012-01-06 15:30:22 +05:30
|
|
|
print 'compiling ',src,'using',compiler
|
2012-01-12 19:16:35 +05:30
|
|
|
os.system('%s -O2 -o %s.exe %s'%(compiler,os.path.splitext(src)[0],src))
|
2011-12-06 22:28:17 +05:30
|
|
|
|
|
|
|
os.chdir(codeDir) # needed for compilation with gfortran and f2py
|
|
|
|
for tasks in execute:
|
|
|
|
for cmd in execute[tasks]:
|
2011-12-22 16:06:59 +05:30
|
|
|
try:
|
|
|
|
print 'executing...:',cmd
|
|
|
|
os.system(cmd)
|
|
|
|
except:
|
|
|
|
print 'failed..!'
|
|
|
|
pass
|
2011-12-06 22:28:17 +05:30
|
|
|
|
2011-12-22 16:06:59 +05:30
|
|
|
os.chdir(damaskEnv.relPath('processing/setup/'))
|
2011-12-06 22:28:17 +05:30
|
|
|
modules = glob.glob('*.mod')
|
|
|
|
for module in modules:
|
|
|
|
print 'removing', module
|
|
|
|
os.remove(module)
|
|
|
|
|
2011-01-26 20:47:03 +05:30
|
|
|
for dir in bin_link:
|
|
|
|
for file in bin_link[dir]:
|
2011-12-06 22:28:17 +05:30
|
|
|
src = os.path.abspath(os.path.join(baseDir,dir,file))
|
2010-10-26 20:32:20 +05:30
|
|
|
if (file == ''):
|
2011-12-15 14:23:18 +05:30
|
|
|
sym_link = os.path.abspath(os.path.join(damaskEnv.binDir(),dir))
|
2010-10-26 20:32:20 +05:30
|
|
|
else:
|
2011-12-15 14:23:18 +05:30
|
|
|
sym_link = os.path.abspath(os.path.join(damaskEnv.binDir(),os.path.splitext(file)[0]))
|
2011-11-09 21:50:52 +05:30
|
|
|
print sym_link,'-->',src
|
|
|
|
if os.path.lexists(sym_link):
|
|
|
|
os.remove(sym_link)
|
|
|
|
os.symlink(src,sym_link)
|
|
|
|
|