new style and test for grain growth
This commit is contained in:
Martin Diehl
parent
c91788d721
commit
41f1994569
|
|
@ -13,18 +13,6 @@ scriptName = os.path.splitext(scriptID.split()[1])[0]
|
||||||
#--------------------------------------------------------------------------------------------------
|
#--------------------------------------------------------------------------------------------------
|
||||||
# MAIN
|
# MAIN
|
||||||
#--------------------------------------------------------------------------------------------------
|
#--------------------------------------------------------------------------------------------------
|
||||||
identifiers = {
|
|
||||||
'grid': ['a','b','c'],
|
|
||||||
'size': ['x','y','z'],
|
|
||||||
'origin': ['x','y','z'],
|
|
||||||
}
|
|
||||||
mappings = {
|
|
||||||
'grid': lambda x: int(x),
|
|
||||||
'size': lambda x: float(x),
|
|
||||||
'origin': lambda x: float(x),
|
|
||||||
'homogenization': lambda x: int(x),
|
|
||||||
'microstructures': lambda x: int(x),
|
|
||||||
}
|
|
||||||
|
|
||||||
parser = OptionParser(option_class=damask.extendableOption, usage='%prog options [file[s]]', description = """
|
parser = OptionParser(option_class=damask.extendableOption, usage='%prog options [file[s]]', description = """
|
||||||
Smoothens out interface roughness by simulated curvature flow.
|
Smoothens out interface roughness by simulated curvature flow.
|
||||||
|
|
@ -52,89 +40,39 @@ parser.set_defaults(immutable = [])
|
||||||
|
|
||||||
options.immutable = map(int,options.immutable)
|
options.immutable = map(int,options.immutable)
|
||||||
|
|
||||||
#--- setup file handles --------------------------------------------------------------------------
|
# --- loop over input files -------------------------------------------------------------------------
|
||||||
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 ------------------------------------------------------------------------
|
if filenames == []: filenames = [None]
|
||||||
for file in files:
|
|
||||||
if file['name'] != 'STDIN': file['croak'].write('\033[1m'+scriptName+'\033[0m: '+file['name']+'\n')
|
|
||||||
else: file['croak'].write('\033[1m'+scriptName+'\033[0m\n')
|
|
||||||
|
|
||||||
theTable = damask.ASCIItable(file['input'],file['output'],labeled = False,buffered = False)
|
for name in filenames:
|
||||||
theTable.head_read()
|
try:
|
||||||
|
table = damask.ASCIItable(name = name,
|
||||||
|
buffered = False, labeled = False)
|
||||||
|
except: continue
|
||||||
|
damask.util.report(scriptName,name)
|
||||||
|
|
||||||
#--- interpret header ----------------------------------------------------------------------------
|
# --- interpret header ----------------------------------------------------------------------------
|
||||||
info = {
|
|
||||||
'grid': np.zeros(3,'i'),
|
|
||||||
'size': np.zeros(3,'d'),
|
|
||||||
'origin': np.zeros(3,'d'),
|
|
||||||
'homogenization': 0,
|
|
||||||
'microstructures': 0,
|
|
||||||
}
|
|
||||||
newInfo = {
|
|
||||||
'microstructures': 0,
|
|
||||||
}
|
|
||||||
extra_header = []
|
|
||||||
|
|
||||||
for header in theTable.info:
|
table.head_read()
|
||||||
headitems = map(str.lower,header.split())
|
info,extra_header = table.head_getGeom()
|
||||||
if len(headitems) == 0: continue
|
|
||||||
if headitems[0] in mappings.keys():
|
|
||||||
if headitems[0] in identifiers.keys():
|
|
||||||
for i in xrange(len(identifiers[headitems[0]])):
|
|
||||||
info[headitems[0]][i] = \
|
|
||||||
mappings[headitems[0]](headitems[headitems.index(identifiers[headitems[0]][i])+1])
|
|
||||||
else:
|
|
||||||
info[headitems[0]] = mappings[headitems[0]](headitems[1])
|
|
||||||
else:
|
|
||||||
extra_header.append(header)
|
|
||||||
|
|
||||||
file['croak'].write('grid a b c: %s\n'%(' x '.join(map(str,info['grid']))) + \
|
damask.util.croak(['grid a b c: %s'%(' x '.join(map(str,info['grid']))),
|
||||||
'size x y z: %s\n'%(' x '.join(map(str,info['size']))) + \
|
'size x y z: %s'%(' x '.join(map(str,info['size']))),
|
||||||
'origin x y z: %s\n'%(' : '.join(map(str,info['origin']))) + \
|
'origin x y z: %s'%(' : '.join(map(str,info['origin']))),
|
||||||
'homogenization: %i\n'%info['homogenization'] + \
|
'homogenization: %i'%info['homogenization'],
|
||||||
'microstructures: %i\n'%info['microstructures'])
|
'microstructures: %i'%info['microstructures'],
|
||||||
|
])
|
||||||
|
|
||||||
if np.any(info['grid'] < 1):
|
errors = []
|
||||||
file['croak'].write('invalid grid a b c.\n')
|
if np.any(info['grid'] < 1): errors.append('invalid grid a b c.')
|
||||||
continue
|
if np.any(info['size'] <= 0.0): errors.append('invalid size x y z.')
|
||||||
if np.any(info['size'] <= 0.0):
|
if errors != []:
|
||||||
file['croak'].write('invalid size x y z.\n')
|
damask.util.croak(errors)
|
||||||
|
table.close(dismiss = True)
|
||||||
continue
|
continue
|
||||||
|
|
||||||
#--- read data ------------------------------------------------------------------------------------
|
# --- read data ------------------------------------------------------------------------------------
|
||||||
microstructure = np.zeros(np.prod(info['grid']),'i') # 2D structures do not work
|
microstructure = np.tile(np.array(table.microstructure_read(info['grid']),'i').reshape(info['grid'],order='F'),
|
||||||
i = 0
|
|
||||||
|
|
||||||
while theTable.data_read(): # read next data line of ASCII table
|
|
||||||
items = theTable.data
|
|
||||||
if len(items) > 2:
|
|
||||||
if items[1].lower() == 'of': items = [int(items[2])]*int(items[0])
|
|
||||||
elif items[1].lower() == 'to': items = xrange(int(items[0]),1+int(items[2]))
|
|
||||||
else: items = map(int,items)
|
|
||||||
else: items = map(int,items)
|
|
||||||
|
|
||||||
s = len(items)
|
|
||||||
microstructure[i:i+s] = items
|
|
||||||
i += s
|
|
||||||
|
|
||||||
#--- reshape, if 2D make copy ---------------------------------------------------------------------
|
|
||||||
|
|
||||||
microstructure = np.tile(microstructure.reshape(info['grid'],order='F'),
|
|
||||||
np.where(info['grid'] == 1, 2,1)) # make one copy along dimensions with grid == 1
|
np.where(info['grid'] == 1, 2,1)) # make one copy along dimensions with grid == 1
|
||||||
grid = np.array(microstructure.shape)
|
grid = np.array(microstructure.shape)
|
||||||
|
|
||||||
|
|
@ -210,33 +148,35 @@ for file in files:
|
||||||
newID += 1
|
newID += 1
|
||||||
microstructure = np.where(microstructure == microstructureID, newID, microstructure)
|
microstructure = np.where(microstructure == microstructureID, newID, microstructure)
|
||||||
|
|
||||||
# --- assemble header -----------------------------------------------------------------------------
|
newInfo = {'microstructures': 0,}
|
||||||
newInfo['microstructures'] = microstructure[0:info['grid'][0],0:info['grid'][1],0:info['grid'][2]].max()
|
newInfo['microstructures'] = microstructure.max()
|
||||||
|
|
||||||
#--- report ---------------------------------------------------------------------------------------
|
# --- report ---------------------------------------------------------------------------------------
|
||||||
if (newInfo['microstructures'] != info['microstructures']):
|
|
||||||
file['croak'].write('--> microstructures: %i\n'%newInfo['microstructures'])
|
|
||||||
|
|
||||||
#--- write header ---------------------------------------------------------------------------------
|
remarks = []
|
||||||
theTable.labels_clear()
|
if (newInfo['microstructures'] != info['microstructures']): remarks.append('--> microstructures: %i'%newInfo['microstructures'])
|
||||||
theTable.info_clear()
|
if remarks != []: damask.util.croak(remarks)
|
||||||
theTable.info_append(extra_header+[
|
|
||||||
scriptID+ ' ' + ' '.join(sys.argv[1:]),
|
# --- write header ---------------------------------------------------------------------------------
|
||||||
"grid\ta %i\tb %i\tc %i"%(info['grid'][0],info['grid'][1],info['grid'][2],),
|
|
||||||
"size\tx %f\ty %f\tz %f"%(info['size'][0],info['size'][1],info['size'][2],),
|
table.labels_clear()
|
||||||
"origin\tx %f\ty %f\tz %f"%(info['origin'][0],info['origin'][1],info['origin'][2],),
|
table.info_clear()
|
||||||
"homogenization\t%i"%info['homogenization'],
|
table.info_append(extra_header+[
|
||||||
"microstructures\t%i"%(newInfo['microstructures']),
|
scriptID + ' ' + ' '.join(sys.argv[1:]),
|
||||||
|
"grid\ta {grid[0]}\tb {grid[1]}\tc {grid[2]}".format(grid=info['grid']),
|
||||||
|
"size\tx {size[0]}\ty {size[1]}\tz {size[2]}".format(size=info['size']),
|
||||||
|
"origin\tx {origin[0]}\ty {origin[1]}\tz {origin[2]}".format(origin=info['origin']),
|
||||||
|
"homogenization\t{homog}".format(homog=info['homogenization']),
|
||||||
|
"microstructures\t{microstructures}".format(microstructures=newInfo['microstructures']),
|
||||||
])
|
])
|
||||||
theTable.head_write()
|
table.head_write()
|
||||||
|
|
||||||
# --- write microstructure information ------------------------------------------------------------
|
# --- write microstructure information ------------------------------------------------------------
|
||||||
formatwidth = int(math.floor(math.log10(microstructure.max())+1))
|
|
||||||
theTable.data = microstructure[0:info['grid'][0],0:info['grid'][1],0:info['grid'][2]].reshape(np.prod(info['grid']),order='F').transpose() # question PE: this assumes that only the Z dimension can be 1!
|
|
||||||
theTable.data_writeArray('%%%ii'%(formatwidth),delimiter=' ')
|
|
||||||
|
|
||||||
#--- output finalization --------------------------------------------------------------------------
|
formatwidth = int(math.floor(math.log10(microstructure.max())+1))
|
||||||
if file['name'] != 'STDIN':
|
table.data = microstructure.reshape((info['grid'][0],info['grid'][1]*info['grid'][2]),order='F').transpose()
|
||||||
theTable.input_close()
|
table.data_writeArray('%%%ii'%(formatwidth),delimiter = ' ')
|
||||||
theTable.output_close()
|
|
||||||
os.rename(file['name']+'_tmp',file['name'])
|
# --- output finalization --------------------------------------------------------------------------
|
||||||
|
|
||||||
|
table.close()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue