diff --git a/processing/post/addTable.py b/processing/post/addTable.py new file mode 100755 index 000000000..d2abe562b --- /dev/null +++ b/processing/post/addTable.py @@ -0,0 +1,77 @@ +#!/usr/bin/env python +# -*- coding: UTF-8 no BOM -*- + +import os,sys,string +from optparse import OptionParser +import damask + +scriptID = string.replace('$Id$','\n','\\n') +scriptName = os.path.splitext(scriptID.split()[1])[0] + +# -------------------------------------------------------------------- +# MAIN +# -------------------------------------------------------------------- + +parser = OptionParser(option_class=damask.extendableOption, usage='%prog options [file[s]]', description = """ +Append data of ASCIItable(s). + +""", version = scriptID) + +parser.add_option('-a', '--add','--table', + dest = 'table', + action = 'extend', metavar = '', + help = 'tables to add') + +(options,filenames) = parser.parse_args() + +# --- loop over input files ------------------------------------------------------------------------- + +if filenames == []: filenames = [None] + +for name in filenames: + try: + table = damask.ASCIItable(name = name, + buffered = False) + except: continue + + table.croak(damask.util.emph(scriptName)+(': '+name if name else '')) + tables = [] + for addTable in options.table: + try: + tables.append(damask.ASCIItable(name = addTable, + buffered = False, + readonly = True) + ) + except: continue + +# ------------------------------------------ read headers ------------------------------------------ + + table.head_read() + for addTable in tables: + addTable.head_read() + +# ------------------------------------------ assemble header -------------------------------------- + + table.info_append(scriptID + '\t' + ' '.join(sys.argv[1:])) + + for addTable in tables: + table.labels_append(addTable.labels) # extend ASCII header with new labels + + table.head_write() + +# ------------------------------------------ process data ------------------------------------------ + + outputAlive = True + while outputAlive and table.data_read(): + for addTable in tables: + outputAlive = addTable.data_read() # read next table's data + if not outputAlive: break + table.data_append(addTable.data) # append to master table + if outputAlive: + outputAlive = table.data_write() # output processed line + +# ------------------------------------------ output finalization ----------------------------------- + + table.close() # close ASCII tables + for addTable in tables: + addTable.close() \ No newline at end of file