223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

now compatible with new table.data_readArray method

This commit is contained in:
Philip Eisenlohr 2013-12-31 21:06:32 +00:00
parent 79307de6a7
commit 4f30a6e05b
1 changed files with 33 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
processing/pre/geom_fromVoronoiTessellation.py
View File

@ -88,7 +88,7 @@ parser.add_option('-c', '--configuration', dest='config', action='store_true', \
help='output material configuration [%default]') help='output material configuration [%default]')
parser.set_defaults(grid = [0,0,0]) parser.set_defaults(grid = [0,0,0])
parser.set_defaults(size = [0.0,0.0,0.0]) parser.set_defaults(size = [0.0,0.0,0.0])
parser.set_defaults(homogenization = 1) parser.set_defaults(homogenization = 1)
parser.set_defaults(phase = 1) parser.set_defaults(phase = 1)
parser.set_defaults(crystallite = 1) parser.set_defaults(crystallite = 1)
@ -99,18 +99,18 @@ parser.set_defaults(config = False)
#--- setup file handles --------------------------------------------------------------------------- #--- setup file handles ---------------------------------------------------------------------------
files = [] files = []
if filenames == []: if filenames == []:
files.append({'name':'STDIN', files.append({'name': 'STDIN',
'input':sys.stdin, 'input': sys.stdin,
'output':sys.stdout, 'output': sys.stdout,
'croak':sys.stderr, 'croak': sys.stderr,
}) })
else: else:
for name in filenames: for name in filenames:
if os.path.exists(name): if os.path.exists(name):
files.append({'name':name, files.append({'name': name,
'input':open(name), 'input': open(name),
'output':open(name+'_tmp','w'), 'output': open(name+'_tmp','w'),
'croak':sys.stdout, 'croak': sys.stdout,
}) })
@ -121,16 +121,31 @@ for file in files:
theTable = damask.ASCIItable(file['input'],file['output']) theTable = damask.ASCIItable(file['input'],file['output'])
theTable.head_read() theTable.head_read()
coords = theTable.data_asArray(['x','y','z']) labels = ['x','y','z']
if numpy.all(theTable.labels_index(['phi1','Phi','phi2']) != -1): index = 0
eulers = theTable.data_asArray(['phi1','Phi','phi2']) if numpy.all(theTable.labels_index(['phi1','Phi','phi2'])) != -1:
if theTable.labels_index('microstructure') != -1: hasEulers = True
grain = theTable.data_asArray(['microstructure']) labels += ['phi1','Phi','phi2']
grainIDs = numpy.unique(grain).astype('i') index += 3
else: else:
grain = 1+numpy.arange(len(eulers)) hasEulers = False
grainIDs = grain eulerCol = index
if theTable.labels_index('microstructure') != -1:
hasGrains = True
labels += ['microstructure']
index += 1
else:
hasGrains = False
grainCol = index
theTable.data_readArray(labels)
coords = theTable.data[:,0:3]
eulers = theTable.data[:,eulerCol:eulerCol+3] if hasEulers else numpy.zeros(3*len(coords))
grain = theTable.data[:,grainCol] if hasGrains else 1+numpy.arange(len(eulers))
grainIDs = numpy.unique(grain).astype('i')
#--- interpret header ---------------------------------------------------------------------------- #--- interpret header ----------------------------------------------------------------------------
info = { info = {