133 lines
5.4 KiB
Python
Executable File
133 lines
5.4 KiB
Python
Executable File
#!/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'])
|
|
|