DAMASK_EICMD/processing/legacy/gmsh_identifySurfaces.py

197 lines
7.4 KiB
Python
Raw Normal View History

#!/usr/bin/env python2.7
# -*- coding: UTF-8 no BOM -*-
2016-03-02 17:13:09 +05:30
import os,re
from optparse import OptionParser
import damask
scriptName = os.path.splitext(os.path.basename(__file__))[0]
scriptID = ' '.join([scriptName,damask.version])
def all_same(items,a):
return all(x == a for x in items)
def func(seq):
for x in seq:
try:
yield float(x)
except ValueError:
yield x
#--------------------------------------------------------------------------------------------------
# MAIN
#--------------------------------------------------------------------------------------------------
parser = OptionParser(option_class=damask.extendableOption, usage='%prog options [file[s]]', description = """
Recognize bounding surfaces and append them as physical sufaces in the geo file.
""", version = scriptID)
parser.add_option('-n','--numvol', dest = 'N', type='int', metavar='int',
2014-10-21 20:12:13 +05:30
help='number of physical volumes' )
parser.add_option('-f','--faces', dest = 'surfaces', action = 'extend', type = 'string', metavar = '<string LIST>',
help = 'surfaces to tag {x, y, z}')
parser.add_option('-s','--size', dest = 'size', type='float', metavar='float',
help='mesh size [%default]' )
parser.add_option('-d','--dim', dest = 'dimension', type='int', metavar='int',
help='dimension of geometry [%default]' )
(options, filename) = parser.parse_args()
parser.set_defaults(size = 0.1,
dimension = 3)
my_geofile = filename[0]
numVol = options.N
PointCount = 0
LineCount = 0
LineLoopCount = 0
point = []
line = []
lineloop = []
f = open(my_geofile,'r')
lines = f.readlines()
f.close()
for eachline in lines:
if eachline.startswith('Point', 0, 5):
PointCount += 1
r = re.compile('{(.*?)}')
m = r.search(eachline)
if m:
a = m.group(1)
point.append(list(func(a.split(","))))
elif eachline.startswith('Line (', 0, 6):
LineCount += 1
r = re.compile('{(.*?)}')
m = r.search(eachline)
if m:
a = m.group(1)
line.append(list(func(a.split(","))))
elif eachline.startswith('Line Loop', 0, 9):
LineLoopCount += 1
r = re.compile('{(.*?)}')
m = r.search(eachline)
if m:
a = m.group(1)
lineloop.append(list(func(a.split(","))))
x_coord = []; y_coord = []; z_coord = []
xp = []; xm = []
yp = []; ym = []
zp = []; zm = []
xmin = min([x[0] for x in point]); xmax = max([x[0] for x in point])
ymin = min([x[1] for x in point]); ymax = max([x[1] for x in point])
zmin = min([x[2] for x in point]); zmax = max([x[2] for x in point])
if (options.dimension == 3):
for i,l in enumerate(lineloop):
for lines in l:
for pts in line[int(abs(lines)-1)]:
x_coord.append(point[int(pts)-1][0])
y_coord.append(point[int(pts)-1][1])
z_coord.append(point[int(pts)-1][2])
if all_same(x_coord,xmax) and any([surface == 'x' for surface in options.surfaces]):
xp.append(int(i+1))
elif all_same(x_coord,xmin) and any([surface == 'x' for surface in options.surfaces]):
xm.append(int(i+1))
elif all_same(y_coord,ymax) and any([surface == 'y' for surface in options.surfaces]):
yp.append(int(i+1))
elif all_same(y_coord,ymin) and any([surface == 'y' for surface in options.surfaces]):
ym.append(int(i+1))
elif all_same(z_coord,zmax) and any([surface == 'z' for surface in options.surfaces]):
zp.append(int(i+1))
elif all_same(z_coord,zmin) and any([surface == 'z' for surface in options.surfaces]):
zm.append(int(i+1))
x_coord = []
y_coord = []
z_coord = []
with open(my_geofile,'a') as f:
f.write('Delete Physicals; \n')
if any([surface == 'x' for surface in options.surfaces]):
f.write('%s%s%s\n' %('Physical Surface(1) = {',','.join(map(str, xp)),'};'))
f.write('%s%s%s\n' %('Physical Surface(2) = {',','.join(map(str, xm)),'};'))
if any([surface == 'y' for surface in options.surfaces]):
f.write('%s%s%s\n' %('Physical Surface(3) = {',','.join(map(str, yp)),'};'))
f.write('%s%s%s\n' %('Physical Surface(4) = {',','.join(map(str, ym)),'};'))
if any([surface == 'z' for surface in options.surfaces]):
f.write('%s%s%s\n' %('Physical Surface(5) = {',','.join(map(str, zp)),'};'))
f.write('%s%s%s\n' %('Physical Surface(6) = {',','.join(map(str, zm)),'};'))
for i in range(numVol):
f.write('%s%d%s%d%s\n' %('Physical Volume (', i+1,') = {',i+1,'};'))
f.write('Field[1] = Box;\n')
f.write('%s%f%s\n' %('Field[1].VIn = ', options.size,';'))
f.write('%s%f%s\n' %('Field[1].VOut = ',options.size,';'))
f.write('%s%f%s\n' %('Field[1].XMin = ',xmin,';'))
f.write('%s%f%s\n' %('Field[1].XMax = ',xmax,';'))
f.write('%s%f%s\n' %('Field[1].YMin = ',ymin,';'))
f.write('%s%f%s\n' %('Field[1].YMax = ',ymax,';'))
f.write('%s%f%s\n' %('Field[1].ZMin = ',zmin,';'))
f.write('%s%f%s\n' %('Field[1].ZMax = ',zmax,';'))
f.write('Background Field = 1;\n')
f.close()
elif (options.dimension == 2):
for i,l in enumerate(line):
2016-03-02 17:13:09 +05:30
# for pts in line[int(abs(lines)-1)]:
for pts in l:
x_coord.append(point[int(pts)-1][0])
y_coord.append(point[int(pts)-1][1])
z_coord.append(point[int(pts)-1][2])
if all_same(x_coord,xmax) and any([surface == 'x' for surface in options.surfaces]):
xp.append(int(i+1))
elif all_same(x_coord,xmin) and any([surface == 'x' for surface in options.surfaces]):
xm.append(int(i+1))
elif all_same(y_coord,ymax) and any([surface == 'y' for surface in options.surfaces]):
yp.append(int(i+1))
elif all_same(y_coord,ymin) and any([surface == 'y' for surface in options.surfaces]):
ym.append(int(i+1))
elif all_same(z_coord,zmax) and any([surface == 'z' for surface in options.surfaces]):
zp.append(int(i+1))
elif all_same(z_coord,zmin) and any([surface == 'z' for surface in options.surfaces]):
zm.append(int(i+1))
x_coord = []
y_coord = []
z_coord = []
with open(my_geofile,'a') as f:
f.write('Delete Physicals; \n')
if any([surface == 'x' for surface in options.surfaces]):
f.write('%s%s%s\n' %('Physical Line(1) = {',','.join(map(str, xp)),'};'))
f.write('%s%s%s\n' %('Physical Line(2) = {',','.join(map(str, xm)),'};'))
if any([surface == 'y' for surface in options.surfaces]):
f.write('%s%s%s\n' %('Physical Line(3) = {',','.join(map(str, yp)),'};'))
f.write('%s%s%s\n' %('Physical Line(4) = {',','.join(map(str, ym)),'};'))
if any([surface == 'z' for surface in options.surfaces]):
f.write('%s%s%s\n' %('Physical Line(5) = {',','.join(map(str, zp)),'};'))
f.write('%s%s%s\n' %('Physical Line(6) = {',','.join(map(str, zm)),'};'))
for i in range(numVol):
f.write('%s%d%s%d%s\n' %('Physical Surface (', i+1,') = {',i+1,'};'))
f.write('Field[1] = Box;\n')
f.write('%s%f%s\n' %('Field[1].VIn = ', options.size,';'))
f.write('%s%f%s\n' %('Field[1].VOut = ',options.size,';'))
f.write('%s%f%s\n' %('Field[1].XMin = ',xmin,';'))
f.write('%s%f%s\n' %('Field[1].XMax = ',xmax,';'))
f.write('%s%f%s\n' %('Field[1].YMin = ',ymin,';'))
f.write('%s%f%s\n' %('Field[1].YMax = ',ymax,';'))
f.write('%s%f%s\n' %('Field[1].ZMin = ',zmin,';'))
f.write('%s%f%s\n' %('Field[1].ZMax = ',zmax,';'))
f.write('Background Field = 1;\n')
f.close()