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

condensed info parsing. 

added info to report new origin of grid (based on --offset).
This commit is contained in:
Philip Eisenlohr 2012-06-26 18:08:50 +00:00
parent fd3d592333
commit 783e1b50e2
1 changed files with 49 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
processing/pre/spectral_geomCanvas.py
View File

@ -29,10 +29,13 @@ class extendedOption(Option):
identifiers = { identifiers = {
'resolution': ['a','b','c'], 'resolution': ['a','b','c'],
'dimension': ['x','y','z'], 'dimension': ['x','y','z'],
'origin': ['x','y','z'],
} }
mappings = { mappings = {
'resolution': lambda x: int(x), 'resolution': lambda x: int(x),
'dimension': lambda x: float(x), 'dimension': lambda x: float(x),
'origin': lambda x: float(x),
'homogenization': lambda x: int(x),
} }
@ -90,49 +93,66 @@ for file in files:
headitems = header.split() headitems = header.split()
if headitems[0] == 'resolution': # located resolution entry if headitems[0] == 'resolution': # located resolution entry
for i in xrange(3): for i in xrange(3):
options.resolution[i] = mappings['resolution'](headitems[headitems.index(identifiers['resolution'][i])+1]) options.resolution[i] = \
mappings['resolution'](headitems[headitems.index(identifiers['resolution'][i])+1])
info = {'resolution': [0,0,0],
'dimension': [0.0,0.0,0.0],
'origin': [0.0,0.0,0.0],
'homogenization': 1,
}
resolution = [0,0,0]
dimension = [0.0,0.0,0.0]
new_header = [] new_header = []
for header in headers: for header in headers:
headitems = header.split() headitems = map(str.lower,header.split())
if headitems[0] == 'resolution': # located resolution entry if headitems[0] in mappings.keys():
for i in xrange(3): if headitems[0] in identifiers.keys():
resolution[i] = mappings['resolution'](headitems[headitems.index(identifiers['resolution'][i])+1]) for i in xrange(len(identifiers[headitems[0]])):
header = "resolution\ta %i\tb %i\tc %i\n"%(options.resolution[0],options.resolution[1],options.resolution[2],) info[headitems[0]][i] = \
if headitems[0] == 'dimension': # located dimension entry mappings[headitems[0]](headitems[headitems.index(identifiers[headitems[0]][i])+1])
for i in xrange(3): else:
dimension[i] = mappings['dimension'](headitems[headitems.index(identifiers['dimension'][i])+1]) info[headitems[0]] = mappings[headitems[0]](headitems[1])
header = "dimension\tx %f\ty %f\tz %f\n"%(dimension[0]/resolution[0]*options.resolution[0],
dimension[1]/resolution[1]*options.resolution[1], if info['resolution'] == [0,0,0]:
dimension[2]/resolution[2]*options.resolution[2],)
new_header.append(header)
if resolution == [0,0,0]:
print 'no resolution info found.' print 'no resolution info found.'
sys.exit(1) sys.exit(1)
if dimension == [0.0,0.0,0.0]: if info['dimension'] == [0.0,0.0,0.0]:
print 'no dimension info found.' print 'no dimension info found.'
sys.exit(1) sys.exit(1)
if file['name'] != 'STDIN': if file['name'] != 'STDIN':
print 'resolution: %s'%(' x '.join(map(str,resolution))) print 'resolution: %s'%(' x '.join(map(str,info['resolution'])))
print 'dimension: %s'%(' x '.join(map(str,dimension))) print 'dimension: %s'%(' x '.join(map(str,info['dimension'])))
print 'origin: %s'%(' x '.join(map(str,info['origin'])))
microstructure = numpy.zeros(resolution,'i')
new_header.append("resolution\ta %i\tb %i\tc %i\n"%(
options.resolution[0],
options.resolution[1],
options.resolution[2],))
new_header.append("dimension\tx %f\ty %f\tz %f\n"%(
info['dimension'][0]/info['resolution'][0]*options.resolution[0],
info['dimension'][1]/info['resolution'][1]*options.resolution[1],
info['dimension'][2]/info['resolution'][2]*options.resolution[2],))
new_header.append("origin\tx %f\ty %f\tz %f\n"%(
info['origin'][0]+info['dimension'][0]/info['resolution'][0]*options.offset[0],
info['origin'][1]+info['dimension'][1]/info['resolution'][1]*options.offset[1],
info['origin'][2]+info['dimension'][2]/info['resolution'][2]*options.offset[2],))
new_header.append("homogenization\t%i\n"%info['homogenization'])
microstructure = numpy.zeros(info['resolution'],'i')
i = 0 i = 0
for line in content: for line in content:
for item in map(int,line.split()): for item in map(int,line.split()):
microstructure[i%resolution[0],(i/resolution[0])%resolution[1],i/resolution[0]/resolution[1]] = item microstructure[i%info['resolution'][0],
(i/info['resolution'][0])%info['resolution'][1],
i/info['resolution'][0]/info['resolution'][1]] = item
i += 1 i += 1
microstructure_cropped = numpy.zeros(options.resolution,'i') microstructure_cropped = numpy.zeros(options.resolution,'i')
microstructure_cropped.fill({True:options.fill,False:microstructure.max()+1}[options.fill>0]) microstructure_cropped.fill({True:options.fill,False:microstructure.max()+1}[options.fill>0])
xindex = list(set(xrange(options.offset[0],options.offset[0]+options.resolution[0])) & set(xrange(resolution[0]))) xindex = list(set(xrange(options.offset[0],options.offset[0]+options.resolution[0])) & set(xrange(info['resolution'][0])))
yindex = list(set(xrange(options.offset[1],options.offset[1]+options.resolution[1])) & set(xrange(resolution[1]))) yindex = list(set(xrange(options.offset[1],options.offset[1]+options.resolution[1])) & set(xrange(info['resolution'][1])))
zindex = list(set(xrange(options.offset[2],options.offset[2]+options.resolution[2])) & set(xrange(resolution[2]))) zindex = list(set(xrange(options.offset[2],options.offset[2]+options.resolution[2])) & set(xrange(info['resolution'][2])))
translate_x = [i - options.offset[0] for i in xindex] translate_x = [i - options.offset[0] for i in xindex]
translate_y = [i - options.offset[1] for i in yindex] translate_y = [i - options.offset[1] for i in yindex]
translate_z = [i - options.offset[2] for i in zindex] translate_z = [i - options.offset[2] for i in zindex]
@ -142,7 +162,8 @@ for file in files:
# ------------------------------------------ assemble header --------------------------------------- # ------------------------------------------ assemble header ---------------------------------------
output = ''.join(new_header) output = '%i\theader\n'%(len(new_header))
output += ''.join(new_header)
# ------------------------------------- regenerate texture information ---------------------------------- # ------------------------------------- regenerate texture information ----------------------------------