DAMASK_EICMD/processing/pre/spectral_geomScale.py

54 lines
2.4 KiB
Python

#!/usr/bin/env python
# -*- coding: UTF-8 no BOM -*-
#scale the size of a given .geom file (first argument, without extension) in x y z direction by given commandline argument
#(integer in each direction)
#be carefull, last line should be empty (line break after last number)
import sys, time, re
scale = [0 for i in range(3)]
scale[0] = int(sys.argv[2])
scale[1] = int(sys.argv[3])
scale[2] = int(sys.argv[4])
resolution = [3 for i in range(3)]
dimension = [0.0 for i in range(3)]
file_in = open(sys.argv[1]+'.geom','r')
for i in range(3):
input = file_in.readline()
if re.match('resolution',input):
resolution[0] = int(re.findall('\S*',''.join(re.findall('a\s*\S*',input)))[-2])
resolution[1] = int(re.findall('\S*',''.join(re.findall('b\s*\S*',input)))[-2])
resolution[2] = int(re.findall('\S*',''.join(re.findall('c\s*\S*',input)))[-2])
if re.match('dimension',input):
dimension[0] = float(re.findall('\S*',''.join(re.findall('x\s*\S*',input)))[-2])
dimension[1] = float(re.findall('\S*',''.join(re.findall('y\s*\S*',input)))[-2])
dimension[2] = float(re.findall('\S*',''.join(re.findall('z\s*\S*',input)))[-2])
output = ''
for x in range(resolution[0]*resolution[1]*resolution[2]):
a=file_in.readline()
for i in range(scale[0]):
output += a
output_separated = output.splitlines(True)
resolution[0] = resolution[0]*scale[0] #scale resolution 1
output = ''
for z in range(resolution[2]):
for y in range(resolution[1]):
for i in range(scale[1]):
output += ''.join(output_separated[z*resolution[0]*resolution[1]+y*resolution[0]:\
z*resolution[0]*resolution[1]+y*resolution[0]+resolution[0]])
resolution[1] = resolution[1]*scale[1] #scale resolution 2
output_separated = output.splitlines(True)
output = ''
for z in range(resolution[2]):
for i in range(scale[2]):
output += ''.join(output_separated[z*resolution[0]*resolution[1]:\
z*resolution[0]*resolution[1]+resolution[0]*resolution[1]])
resolution[2] = resolution[2]*scale[2] #scale resolution 2
file_out = open(sys.argv[1]+'_scaled_'+str(sys.argv[2])+'-'+str(sys.argv[3])+'-'+str(sys.argv[4])+'.geom','w')
file_out.write('resolution a '+str(resolution[0])+' b '+str(resolution[1])+' z '+str(resolution[2])+'\n')
file_out.write('dimension x '+str(dimension[0])+' y '+str(dimension[1])+' z '+str(dimension[2])+'\n')
file_out.write('homogenization 1\n')
file_out.write(output)