added class ASCII_TABLE for easier processing of those...

lib/damask_tools.py

@@ -1,5 +1,6 @@
 import os,string,re
 
+
 class DAMASK_TOOLS():
 
   def check_env(self):
@@ -10,9 +11,141 @@ class DAMASK_TOOLS():
     else:
       return True       
 
-class MATERIAL_CONFIG():
-  __slots__ = ['data']
 
+class ASCII_TABLE():
+  import sys
+  
+  __slots__ = ['__IO__',
+               'info',
+               'labels',
+               'data',
+              ]
+
+
+#.............................................
+  def __init__(self,
+               fileIn = sys.stdin,
+               fileOut = sys.stdout,
+               buffered = True):
+
+    self.__IO__ = {'in': fileIn,
+                   'out':fileOut,
+                   'output':[],
+                   'buffered':buffered,
+                   'validReadSize': 0,
+                  }
+    self.info = []
+    self.labels = []
+    self.data = []
+
+
+#.............................................
+  def output_write(self,
+                   what):
+
+    if isinstance(what,list):
+      for item in what: self.output_write(item)
+    else:
+      self.__IO__['output'] += [str(what)]
+      self.__IO__['buffered'] or self.output_flush()
+
+
+#.............................................
+  def output_flush(self,
+                   clear = True):
+
+    self.__IO__['output'] == [] or self.__IO__['out'].write('\n'.join(self.__IO__['output']) + '\n')
+    if clear: self.output_clear()
+
+
+#.............................................
+  def output_clear(self):
+
+    self.__IO__['output'] = []
+
+
+#.............................................
+  def head_read(self):
+  #  get column labels by either read the first row, or - if keyword "head[*]" is present - the last line of the header
+
+    try:
+      self.__IO__['in'].seek(0)
+    except:
+      pass
+
+    firstline = self.__IO__['in'].readline()
+    m = re.search('(\d+)\s*head', firstline.lower())
+    if m:
+      self.info     = [self.__IO__['in'].readline().strip() for i in xrange(1,int(m.group(1)))]
+      self.labels   =  self.__IO__['in'].readline().split()
+    else:
+      self.info     = []
+      self.labels   = firstline.split()
+  
+    self.__IO__['validReadSize'] = len(self.labels)
+
+
+#.............................................
+  def head_write(self):
+
+    self.output_write (['%i\theader'%(len(self.info)+1),
+                        self.info,
+                        '\t'.join(self.labels)])
+
+
+#.............................................
+  def labels_append(self,
+                   what):
+
+    if isinstance(what,list):
+      for item in what: self.labels_append(item)
+    else:               self.labels += [str(what)]
+
+
+#.............................................
+  def info_append(self,
+                  what):
+
+    if isinstance(what,list):
+      for item in what: self.info_append(item)
+    else:               self.info += [str(what)]
+
+
+#.............................................
+  def data_read(self):
+
+    line = self.__IO__['in'].readline()
+    items = line.split()[:self.__IO__['validReadSize']]                             # get next data row
+    self.data = {False:   [],
+                  True: items}[len(items) == self.__IO__['validReadSize']]          # take if correct number of entries
+
+    return line != ''
+
+#.............................................
+  def data_write(self):
+
+    if isinstance(self.data[0],list):
+      self.output_write (['\t'.join(map(str,items)) for items in self.data])
+    else:
+      self.output_write ('\t'.join(map(str,self.data)))
+
+
+#.............................................
+  def data_append(self,
+                  what):
+
+    if isinstance(what,list):
+      for item in what: self.data_append(item)
+    else:               self.data += [str(what)]
+
+
+
+
+
+    
+class MATERIAL_CONFIG():
+
+  __slots__ = ['data']
   
 
   def __init__(self):