From a87e396b83a9b5a438812fb90fa3d87361f1626a Mon Sep 17 00:00:00 2001
From: Martin Diehl <mail@martin-diehl.net>
Date: Thu, 2 Jan 2020 11:31:33 +0100
Subject: [PATCH] bugfix: need to addData to vtk

not really sure why 'SetScalars' was used for color. 'AddArray' seems to
work. Also, there seems to be no difference between 'data' and 'tensor'
---
 processing/post/vtk_addGridData.py       | 13 ++++---------
 processing/post/vtk_addPointCloudData.py |  8 +++++---
 2 files changed, 9 insertions(+), 12 deletions(-)

diff --git a/processing/post/vtk_addGridData.py b/processing/post/vtk_addGridData.py
index 1596cd7fe..8e76cfca8 100755
--- a/processing/post/vtk_addGridData.py
+++ b/processing/post/vtk_addGridData.py
@@ -2,8 +2,8 @@
 
 import os
 import sys
+from io import StringIO
 from optparse import OptionParser
-from collections import defaultdict
 
 import vtk
 from vtk.util import numpy_support
@@ -120,14 +120,9 @@ for name in filenames:
 
   damask.util.croak('{} mode...'.format(mode))
 
-  for datatype,labels in active.items():                                                            # loop over scalar,color
-    if datatype == 'color':
-      if   mode == 'cell':  rGrid.GetCellData().SetScalars(VTKarray[active['color'][0]])
-      elif mode == 'point': rGrid.GetPointData().SetScalars(VTKarray[active['color'][0]])
-    for me in labels:                                                                               # loop over all requested items
-      if   mode == 'cell':  rGrid.GetCellData().AddArray(VTKarray[me])
-      elif mode == 'point': rGrid.GetPointData().AddArray(VTKarray[me])
-
+  for data in VTKarray:
+    if   mode == 'cell':  rGrid.GetCellData().AddArray(VTKarray[data])
+    elif mode == 'point': rGrid.GetPointData().AddArray(VTKarray[data])
   rGrid.Modified()
 
 # ------------------------------------------ output result ---------------------------------------
diff --git a/processing/post/vtk_addPointCloudData.py b/processing/post/vtk_addPointCloudData.py
index 5a40d967a..833bfc88e 100755
--- a/processing/post/vtk_addPointCloudData.py
+++ b/processing/post/vtk_addPointCloudData.py
@@ -2,8 +2,8 @@
 
 import os
 import sys
+from io import StringIO
 from optparse import OptionParser
-from collections import defaultdict
 
 import vtk
 from vtk.util import numpy_support
@@ -104,10 +104,12 @@ for name in filenames:
     VTKarray[tensor].SetName(tensor)
 
 
-# ------------------------------------------ output result ---------------------------------------
-
+  for data in VTKarray:
+     Polydata.GetPointData().AddArray(VTKarray[data])
   Polydata.Modified()
 
+# ------------------------------------------ output result ---------------------------------------
+
   writer = vtk.vtkXMLPolyDataWriter()
   writer.SetDataModeToBinary()
   writer.SetCompressorTypeToZLib()