renamed script to new convention
aligned output behavior to that of fromVoronoiTessellation, i.e., standard output is geom, --config gives associated material.config stud
This commit is contained in:
Philip Eisenlohr
parent
af2aa8500d
commit
025d6c9048
|
|
@ -0,0 +1,127 @@
|
|||
#!/usr/bin/env python
|
||||
|
||||
import os,sys,math,string,numpy
|
||||
from optparse import OptionParser, Option
|
||||
|
||||
# -----------------------------
|
||||
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 [file[s]]', description = """
|
||||
Generate geometry description and material configuration from EBSD data in given square-gridded 'ang' file.
|
||||
Dual phases can be discriminated based on threshold value in a given data column.
|
||||
""" + string.replace('$Id$','\n','\\n')
|
||||
)
|
||||
|
||||
|
||||
parser.add_option('--column', dest='column', type='int', \
|
||||
help='data column to separate phase 1 and 2 [%default]')
|
||||
parser.add_option('-f','--threshold', dest='threshold', type='float', \
|
||||
help='threshold value to discriminate phase 1 from 2')
|
||||
parser.add_option('--homogenization', dest='homogenization', type='int', \
|
||||
help='homogenization index to be used')
|
||||
parser.add_option('--phase', dest='phase', type='int', nargs = 2, \
|
||||
help='two phase indices to be used %default')
|
||||
parser.add_option('--crystallite', dest='crystallite', type='int', \
|
||||
help='crystallite index to be used')
|
||||
parser.add_option('-c', '--configuration', dest='config', action='store_true', \
|
||||
help='output material configuration')
|
||||
|
||||
parser.set_defaults(column = 1)
|
||||
parser.set_defaults(threshold = sys.maxint)
|
||||
parser.set_defaults(homogenization = 1)
|
||||
parser.set_defaults(phase = [1,2])
|
||||
parser.set_defaults(crystallite = 1)
|
||||
parser.set_defaults(config = False)
|
||||
|
||||
(options,filenames) = parser.parse_args()
|
||||
|
||||
# ------------------------------------------ setup file handles ---------------------------------------
|
||||
|
||||
files = []
|
||||
if filenames == []:
|
||||
files.append({'name':'STDIN',
|
||||
'input':sys.stdin,
|
||||
'output':sys.stdout,
|
||||
'croak':sys.stderr,
|
||||
})
|
||||
else:
|
||||
for name in filenames:
|
||||
if os.path.exists(name):
|
||||
files.append({'name':name,
|
||||
'input':open(name),
|
||||
'output':open(name+'_tmp','w'),
|
||||
'croak':sys.stdout,
|
||||
})
|
||||
|
||||
|
||||
# ------------------------------------------ loop over input files ---------------------------------------
|
||||
|
||||
for file in files:
|
||||
if file['name'] != 'STDIN': file['croak'].write(file['name']+'\n')
|
||||
|
||||
point = 0
|
||||
step = [0,0]
|
||||
resolution = [1,1]
|
||||
microstructure = ['<microstructure>']
|
||||
texture = ['<texture>']
|
||||
|
||||
for line in file['input']:
|
||||
words = line.split()
|
||||
if words[0] == '#': # process initial comments block
|
||||
if len(words) > 2:
|
||||
if words[1] == 'HexGrid': file['croak'].write('The file has HexGrid format. Please first convert to SquareGrid...\n'); break
|
||||
if words[1] == 'XSTEP:': step[0] = float(words[2])
|
||||
if words[1] == 'YSTEP:': step[1] = float(words[2])
|
||||
if words[1] == 'NCOLS_ODD:': resolution[0] = int(words[2]); formatwidth = 1+int(math.log10(resolution[0]*resolution[1]))
|
||||
if words[1] == 'NROWS:': resolution[1] = int(words[2]); formatwidth = 1+int(math.log10(resolution[0]*resolution[1]))
|
||||
else: # finished with comments block
|
||||
if options.config: # write configuration (line by line)
|
||||
point += 1
|
||||
me = str(point).rjust(formatwidth)
|
||||
microstructure += ['[Grain%s]\n'%me + \
|
||||
'crystallite\t%i\n'%options.crystallite + \
|
||||
'(constituent)\tphase %i\ttexture %s\tfraction 1.0\n'%(options.phase[{True:0,False:1}[float(words[options.column])<options.threshold]],me)
|
||||
]
|
||||
texture += ['[Grain%s]\n'%me + \
|
||||
'(gauss)\tphi1 %4.2f\tPhi %4.2f\tphi2 %4.2f\tscatter 0.0\tfraction 1.0\n'%tuple(map(lambda x: float(x)*180.0/math.pi, words[:3]))
|
||||
]
|
||||
else:
|
||||
file['output'].write("4 header\n" + \
|
||||
"resolution\ta %i\tb %i\tc 1\n"%(resolution[0],resolution[1]) + \
|
||||
"dimension\tx %g\ty %g\tz %g\n"%(step[0]*resolution[0],step[1]*resolution[1],min(step)) + \
|
||||
"origin\tx 0\ty 0\tz 0\n" + \
|
||||
"homogenization %i\n"%options.homogenization + \
|
||||
"1 to %i\n"%(resolution[0]*resolution[1]))
|
||||
break
|
||||
|
||||
if options.config:
|
||||
file['output'].write('\n'.join(microstructure) + \
|
||||
'\n'.join(texture))
|
||||
|
||||
# ------------------------------------------ output finalization ---------------------------------------
|
||||
|
||||
if file['name'] != 'STDIN':
|
||||
file['output'].close()
|
||||
os.rename(file['name']+'_tmp',os.path.splitext(file['name'])[0] + \
|
||||
{True: '_material.config',
|
||||
False:'.geom'}[options.config])
|
||||
|
|
@ -1,107 +0,0 @@
|
|||
#!/usr/bin/env python
|
||||
|
||||
import os,sys,math,string,numpy
|
||||
from optparse import OptionParser, Option
|
||||
|
||||
# -----------------------------
|
||||
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 [file[s]]', description = """
|
||||
Converts EBSD data from cubic ang files into description for spectral solver (*.geom + material.config)
|
||||
Can discriminate two phases depending on threshold value
|
||||
|
||||
""" + string.replace('$Id$','\n','\\n')
|
||||
)
|
||||
|
||||
|
||||
parser.add_option('-c','--column', dest='column', type='int', \
|
||||
help='data column to separate phase 1 and 2 [%default]')
|
||||
parser.add_option('-t','--threshold', dest='threshold', type='float', \
|
||||
help='threshold used to separate phases. value < threshold: phase = 1')
|
||||
|
||||
parser.set_defaults(column = 1)
|
||||
parser.set_defaults(threshold = sys.maxint)
|
||||
|
||||
(options,filenames) = parser.parse_args()
|
||||
|
||||
# ------------------------------------------ setup file handles ---------------------------------------
|
||||
eulers = numpy.array([0.0,0.0,0.0],'f')
|
||||
geomdim = numpy.array([0.0,0.0,0.0],'f')
|
||||
res = numpy.array([0,0,1],'i')
|
||||
|
||||
files = []
|
||||
if filenames == []:
|
||||
files.append({'name':'STDIN', 'input':sys.stdin, 'material':sys.stdout, 'geom':sys.stdout})
|
||||
else:
|
||||
for name in filenames:
|
||||
if os.path.exists(name):
|
||||
files.append( {'name':name, 'input':open(name),\
|
||||
'material':open(os.path.splitext(name)[0]+'_material.config', 'w'),\
|
||||
'geom':open(os.path.splitext(name)[0]+'.geom', 'w')})
|
||||
|
||||
# ------------------------------------------ loop over input files ---------------------------------------
|
||||
for file in files:
|
||||
point=0
|
||||
if file['name'] != 'STDIN': print file['name']
|
||||
file['material'].write('#---\n<homogenization>\n#---\n'+
|
||||
'\n[SX]\ntype isostrain\nNgrains 1\n\n'+
|
||||
'#---\n<microstructure>\n#---\n\n')
|
||||
tempPart2= '#---\n<texture>\n#---\n\n'
|
||||
for line in file['input']:
|
||||
lineSplit=line.split()
|
||||
|
||||
if line[0]=='#':
|
||||
if len(lineSplit)>2:
|
||||
if line.split()[2]=='HexGrid':
|
||||
print 'The file is a hex grid file. Convert it first to sqr grid'
|
||||
sys.exit()
|
||||
if lineSplit[1]=='XSTEP:': stepSizeX = float(lineSplit[2])
|
||||
if lineSplit[1]=='YSTEP:': stepSizeY = float(lineSplit[2])
|
||||
if lineSplit[1]=='NCOLS_ODD:': res[0] = int(lineSplit[2])
|
||||
if lineSplit[1]=='NROWS:': res[1] = int(lineSplit[2])
|
||||
else:
|
||||
point+=1
|
||||
eulers = (float(lineSplit[0])/2.0/math.pi*360.0, \
|
||||
float(lineSplit[1])/2.0/math.pi*360.0, \
|
||||
float(lineSplit[2])/2.0/math.pi*360.0)
|
||||
|
||||
if float(lineSplit[options.column-1])<options.threshold:
|
||||
phase=1
|
||||
else:
|
||||
phase=2
|
||||
|
||||
file['material'].write(\
|
||||
'[Grain%08d]\ncrystallite 1\n(constituent) phase %1d texture %08d fraction 1.0\n' \
|
||||
%(point,phase,point))
|
||||
tempPart2+=\
|
||||
'[Grain%08d]\n(gauss) phi1 %4.2f Phi %4.2f phi2 %4.2f scatter 0.0 fraction 1.0\n'\
|
||||
%(point,eulers[0],eulers[1],eulers[2])
|
||||
geomdim[0] = stepSizeX*res[0]
|
||||
geomdim[1] = stepSizeY*res[1]
|
||||
geomdim[2] = min(stepSizeX,stepSizeY)
|
||||
|
||||
file['material'].write(tempPart2)
|
||||
file['geom'].write(
|
||||
'3 header\nresolution a %4d b %4d c %1d \ndimension x %5.3f y %5.3f z %5.3f\nhomogenization 1\n'\
|
||||
%(res[0],res[1],res[2],geomdim[0],geomdim[1],geomdim[2]))
|
||||
file['geom'].write('1 to %d\n'%(res[0]*res[1]))
|
||||
Loading…
Reference in New Issue