DAMASK_EICMD/processing/pre/gmsh_identifySurfaces.py

81 lines
1.7 KiB
Python
Raw Normal View History

import numpy as np
import re
def func(seq):
for x in seq:
try:
yield float(x)
except ValueError:
yield x
my_geofile = 'polyXtal_5grains.geo'
PointCount = 0
LineCount = 0
LineLoopCount = 0
PlaneSurfaceCount = 0
SurfaceLoopCount = 0
point = []
line = []
lineloop = []
plane = []
surface = []
f = open(my_geofile,'r')
lines = f.readlines()
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(","))))
elif eachline.startswith('Plane', 0, len(eachline)):
PlaneSurfaceCount += 1
r = re.compile('{(.*?)}')
m = r.search(eachline)
if m:
a = m.group(1)
plane.append(list(func(a.split(","))))
elif eachline.startswith('Surface', 0,len(eachline)):
SurfaceLoopCount += 1
r = re.compile('{(.*?)}')
m = r.search(eachline)
if m:
a = m.group(1)
surface.append(list(func(a.split(","))))
print point
print line
print 'll'
print np.size(lineloop)
print lineloop
print 'ss'
print surface
#for sn, ll in enumerate(lineloop):
# for l in ll:
# for points in l[abs(int(l))]:
# if point[int(points)-1][0] == float(0):
# print 'abc'