Script to merge multiple ASCIItables. Stops at end of shortest one.

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 = '<string LIST>',
+                  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()