updated to new ASCII table style

processing/post/vtk_addVoxelcloudData.py

@@ -3,6 +3,7 @@
 
 import os,sys,string,re,vtk
 import damask
+from collections import defaultdict
 from optparse import OptionParser
 
 scriptID   = string.replace('$Id$','\n','\\n')
@@ -24,18 +25,13 @@ parser.add_option('-s', '--scalar',   dest='scalar', action='extend', \
 parser.add_option('-c', '--color',   dest='color', action='extend', \
                   help = 'RGB color tuples')
 
-parser.set_defaults(scalar = [])
+parser.set_defaults(scalar = [],
-parser.set_defaults(color = [])
+                    color = [],
+                    render = False,
+)
 
 (options, filenames) = parser.parse_args()
 
-datainfo = {                                                               # list of requested labels per datatype
-             'scalar':     {'len':1,
-                            'label':[]},
-             'color':      {'len':3,
-                            'label':[]},
-           }
-
 if options.vtk == None or not os.path.exists(options.vtk):
   parser.error('VTK file does not exist')
 
@@ -52,86 +48,75 @@ elif os.path.splitext(options.vtk)[1] == '.vtk':
 else:
   parser.error('unsupported VTK file type extension')
 
-
 Npoints = uGrid.GetNumberOfPoints()
 Ncells  = uGrid.GetNumberOfCells()
 
-#if Npoints != Ncells or Npoints != Nvertices:
+sys.stderr.write('{}: {} points and {} cells...\n'.format(damask.util.emph(options.vtk),Npoints,Ncells))
-#  parser.error('Number of points, cells, and vertices in VTK differ from each other'); sys.exit()
-if options.scalar != None:  datainfo['scalar']['label'] += options.scalar
-if options.color  != None:  datainfo['color']['label']  += options.color
 
-# ------------------------------------------ setup file handles ---------------------------------------  
+# --- loop over input files -------------------------------------------------------------------------
 
-files = []
+if filenames == []: filenames = [None]
-if filenames == []:
-  files.append({'name':'STDIN', 'input':sys.stdin, 'output':sys.stdout, 'croak':sys.stderr})
-else:
-  for name in filenames:
-    if os.path.exists(name):
-      files.append({'name':name, 'input':open(name), 'output':sys.stderr, 'croak':sys.stderr})
 
-#--- loop over input files ------------------------------------------------------------------------
+for name in filenames:
-for file in files:
+  try:
-  if file['name'] != 'STDIN': file['croak'].write('\033[1m'+scriptName+'\033[0m: '+file['name']+'\n')
+    table = damask.ASCIItable(name = name,
-  else: file['croak'].write('\033[1m'+scriptName+'\033[0m\n')
+                              buffered = False, readonly = True)
+  except: continue
+  table.croak(damask.util.emph(scriptName)+(': '+name if name else ''))
 
-  table = damask.ASCIItable(file['input'],file['output'],False)             # make unbuffered ASCII_table
+# --- interpret header ----------------------------------------------------------------------------
-  table.head_read()                                                         # read ASCII header info
 
-# --------------- figure out columns to process
+  table.head_read()
-  active = {}
-  column = {}
 
-  array = {}
+  remarks = []
+  errors  = []
+  VTKarray = {}
+  active = defaultdict(list)
   
-  for datatype,info in datainfo.items():
+  for datatype,dimension,label in [['scalar',1,options.scalar],
-    for label in info['label']:
+                                   ['color',3,options.color],
-      foundIt = False
+                                   ]:
-      for key in ['1_'+label,label]:
+    for i,dim in enumerate(table.label_dimension(label)):
-        if key in table.labels:
+      me = label[i]
-          foundIt = True
+      if dim == -1: remarks.append('{} "{}" not found...'.format(datatype,me))
-          if datatype not in active: active[datatype] = []
+      elif dim > dimension: remarks.append('"{}" not of dimension{}...'.format(me,dimension))
-          if datatype not in column: column[datatype] = {}
+      else:
-          if datatype not in array:   array[datatype] = {}
+        table.croak('adding {} {}'.format(datatype,me))
-          active[datatype].append(label)
+        active[datatype].append(me)
-          column[datatype][label] = table.labels.index(key)                   # remember columns of requested data
+
-          if datatype == 'scalar':
+        if datatype in ['scalar']:
-            array[datatype][label] = vtk.vtkDoubleArray()
+          VTKarray[me] = vtk.vtkDoubleArray()
-            array[datatype][label].SetNumberOfComponents(1)
-            array[datatype][label].SetName(label)
         elif datatype == 'color':
-            array[datatype][label] = vtk.vtkUnsignedCharArray()
+          VTKarray[me] = vtk.vtkUnsignedCharArray()
-            array[datatype][label].SetNumberOfComponents(3)
+
-            array[datatype][label].SetName(label)
+        VTKarray[me].SetNumberOfComponents(dimension)
-      if not foundIt:
+        VTKarray[me].SetName(label[i])
-        file['croak'].write('column %s not found...\n'%label)
+
+  if remarks != []: table.croak(remarks)
+  if errors  != []:
+    table.croak(errors)
+    table.close(dismiss=True)
+    continue
        
 # ------------------------------------------ process data ---------------------------------------  
 
   while table.data_read():                                                                          # read next data line of ASCII table
     
     for datatype,labels in active.items():                                                          # loop over scalar,color
-      for label in labels:                                                  # loop over all requested items
+      for me in labels:                                                                             # loop over all requested items
-        theData = table.data[column[datatype][label]:\
+        theData = [table.data[i] for i in table.label_indexrange(me)]                               # read strings
-                             column[datatype][label]+datainfo[datatype]['len']]  # read strings
         if datatype == 'color':
-          theData = map(lambda x: int(255.*float(x)),theData)
+          VTKarray[me].InsertNextTuple3(*map(lambda x: int(255.*float(x)),theData))
-          array[datatype][label].InsertNextTuple3(theData[0],theData[1],theData[2],)
         elif datatype == 'scalar':
-          array[datatype][label].InsertNextValue(float(theData[0]))
+          VTKarray[me].InsertNextValue(float(theData[0]))
-
-  table.input_close()                                                     # close input ASCII table
 
 # ------------------------------------------ add data ---------------------------------------  
 
   for datatype,labels in active.items():                                   # loop over scalar,color
     if datatype == 'color':
-#      Polydata.GetPointData().SetScalars(array[datatype][labels[0]])
+      uGrid.GetCellData().SetScalars(VTKarray[active['color'][0]])
-      uGrid.GetCellData().SetScalars(array[datatype][labels[0]])
     for label in labels:                                                   # loop over all requested items
-#      Polydata.GetPointData().AddArray(array[datatype][label])
+      uGrid.GetCellData().AddArray(VTKarray[me])
-      uGrid.GetCellData().AddArray(array[datatype][label])
 
   uGrid.Modified()
   if vtk.VTK_MAJOR_VERSION <= 5: