DAMASK_EICMD/processing/pre/spectral_geomPack.py

133 lines
5.4 KiB
Python
Raw Normal View History

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os,re,sys,math,string
from optparse import OptionParser, OptionGroup, Option, SUPPRESS_HELP
def writeKey(outType, begin, counter):
if outType == 'range':
a=str(begin) +' to '+ str(begin+counter-1)+'\n'
file['output'].write(a)
if outType == 'copy':
b=str(counter)+ ' copies of '+str(begin)+'\n'
file['output'].write(b)
if outType == 'none':
c=str(begin)+'\n'
file['output'].write(c)
# -----------------------------
class extendedOption(Option):
# -----------------------------
# used for definition of new option parser action 'extend', which enables to take multiple option arguments
# taken from online tutorial http://docs.python.org/library/optparse.html
ACTIONS = Option.ACTIONS + ("extend",)
STORE_ACTIONS = Option.STORE_ACTIONS + ("extend",)
TYPED_ACTIONS = Option.TYPED_ACTIONS + ("extend",)
ALWAYS_TYPED_ACTIONS = Option.ALWAYS_TYPED_ACTIONS + ("extend",)
def take_action(self, action, dest, opt, value, values, parser):
if action == "extend":
lvalue = value.split(",")
values.ensure_value(dest, []).extend(lvalue)
else:
Option.take_action(self, action, dest, opt, value, values, parser)
# --------------------------------------------------------------------
# MAIN
# --------------------------------------------------------------------
parser = OptionParser(option_class=extendedOption, usage='%prog options [file[s]]', description = """
Pack/unpack geometry files by adding or removing range indicator "to" and copy indicator "copies of".
""" + string.replace('$Id$','\n','\\n')
)
parser.add_option('-p','--pack', dest='pack', action='store_true', \
help='pack input file [%default]')
parser.add_option('-u','--unpack', dest='unpack', action='store_true', \
help='unpack input file [%default]')
parser.add_option('-l','--linelength', dest='lineLength',type='int', nargs=1, \
help='length of line, leave blank for auto (FPs in x direction)')
parser.set_defaults(pack = False)
parser.set_defaults(unpack = False)
parser.set_defaults(lineLength = 0)
(options,filenames) = parser.parse_args()
if options.pack and options.unpack:
print 'cannot pack and unpack at the same time'
# ------------------------------------------ setup file handles ---------------------------------------
files = []
if filenames == []:
files.append({'name':'STDIN', 'input':sys.stdin, 'output':sys.stdout})
else:
for name in filenames:
if os.path.exists(name):
files.append({'name':name, 'input':open(name), 'output':open(name+'_tmp','w+')})
# ------------------------------------------ loop over input files ---------------------------------------
for file in files:
digits=2
counter = 1
begin = -1
outType = ''
current=-1 # current material ID
wordsWritten=0 # number of words per line in the output file
currentLine=0
if file['name'] != 'STDIN': print file['name']
for line in file['input']:
words=line.split()
currentLine+=1
if('head' in line): headNumber=int(words[0])+1
if(currentLine<=headNumber):
if('maxgraincount'==words[0].lower()): digits = 1+int(math.log10(int(words[1])))
if(words[0].lower()=='resolution' and words[1].lower()=='a' and options.lineLength==0):
options.lineLength = int(words[2].lower())
file['output'].write(line)
else:
if options.unpack: # unpacking the geometry file
words = map(str.lower,line.split())
if len(words) > 1: # any packing keywords?
if (words[1] == 'to'): line = ' '.join(map(str,range(int(words[0]),int(words[2])+1)))
if (words[1] == 'copies' and words[2] == 'of'): line = ' '.join([words[3]]*int(words[0]))
for word in line.split():
wordsWritten += 1
file['output'].write(word.zfill(digits)+{True:'\n',False:' '}[wordsWritten%options.lineLength == 0]) # newline every options.linelength words
if options.pack: # packing the geometry file
numbers=line.split()
i=len(numbers)
for x in xrange(i):
last=current
current = int(numbers[x])
if current == last+1 and current==begin+counter:
counter+=1
outType = 'range'
elif current == begin and current == last:
counter+=1
outType = 'copy'
else:
writeKey(outType, begin, counter)
counter = 1
outType= 'none'
begin = current
writeKey(outType, begin, counter) # just for packing (last line), outType not defined for unpacking
file['output'].close()
file['input'].close()
os.rename(file['name']+'_tmp',file['name'])