DAMASK_EICMD/processing/pre/mentat_spectralBox

321 lines
7.3 KiB
Python
Executable File

#!/usr/bin/env python
import os, sys, math, re, threading, time
from optparse import OptionParser, OptionGroup, Option, SUPPRESS_HELP
releases = {'2010':['linux64',''],
'2008r1':[''],
'2007r1':[''],
'2005r3':[''],
}
# ----------------------- FUNCTIONS ----------------------------
def outMentat(cmd,locals):
if cmd[0:3] == '(!)':
exec(cmd[3:])
elif cmd[0:3] == '(?)':
cmd = eval(cmd[3:])
py_send(cmd)
else:
py_send(cmd)
return
def outStdout(cmd,locals):
if cmd[0:3] == '(!)':
exec(cmd[3:])
elif cmd[0:3] == '(?)':
cmd = eval(cmd[3:])
print cmd
else:
print cmd
return
def output(cmds,locals,dest):
for cmd in cmds:
if isinstance(cmd,list):
output(cmd,locals,dest)
else:
{\
'Mentat': outMentat,\
'Stdout': outStdout,\
}[dest](str(cmd),locals)
return
#--------------------
def init():
#--------------------
return ["*new_model yes",
"*reset",
"*select_clear",
"*set_element_class hex8",
"*set_nodes off",
"*elements_solid",
"*show_view 4",
"*reset_view",
"*view_perspective",
"*redraw",
]
#--------------------
def mesh(r,d):
#--------------------
return [
"*add_nodes",
"%f %f %f"%(0.0,0.0,0.0),
"%f %f %f"%(0.0,0.0,d[2]),
"%f %f %f"%(0.0,d[1],d[2]),
"%f %f %f"%(0.0,d[1],0.0),
"%f %f %f"%(-d[0],0.0,0.0),
"%f %f %f"%(-d[0],0.0,d[2]),
"%f %f %f"%(-d[0],d[1],d[2]),
"%f %f %f"%(-d[0],d[1],0.0),
"*add_elements",
range(1,9),
"*sub_divisions",
"%i %i %i"%(r[2],r[1],r[0]),
"*subdivide_elements",
"all_existing",
"*set_sweep_tolerance",
"%f"%(float(min(d))/max(r)/2.0),
"*sweep_all",
"*renumber_all",
"*set_move_scale_factor x -1",
"*move_elements",
"all_existing",
"*flip_elements",
"all_existing",
"*fill_view",
]
#--------------------
def material():
#--------------------
cmds = [\
"*new_mater standard",
"*mater_option general:state:solid",
"*mater_option structural:type:hypo_elast",
"*mater_name",
"hypela2",
"*add_mater_elements",
"all_existing",
"*geometry_type mech_three_solid",
"*geometry_option red_integ_capacity:on",
"*add_geometry_elements",
"all_existing",
]
return cmds
#--------------------
def geometry():
#--------------------
cmds = [\
"*geometry_type mech_three_solid",
"*geometry_option red_integ_capacity:on",
"*add_geometry_elements",
"all_existing",
"*element_type 117",
"all_existing",
]
return cmds
#--------------------
def initial_conditions(homogenization,grains):
#--------------------
elements = []
element = 0
for id in grains:
element += 1
if len(elements) < id:
for i in range(id-len(elements)):
elements.append([])
elements[id-1].append(element)
cmds = [\
"*new_icond",
"*icond_name _temperature",
"*icond_type state_variable",
"*icond_param_value state_var_id 1",
"*icond_dof_value var 300",
"*add_icond_elements",
"all_existing",
"*new_icond",
"*icond_name _homogenization",
"*icond_type state_variable",
"*icond_param_value state_var_id 2",
"*icond_dof_value var %i"%homogenization,
"*add_icond_elements",
"all_existing",
]
for grain,elementList in enumerate(elements):
cmds.append([\
"*new_icond",
"*icond_name microstructure_%i"%(grain+1),
"*icond_type state_variable",
"*icond_param_value state_var_id 3",
"*icond_dof_value var %i"%(grain+1),
"*add_icond_elements",
elementList,
"#",
])
return cmds
#--------------------
def parse_geomFile(content,homog):
#--------------------
(skip,key) = content[0].split()[:2]
if key[:4].lower() == 'head':
skip = int(skip)+1
else:
skip = 0
res = [0,0,0]
dim = [0.0,0.0,0.0]
homog = 0
for line in content[:skip]:
data = line.split()
if data[0].lower() == 'resolution':
res = map(int,data[2:8:2])
if data[0].lower() == 'dimension':
dim = map(float,data[2:8:2])
if data[0].lower() == 'homogenization':
homog = int(data[1])
grains = []
for line in content[skip:]:
grains.append(int(line.split()[0]))
return (res,dim,homog,grains)
#--------------------
def parse_spectralFile(content,homog):
#--------------------
coords = [{},{},{}]
maxBox = [-1.0e20,-1.0e20,-1.0e20]
minBox = [ 1.0e20, 1.0e20, 1.0e20]
dim = [0.0,0.0,0.0]
res = [0,0,0]
grains = []
for line in content:
data = line.split()[3:7]
grains.append(int(data[3]))
for i in range(3):
maxBox[i] = max(maxBox[i],float(data[i]))
minBox[i] = min(minBox[i],float(data[i]))
coords[i][data[i]] = True
for i in range(3):
res[i] = len(coords[i])
dim[i] = (maxBox[i]-minBox[i])*res[i]/(res[i]-1.0)
return (res,dim,homog,grains)
# ----------------------- MAIN -------------------------------
parser = OptionParser(usage='%prog [options] spectral.datafile', description = """
Generate FE hexahedral mesh from spectral description file.
Acceptable formats are
geom: header plus list of grain numbers or
spectral: phi1,Phi,phi2,x,y,z,id,phase.
$Id$
""")
parser.add_option("-p", "--port", type="int",\
dest="port",\
help="Mentat connection port")
parser.add_option("-d", "--dimension", type="int", nargs=3,\
dest="d",\
help="physical dimension")
parser.add_option("--homogenization", type="int",\
dest="homogenization",\
help="index of homogenization to be chosen from material.config file")
parser.add_option("-s", "--spectral", action="store_const", const="spectral",\
dest="filetype",\
help="file has 'spectral' format")
parser.add_option("-g", "--geom", action="store_const", const="geom",\
dest="filetype",\
help="file has 'geom' format")
parser.set_defaults(homogenization = 1)
(options, args) = parser.parse_args()
try:
file = open('%s/../MSCpath'%os.path.dirname(os.path.realpath(sys.argv[0])))
MSCpath = os.path.normpath(file.readline().strip())
file.close()
except:
MSCpath = '/msc'
for release,subdirs in sorted(releases.items(),reverse=True):
for subdir in subdirs:
libPath = '%s/mentat%s/shlib/%s'%(MSCpath,release,subdir)
if os.path.exists(libPath):
sys.path.append(libPath)
break
else:
continue
break
try:
from py_mentat import *
except:
print('no valid Mentat release found in %s'%MSCpath)
if options.port != None: sys.exit(-1)
if not os.path.isfile(args[0]):
parser.error("cannot open %s"%args[0])
file = open(args[0])
content = file.readlines()
file.close()
if options.filetype not in ['spectral','geom']:
options.filetype = os.path.splitext(args[0])[1][1:]
print '\nparsing %s...'%options.filetype,
sys.stdout.flush()
(res,dim,homog,grains) = {\
'geom': parse_geomFile,
'spectral': parse_spectralFile,
}[options.filetype](content,options.homogenization)
print '%i grains in %s with resolution %s and homogenization %i\n'%(len(list(set(grains))),str(dim),str(res),homog)
cmds = [\
init(),
mesh(res,dim),
material(),
geometry(),
initial_conditions(homog,grains),
'*identify_sets',
'*redraw',
]
outputLocals = {}
if (options.port != None):
py_connect('',options.port)
output(cmds,outputLocals,'Mentat')
py_disconnect()
else:
output(cmds,outputLocals,'Stdout')